正确保存模型名称到Database

2025-03-20 16:47:50 +08:00
parent 8c178b80d7
commit e9bd3196ba
2 changed files with 4 additions and 3 deletions
--- a/src/plugins/models/utils_model.py
+++ b/src/plugins/models/utils_model.py
@@ -526,7 +526,7 @@ class LLM_request:
        """根据输入的提示生成模型的异步响应"""

        content, reasoning_content = await self._execute_request(endpoint="/chat/completions", prompt=prompt)
-        return content, reasoning_content
+        return content, reasoning_content, self.model_name

    async def generate_response_for_image(self, prompt: str, image_base64: str, image_format: str) -> Tuple[str, str]:
        """根据输入的提示和图片生成模型的异步响应"""