liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix proxy

Browse Source
This commit is contained in:
Changhua
2024-02-18 17:36:58 +08:00
parent 27277c97d2
commit c41e81f51f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
base/func_chatgpt.py
View File

@@ -4,6 +4,7 @@
import logging import logging
from datetime import datetime from datetime import datetime
import httpx
from openai import APIConnectionError, APIError, AuthenticationError, OpenAI from openai import APIConnectionError, APIError, AuthenticationError, OpenAI
@@ -16,7 +17,7 @@ class ChatGPT():
self.model = conf.get("model", "gpt-3.5-turbo") self.model = conf.get("model", "gpt-3.5-turbo")
self.LOG = logging.getLogger("ChatGPT") self.LOG = logging.getLogger("ChatGPT")
if proxy: if proxy:
self.client = OpenAI(api_key=key, base_url=api, proxy={"http": proxy, "https": proxy}) self.client = OpenAI(api_key=key, base_url=api, http_client=httpx.Client(proxy=proxy))
else: else:
self.client = OpenAI(api_key=key, base_url=api) self.client = OpenAI(api_key=key, base_url=api)
self.conversation_list = {} self.conversation_list = {}