From 59d9a3d4eca2f7e4e156207e87fc6ac3e52f1d97 Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Tue, 19 Dec 2023 15:39:22 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20chatAI=E4=BB=A3=E7=90=86=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E9=94=99=E8=AF=AF=EF=BC=8C=E6=9C=8D=E5=8A=A1=E5=99=A8?= =?UTF-8?q?=E6=8A=A5=E9=94=99500?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/settings/api/chat.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/apps/settings/api/chat.py b/apps/settings/api/chat.py index 830ab45ea..68a80f013 100644 --- a/apps/settings/api/chat.py +++ b/apps/settings/api/chat.py @@ -47,16 +47,17 @@ class ChatAITestingAPI(GenericAPIView): 'base_url': config['GPT_BASE_URL'] or None, 'api_key': config['GPT_API_KEY'], } - if proxy: - kwargs['http_client'] = httpx.Client( - proxies=proxy, - transport=httpx.HTTPTransport(local_address='0.0.0.0') - ) - client = openai.OpenAI(**kwargs) - - ok = False - error = '' try: + if proxy: + kwargs['http_client'] = httpx.Client( + proxies=proxy, + transport=httpx.HTTPTransport(local_address='0.0.0.0') + ) + client = openai.OpenAI(**kwargs) + + ok = False + error = '' + client.chat.completions.create( messages=[ {