Add langchain embedding, update langchain LLM and version bump -> 0.1.84 (#2510)
This commit is contained in:
@@ -13,7 +13,7 @@ class BaseLlmConfig(ABC):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
model: Optional[str] = None,
|
||||
model: Optional[Union[str, Dict]] = None,
|
||||
temperature: float = 0.1,
|
||||
api_key: Optional[str] = None,
|
||||
max_tokens: int = 2000,
|
||||
@@ -41,8 +41,6 @@ class BaseLlmConfig(ABC):
|
||||
xai_base_url: Optional[str] = None,
|
||||
# LM Studio specific
|
||||
lmstudio_base_url: Optional[str] = "http://localhost:1234/v1",
|
||||
# Langchain specific
|
||||
langchain_provider: Optional[str] = None,
|
||||
):
|
||||
"""
|
||||
Initializes a configuration class instance for the LLM.
|
||||
@@ -89,8 +87,6 @@ class BaseLlmConfig(ABC):
|
||||
:type xai_base_url: Optional[str], optional
|
||||
:param lmstudio_base_url: LM Studio base URL to be use, defaults to "http://localhost:1234/v1"
|
||||
:type lmstudio_base_url: Optional[str], optional
|
||||
:param langchain_provider: Langchain provider to be use, defaults to None
|
||||
:type langchain_provider: Optional[str], optional
|
||||
"""
|
||||
|
||||
self.model = model
|
||||
@@ -127,6 +123,3 @@ class BaseLlmConfig(ABC):
|
||||
|
||||
# LM Studio specific
|
||||
self.lmstudio_base_url = lmstudio_base_url
|
||||
|
||||
# Langchain specific
|
||||
self.langchain_provider = langchain_provider
|
||||
|
||||
Reference in New Issue
Block a user