add: 添加ollama调用AI chat逻辑

This commit is contained in:
taylor
2024-10-13 17:07:42 +08:00
parent bab547198f
commit 5821623aae
4 changed files with 19 additions and 16 deletions

View File

@@ -33,8 +33,9 @@ class Platform:
additional_prompt: List[dict]
system_prompt: str
max_context_messages: int
name: str
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
def __init__(self, config: BaseProxyConfig, name: str, http: ClientSession) -> None:
self.http = http
self.config = config['platforms'][self.get_type()]
self.url = self.config['url']
@@ -44,6 +45,7 @@ class Platform:
self.max_context_messages = self.config['max_context_messages']
self.additional_prompt = config['additional_prompt']
self.system_prompt = config['system_prompt']
self.name = name
"""a
调用AI对话接口, 响应结果