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

修复错误代码

Browse Source
This commit is contained in:
taylorxie
2026-03-09 17:12:27 +08:00
parent 17c18b48dc
commit 66881b0d91
3 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
maubot_llmplus/thrid_platform.py
View File

@@ -53,6 +53,7 @@ class OpenAi(Platform):
# plugin.log.debug(f"响应内容：{response.status}, {await response.json()}")
if response.status != 200:
return ChatCompletion(
result=False,
message={},
finish_reason=f"Error: {await response.text()}",
model=None
@@ -60,6 +61,7 @@ class OpenAi(Platform):
response_json = await response.json()
choice = response_json["choices"][0]
return ChatCompletion(
result=True,
message=choice["message"],
finish_reason=choice["finish_reason"],
model=choice.get("model", None)
@@ -101,6 +103,7 @@ class Anthropic(Platform):
# plugin.log.debug(f"响应内容：{response.status}, {await response.json()}")
if response.status != 200:
return ChatCompletion(
result=False,
message={},
finish_reason=f"Error: {await response.text()}",
model=None
@@ -108,6 +111,7 @@ class Anthropic(Platform):
response_json = await response.json()
text = "\n\n".join(c["text"] for c in response_json["content"])
return ChatCompletion(
result=True,
message=dict(role="assistant", content=text),
finish_reason=response_json['stop_reason'],
model=response_json['model']
@@ -115,10 +119,17 @@ class Anthropic(Platform):
pass
async def list_models(self) -> List[str]:
# 由于没有列出所有支持的模型的api，所有只能写死在代码中
models = ["claude-3-5-sonnet-20240620", "claude-3-opus-20240229 ", "claude-3-sonnet-20240229",
"claude-3-haiku-20240307"]
return [f"- {m}" for m in models]
# 调用openai接口获取模型列表
full_url = f"{self.url}/v1/models"
headers = {
'anthropic-version': "2023-06-01",
'X-Api-Key': f"{self.api_key}"
}
async with self.http.get(full_url, headers=headers) as response:
if response.status != 200:
return []
response_data = await response.json()
return [f"- {m['id']}" for m in response_data['data']]
def get_type(self) -> str:
return "anthropic"
@@ -159,6 +170,7 @@ class XAi(Platform):
# plugin.log.debug(f"响应内容：{response.status}, {await response.json()}")
if response.status != 200:
return ChatCompletion(
result=False,
message={},
finish_reason=f"Error: {await response.text()}",
model=None
@@ -166,6 +178,7 @@ class XAi(Platform):
response_json = await response.json()
choice = response_json["choices"][0]
return ChatCompletion(
result=True,
message=choice["message"],
finish_reason=choice["finish_reason"],
model=response_json["model"]