github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 10:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

mistralai[patch]: standardize model params (#20163)

Browse Source 
Related to #20085
This commit is contained in:
Bagatur
2024-04-08 11:48:38 -05:00
committed by GitHub
parent 17182406f3
commit 3490d70238
7 changed files with 32 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/partners/mistralai/langchain_mistralai/chat_models.py
View File

@@ -186,7 +186,7 @@ class ChatMistralAI(BaseChatModel):
client: httpx.Client = Field(default=None) #: :meta private:
async_client: httpx.AsyncClient = Field(default=None) #: :meta private:
mistral_api_key: Optional[SecretStr] = None
mistral_api_key: Optional[SecretStr] = Field(default=None, alias="api_key")
endpoint: str = "https://api.mistral.ai/v1"
max_retries: int = 5
timeout: int = 120
@@ -202,6 +202,12 @@ class ChatMistralAI(BaseChatModel):
safe_mode: bool = False
streaming: bool = False
class Config:
"""Configuration for this pydantic object."""
allow_population_by_field_name = True
arbitrary_types_allowed = True
@property
def _default_params(self) -> Dict[str, Any]:
"""Get the default parameters for calling the API."""

6
libs/partners/mistralai/langchain_mistralai/embeddings.py
View File

@@ -29,15 +29,16 @@ class MistralAIEmbeddings(BaseModel, Embeddings):
.. code-block:: python
from langchain_mistralai import MistralAIEmbeddings
mistral = MistralAIEmbeddings(
model="mistral-embed",
mistral_api_key="my-api-key"
api_key="my-api-key"
)
"""
client: httpx.Client = Field(default=None) #: :meta private:
async_client: httpx.AsyncClient = Field(default=None) #: :meta private:
mistral_api_key: Optional[SecretStr] = None
mistral_api_key: Optional[SecretStr] = Field(default=None, alias="api_key")
endpoint: str = "https://api.mistral.ai/v1/"
max_retries: int = 5
timeout: int = 120
@@ -49,6 +50,7 @@ class MistralAIEmbeddings(BaseModel, Embeddings):
class Config:
extra = Extra.forbid
arbitrary_types_allowed = True
allow_population_by_field_name = True
@root_validator()
def validate_environment(cls, values: Dict) -> Dict: