From 60a0f2f039dbbfe5e28d2909617958df2187264d Mon Sep 17 00:00:00 2001 From: taylor Date: Sun, 13 Oct 2024 18:02:15 +0800 Subject: [PATCH] =?UTF-8?q?add:=20=E6=B7=BB=E5=8A=A0ollama=E8=B0=83?= =?UTF-8?q?=E7=94=A8AI=20chat=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- maubot_llmplus/local_paltform.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/maubot_llmplus/local_paltform.py b/maubot_llmplus/local_paltform.py index 67ac655..d56adb0 100644 --- a/maubot_llmplus/local_paltform.py +++ b/maubot_llmplus/local_paltform.py @@ -24,19 +24,20 @@ class Ollama(Platform): endpoint = f"{self.url}/api/chat" req_body = {'model': self.model, 'messages': full_context, 'steam': False} - headers = {} + headers = {'Content-Type': 'application/json'} if self.api_key is not None: headers['Authorization'] = self.api_key - headers['Content-Type'] = 'application/json' plugin.log.debug(f"{json.dumps(req_body)}") async with self.http.post(endpoint, headers=headers, data=json.dumps(req_body)) as response: - plugin.log.debug(f"响应内容:{response.status}, {response.json()}") + plugin.log.debug(f"响应内容:{response.status}, {await response.json()}") if response.status != 200: return ChatCompletion( message={}, finish_reason=f"http status {response.status}", model=None ) + text = await response.text() + plugin.log.debug(f"解析后的响应内容: {text}") response_json = await response.json() return ChatCompletion( message=response_json['message'],