mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-28 10:03:54 +08:00
fix:部分支持openai的LLM的url配置名不同
This commit is contained in:
@@ -7,7 +7,10 @@ class LLMProvider(LLMProviderBase):
|
|||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
self.model_name = config.get("model_name")
|
self.model_name = config.get("model_name")
|
||||||
self.api_key = config.get("api_key")
|
self.api_key = config.get("api_key")
|
||||||
self.url = config.get("url")
|
if 'base_url' in config:
|
||||||
|
self.base_url = config.get("base_url")
|
||||||
|
else:
|
||||||
|
self.base_url = config.get("url")
|
||||||
self.top_p = config.get("top_p")
|
self.top_p = config.get("top_p")
|
||||||
self.top_k = config.get("top_k")
|
self.top_k = config.get("top_k")
|
||||||
self.temperature = config.get("temperature")
|
self.temperature = config.get("temperature")
|
||||||
@@ -15,7 +18,7 @@ class LLMProvider(LLMProviderBase):
|
|||||||
self.frequency_penalty = config.get("frequency_penalty")
|
self.frequency_penalty = config.get("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.url)
|
self.client = openai.OpenAI(api_key=self.api_key, base_url=self.base_url)
|
||||||
|
|
||||||
def response(self, session_id, dialogue):
|
def response(self, session_id, dialogue):
|
||||||
try:
|
try:
|
||||||
@@ -71,4 +74,4 @@ class LLMProvider(LLMProviderBase):
|
|||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.logger.bind(tag=TAG).error(f"Error in function call streaming: {e}")
|
self.logger.bind(tag=TAG).error(f"Error in function call streaming: {e}")
|
||||||
yield {"type": "content", "content": f"【OpenAI服务响应异常: {e}】"}
|
yield {"type": "content", "content": f"【OpenAI服务响应异常: {e}】"}
|
||||||
|
|||||||
Reference in New Issue
Block a user