fix(llms,embeddings): repair HTTP proxy support (httpx>=0.28) and preserve proxies in LlmFactory (#5447)
Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>
This commit is contained in:
co-authored by
kartik-mem0
parent
48f1d6f010
commit
7a9f03af3f
@@ -100,7 +100,7 @@ class LlmFactory:
|
||||
"top_k": config.top_k,
|
||||
"enable_vision": config.enable_vision,
|
||||
"vision_details": config.vision_details,
|
||||
"http_client_proxies": config.http_client,
|
||||
"http_client_proxies": config.http_client_proxies,
|
||||
}
|
||||
config_dict.update(kwargs)
|
||||
config = config_class(**config_dict)
|
||||
|
||||
@@ -0,0 +1,13 @@
|
||||
from typing import Dict, Optional, Union
|
||||
|
||||
import httpx
|
||||
|
||||
|
||||
def build_http_client(http_client_proxies: Optional[Union[Dict, str]]) -> Optional[httpx.Client]:
|
||||
if not http_client_proxies:
|
||||
return None
|
||||
if isinstance(http_client_proxies, dict):
|
||||
return httpx.Client(
|
||||
mounts={scheme: httpx.HTTPTransport(proxy=url) for scheme, url in http_client_proxies.items()}
|
||||
)
|
||||
return httpx.Client(proxy=http_client_proxies)
|
||||
Reference in New Issue
Block a user