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 logging
import os import os
import re import re
import ssl
import uuid import uuid
from operator import itemgetter from operator import itemgetter
from typing import ( from typing import (
@ -24,6 +25,7 @@ from typing import (
cast, cast,
) )
import certifi
import httpx import httpx
from httpx_sse import EventSource, aconnect_sse, connect_sse from httpx_sse import EventSource, aconnect_sse, connect_sse
from langchain_core.callbacks import ( 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}$") 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( def _create_retry_decorator(
llm: ChatMistralAI, llm: ChatMistralAI,
run_manager: Optional[ run_manager: Optional[
@ -518,6 +525,7 @@ class ChatMistralAI(BaseChatModel):
"Authorization": f"Bearer {api_key_str}", "Authorization": f"Bearer {api_key_str}",
}, },
timeout=self.timeout, timeout=self.timeout,
verify=global_ssl_context,
) )
# todo: handle retries and max_concurrency # todo: handle retries and max_concurrency
if not self.async_client: if not self.async_client:
@ -529,6 +537,7 @@ class ChatMistralAI(BaseChatModel):
"Authorization": f"Bearer {api_key_str}", "Authorization": f"Bearer {api_key_str}",
}, },
timeout=self.timeout, timeout=self.timeout,
verify=global_ssl_context,
) )
if self.temperature is not None and not 0 <= self.temperature <= 1: if self.temperature is not None and not 0 <= self.temperature <= 1: