xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-04 12:56:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 正向代理支持

Browse Source
This commit is contained in:
RockChinQ
2024-02-07 23:58:22 +08:00
parent cf583486e3
commit d86ad25f86
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/provider/requester/apis/chatcmpl.py
View File

@@ -7,6 +7,7 @@ from typing import AsyncGenerator
import openai
import openai.types.chat.chat_completion as chat_completion
import httpx
from pkg.provider.entities import Message
@@ -24,6 +25,9 @@ class OpenAIChatCompletion(api.LLMAPIRequester):
api_key="",
base_url=self.ap.provider_cfg.data['openai-config']['base_url'],
timeout=self.ap.provider_cfg.data['openai-config']['request-timeout'],
http_client=httpx.AsyncClient(
proxies=self.ap.proxy_mgr.get_forward_proxies()
)
)
async def _req(

10
pkg/utils/proxy.py
View File

@@ -18,14 +18,14 @@ class ProxyManager:
async def initialize(self):
self.forward_proxies = {
"http": os.getenv("HTTP_PROXY") or os.getenv("http_proxy"),
"https": os.getenv("HTTPS_PROXY") or os.getenv("https_proxy"),
"http://": os.getenv("HTTP_PROXY") or os.getenv("http_proxy"),
"https://": os.getenv("HTTPS_PROXY") or os.getenv("https_proxy"),
}
if 'http' in self.ap.system_cfg.data['network-proxies']:
self.forward_proxies['http'] = self.ap.system_cfg.data['network-proxies']['http']
self.forward_proxies['http://'] = self.ap.system_cfg.data['network-proxies']['http']
if 'https' in self.ap.system_cfg.data['network-proxies']:
self.forward_proxies['https'] = self.ap.system_cfg.data['network-proxies']['https']
self.forward_proxies['https://'] = self.ap.system_cfg.data['network-proxies']['https']
def get_forward_proxies(self) -> dict:
return self.forward_proxies
return self.forward_proxies.copy()