add: 添加ollama调用AI chat逻辑

This commit is contained in:
taylor
2024-10-13 18:31:18 +08:00
parent 78b44a08fc
commit 7e1af58c84
2 changed files with 4 additions and 11 deletions

View File

@@ -25,9 +25,6 @@ class Ollama(Platform):
endpoint = f"{self.url}/api/chat"
req_body = {'model': self.model, 'messages': full_context, 'stream': False}
headers = {'Content-Type': 'application/json'}
if self.api_key is not None:
headers['Authorization'] = self.api_key
plugin.log.debug(f"{json.dumps(req_body)}")
async with self.http.post(endpoint, headers=headers, json=req_body) as response:
# plugin.log.debug(f"响应内容:{response.status}, {await response.json()}")
if response.status != 200:
@@ -36,8 +33,6 @@ class Ollama(Platform):
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'],