add openai error logs

This commit is contained in:
harry
2024-04-11 18:25:15 +08:00
parent 56a6d5c523
commit 759c1ce652

View File

@@ -5,6 +5,8 @@ from typing import List
from loguru import logger from loguru import logger
from openai import OpenAI from openai import OpenAI
from openai import AzureOpenAI from openai import AzureOpenAI
from openai.types.chat import ChatCompletion
from app.config import config from app.config import config
@@ -133,7 +135,15 @@ def _generate_response(prompt: str) -> str:
messages=[{"role": "user", "content": prompt}] messages=[{"role": "user", "content": prompt}]
) )
if response: if response:
content = response.choices[0].message.content if isinstance(response, ChatCompletion):
content = response.choices[0].message.content
else:
raise Exception(
f"[{llm_provider}] returned an invalid response: \"{response}\", please check your network "
f"connection and try again.")
else:
raise Exception(
f"[{llm_provider}] returned an empty response, please check your network connection and try again.")
return content.replace("\n", "") return content.replace("\n", "")