github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 07:09:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

mistral[patch]: set global ssl context (#30189)

Browse Source
This commit is contained in:
ccurme 2025-03-09 17:27:41 -04:00 committed by GitHub
parent df459d0d5e
commit b209d46eb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5,6 +5,7 @@ import json
import logging
import os
import re
import ssl
import uuid
from operator import itemgetter
from typing import (
@ -24,6 +25,7 @@ from typing import (
cast,
)
import certifi
import httpx
from httpx_sse import EventSource, aconnect_sse, connect_sse
from langchain_core.callbacks import (
@ -87,6 +89,11 @@ logger = logging.getLogger(__name__)
TOOL_CALL_ID_PATTERN = re.compile(r"^[a-zA-Z0-9]{9}$")
# This SSL context is equivelent to the default `verify=True`.
# https://www.python-httpx.org/advanced/ssl/#configuring-client-instances
global_ssl_context = ssl.create_default_context(cafile=certifi.where())
def _create_retry_decorator(
llm: ChatMistralAI,
run_manager: Optional[
@ -518,6 +525,7 @@ class ChatMistralAI(BaseChatModel):
"Authorization": f"Bearer {api_key_str}",
},
timeout=self.timeout,
verify=global_ssl_context,
)
# todo: handle retries and max_concurrency
if not self.async_client:
@ -529,6 +537,7 @@ class ChatMistralAI(BaseChatModel):
"Authorization": f"Bearer {api_key_str}",
},
timeout=self.timeout,
verify=global_ssl_context,
)
if self.temperature is not None and not 0 <= self.temperature <= 1: