From 94ebf027197e5bf8faa58e83c6ce1f0a65441bfd Mon Sep 17 00:00:00 2001 From: mindspawn Date: Mon, 3 Jun 2024 20:41:59 -0700 Subject: [PATCH] Enable http_proxy use for openai calls. --- backend/apps/openai/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/apps/openai/main.py b/backend/apps/openai/main.py index 7429d94ed..472699f1d 100644 --- a/backend/apps/openai/main.py +++ b/backend/apps/openai/main.py @@ -186,7 +186,7 @@ async def fetch_url(url, key): timeout = aiohttp.ClientTimeout(total=5) try: headers = {"Authorization": f"Bearer {key}"} - async with aiohttp.ClientSession(timeout=timeout) as session: + async with aiohttp.ClientSession(timeout=timeout, trust_env=True) as session: async with session.get(url, headers=headers) as response: return await response.json() except Exception as e: @@ -462,7 +462,7 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)): streaming = False try: - session = aiohttp.ClientSession() + session = aiohttp.ClientSession(trust_env=True) r = await session.request( method=request.method, url=target_url,