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 18:08:30 +08:00
parent 4e936b35b2
commit d6339be66d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
maubot_llmplus/local_paltform.py
View File

@@ -23,7 +23,7 @@ class Ollama(Platform):
full_context.extend(list(context)) full_context.extend(list(context))
endpoint = f"{self.url}/api/chat" endpoint = f"{self.url}/api/chat"
req_body = {'model': self.model, 'messages': full_context, 'steam': False} req_body = {'model': self.model, 'messages': full_context, 'stream': False}
headers = {'Content-Type': 'application/json'} headers = {'Content-Type': 'application/json'}
if self.api_key is not None: if self.api_key is not None:
headers['Authorization'] = self.api_key headers['Authorization'] = self.api_key