Hass/xiaozhi-esp32-server
1
0
Fork 0
mirror of https://github.com/xinnan-tech/xiaozhi-esp32-server.git synced 2026-07-27 01:23:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加timeout配置项，单位为秒

Browse Source
This commit is contained in:
Liam Mazy
2025-06-11 14:05:54 +08:00
parent 0e57aad12d
commit 0f31bfc78f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/xiaozhi-server/core/providers/llm/openai/openai.py
+4 -1
View File
@@ -1,3 +1,4 @@
import httpx
import openai import openai
from openai.types import CompletionUsage from openai.types import CompletionUsage
from config.logger import setup_logging from config.logger import setup_logging
@@ -16,6 +17,8 @@ class LLMProvider(LLMProviderBase):
self.base_url = config.get("base_url") self.base_url = config.get("base_url")
else: else:
self.base_url = config.get("url") self.base_url = config.get("url")
# 增加timeout的配置项，单位为秒
self.timeout = config.get("timeout", 300.0)
param_defaults = { param_defaults = {
"max_tokens": (500, int), "max_tokens": (500, int),
@@ -35,7 +38,7 @@ class LLMProvider(LLMProviderBase):
f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.frequency_penalty}") f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.frequency_penalty}")
check_model_key("LLM", self.api_key) check_model_key("LLM", self.api_key)
self.client = openai.OpenAI(api_key=self.api_key, base_url=self.base_url) self.client = openai.OpenAI(api_key=self.api_key, base_url=self.base_url, timeout=httpx.Timeout(self.timeout))
def response(self, session_id, dialogue, **kwargs): def response(self, session_id, dialogue, **kwargs):
try: try: