fix deepseek

This commit is contained in:
Chill
2025-02-10 16:01:06 +08:00
parent 6f126d9abe
commit 1a12e1d6a5
4 changed files with 32 additions and 8 deletions

View File

@@ -69,3 +69,27 @@ class AiRobot(models.Model):
return content, usage, True
else:
return super().get_ai_post(res, author_id, answer_id, param)
def get_ai_list_model(self):
self.ensure_one()
if self.provider == 'deepseek':
api_key = self.openapi_api_key
if not api_key:
raise UserError(_("Please provide Ai Robot [%s] API Key." % self.name))
payload = {}
headers = {
'Accept': 'application/json',
'Authorization': f'Bearer {self.openapi_api_key}'
}
R_TIMEOUT = self.ai_timeout or 120
o_url = "https://api.deepseek.com/models"
response = requests.get(o_url, headers=headers, data=payload, timeout=R_TIMEOUT)
response.close()
if response:
res = response.json()
r_text = json.dumps(res, indent=2)
else:
r_text = 'No response.'
raise UserError(r_text)
else:
return super().get_ai_list_model()