mirrors/langchain-gpt4free
0
0
Fork 0
mirror of https://github.com/MIDORIBIN/langchain-gpt4free.git synced 2024-12-24 03:32:57 +03:00
Code

fix: suppoer g4f v0.0.2

Browse Source
This commit is contained in:
MIDORIBIN 2023-08-18 21:22:22 +09:00
parent 1bdad4b922
commit a6da9a5d5f
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain_g4f/G4FLLM.py
View File

@ -1,18 +1,16 @@
from types import ModuleType
from typing import Any, List, Mapping, Optional, Union
from g4f import ChatCompletion
from g4f.models import Model
from g4f.Provider.base_provider import BaseProvider
from langchain.callbacks.manager import CallbackManagerForLLMRun
from langchain.llms.base import LLM
from langchain.llms.utils import enforce_stop_tokens
class G4FLLM(LLM):
# Model.model or str
model: Union[Model, str]
# Provider.Provider
provider: Optional[ModuleType] = None
provider: Optional[type[BaseProvider]] = None
auth: Optional[Union[str, bool]] = None
create_kwargs: Optional[dict[str, Any]] = None