diff --git a/app/service/chat/gemini_chat_service.py b/app/service/chat/gemini_chat_service.py index 8bec12b..61955e9 100644 --- a/app/service/chat/gemini_chat_service.py +++ b/app/service/chat/gemini_chat_service.py @@ -112,7 +112,7 @@ class GeminiChatService: """聊天服务""" def __init__(self, base_url: str, key_manager: KeyManager): - self.api_client = GeminiApiClient(base_url) + self.api_client = GeminiApiClient(base_url, settings.TIME_OUT) self.key_manager = key_manager self.response_handler = GeminiResponseHandler() diff --git a/app/service/chat/openai_chat_service.py b/app/service/chat/openai_chat_service.py index 4f3cf95..be87f59 100644 --- a/app/service/chat/openai_chat_service.py +++ b/app/service/chat/openai_chat_service.py @@ -146,7 +146,7 @@ class OpenAIChatService: def __init__(self, base_url: str, key_manager: KeyManager = None): self.message_converter = OpenAIMessageConverter() self.response_handler = OpenAIResponseHandler(config=None) - self.api_client = GeminiApiClient(base_url) + self.api_client = GeminiApiClient(base_url, settings.TIME_OUT) self.key_manager = key_manager self.image_create_service = ImageCreateService()