diff --git a/app_deepseek/models/ai_robot.py b/app_deepseek/models/ai_robot.py index 3e42ff32..32c6b740 100644 --- a/app_deepseek/models/ai_robot.py +++ b/app_deepseek/models/ai_robot.py @@ -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() diff --git a/app_deepseek/views/ai_robot_views.xml b/app_deepseek/views/ai_robot_views.xml index 4d2f2abc..e8ddd481 100644 --- a/app_deepseek/views/ai_robot_views.xml +++ b/app_deepseek/views/ai_robot_views.xml @@ -1,13 +1,13 @@ - +