add: 添加ollama调用AI chat逻辑
This commit is contained in:
@@ -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'],
|
||||
|
||||
Reference in New Issue
Block a user