This commit is contained in:
Bagatur
2024-09-04 11:05:36 -07:00
parent 40634d441a
commit e409a85a28
4 changed files with 13 additions and 20 deletions

View File

@@ -361,8 +361,8 @@ def _convert_message_to_mistral_chat_message(
class ChatMistralAI(BaseChatModel):
"""A chat model that uses the MistralAI API."""
client: httpx.Client = Field(default=None) #: :meta private:
async_client: httpx.AsyncClient = Field(default=None) #: :meta private:
client: httpx.Client = Field(default=None, exclude=True) #: :meta private:
async_client: httpx.AsyncClient = Field(default=None, exclude=True) #: :meta private:
mistral_api_key: Optional[SecretStr] = Field(
alias="api_key",
default_factory=secret_from_env("MISTRAL_API_KEY", default=None),

View File

@@ -45,24 +45,6 @@
'ChatMistralAI',
]),
'kwargs': dict({
'async_client': dict({
'id': list([
'httpx',
'AsyncClient',
]),
'lc': 1,
'repr': '<httpx.AsyncClient object at 0x1119d6990>',
'type': 'not_implemented',
}),
'client': dict({
'id': list([
'httpx',
'Client',
]),
'lc': 1,
'repr': '<httpx.Client object at 0x1119d3b90>',
'type': 'not_implemented',
}),
'endpoint': 'boo',
'max_concurrent_requests': 64,
'max_retries': 2,