code-repository/maubot-llmplus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add: 添加ollama调用AI chat逻辑

Browse Source
This commit is contained in:
taylor
2024-10-13 17:51:31 +08:00
parent fa826eadbe
commit e4ba651dfd
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
maubot_llmplus/local_paltform.py
View File

@@ -27,6 +27,7 @@ class Ollama(Platform):
headers = {} headers = {}
if self.api_key is not None: if self.api_key is not None:
headers['Authorization'] = self.api_key headers['Authorization'] = self.api_key
headers['Content-Type'] = 'application/json'
plugin.log.debug(f"{json.dumps(req_body)}") plugin.log.debug(f"{json.dumps(req_body)}")
async with self.http.post(endpoint, headers=headers, data=json.dumps(req_body)) as response: 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}, {response.json()}")
@@ -36,11 +37,11 @@ class Ollama(Platform):
finish_reason=f"http status {response.status}", finish_reason=f"http status {response.status}",
model=None model=None
) )
response_json = await response.json() response_json = response.json()
return ChatCompletion( return ChatCompletion(
message=response_json['message'], message=response_json['message'],
finish_reason='success', finish_reason='success',
model=response_json.get('model', None) model=response_json['model']
) )
def get_type(self) -> str: def get_type(self) -> str: