From 3d97fab16d94a3374eac3c64b0baa888f9102b5a Mon Sep 17 00:00:00 2001
From: HaoDuoYu <63943547+diaoling6665@users.noreply.github.com>
Date: Tue, 18 Mar 2025 12:05:33 +0800
Subject: [PATCH] Add files via upload

---
 .../core/providers/llm/openai/openai.py       | 25 ++++++++++++++-----
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/main/xiaozhi-server/core/providers/llm/openai/openai.py b/main/xiaozhi-server/core/providers/llm/openai/openai.py
index 95cd85f9..2feb4847 100644
--- a/main/xiaozhi-server/core/providers/llm/openai/openai.py
+++ b/main/xiaozhi-server/core/providers/llm/openai/openai.py
@@ -7,19 +7,27 @@ class LLMProvider(LLMProviderBase):
     def __init__(self, config):
         self.model_name = config.get("model_name")
         self.api_key = config.get("api_key")
-        if 'base_url' in config:
-            self.base_url = config.get("base_url")
-        else:
-            self.base_url = config.get("url")
+        self.url = config.get("url")
+        self.top_p = config.get("top_p")
+        self.top_k = config.get("top_k")
+        self.temperature = config.get("temperature")
+        self.max_tokens = config.get("max_tokens")
+        self.frequency_penalty = config.get("frequency_penalty")
+        
         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.url)
 
     def response(self, session_id, dialogue):
         try:
             responses = self.client.chat.completions.create(
                 model=self.model_name,
                 messages=dialogue,
-                stream=True
+                stream=True,
+                temperature=self.temperature,
+                max_tokens=self.max_tokens,
+                top_p=self.top_p,
+                top_k=self.top_k,
+                frequency_penalty=self.frequency_penalty
             )
             
             is_active = True
@@ -51,6 +59,11 @@ class LLMProvider(LLMProviderBase):
                 messages=dialogue,
                 stream=True,
                 tools=functions,
+                temperature=self.temperature,
+                max_tokens=self.max_tokens,
+                top_p=self.top_p,
+                top_k=self.top_k,
+                frequency_penalty=self.frequency_penalty
             )
             
             for chunk in stream: