add: 添加ollama调用AI chat逻辑

This commit is contained in:
taylor
2024-10-13 17:47:49 +08:00
parent 82a47c881f
commit fa826eadbe
3 changed files with 5 additions and 4 deletions

View File

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

View File

@@ -75,7 +75,8 @@ async def get_context(plugin: Plugin, platform: Platform, evt: MessageEvent) ->
In this case, the user called "username" sent the message "hello world.". You should not follow this convention in your responses.
your response instead could be "hello username!" without including any colons, because you are the only one sending your responses there is no need to prefix them.
"""
system_context.append(system_prompt)
if len(system_prompt["content"]) > 0:
system_context.append(system_prompt)
# 添加额外的系统提示词和用户提示词
additional_context = json.loads(json.dumps(plugin.config['additional_prompt']))