From 84cc7d284b6e9efd265edea0cc8000ea4fd6bb43 Mon Sep 17 00:00:00 2001
From: Chill <chill421@qq.com>
Date: Mon, 10 Feb 2025 15:59:02 +0800
Subject: [PATCH] fix deepseek

---
 app_deepseek/models/ai_robot.py       | 24 ++++++++++++++++++++++++
 app_deepseek/views/ai_robot_views.xml | 10 +++++-----
 2 files changed, 29 insertions(+), 5 deletions(-)

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 @@
 <?xml version="1.0" encoding="UTF-8" ?>
 <odoo>
-    <!-- <record id="baidu_ai_robot_form_view" model="ir.ui.view">
-        <field name="name">baidu.ai.robot.form</field>
+    <record id="app_ai_robot_deepseek_form_view" model="ir.ui.view">
+        <field name="name">app.ai.robot.deepseek.form</field>
         <field name="model">ai.robot</field>
         <field name="inherit_id" ref="app_chatgpt.ai_robot_form_view"/>
         <field name="arch" type="xml">
-            <xpath expr="//field[@name='openapi_api_key']" position="after">
-                <field name="client_secret" password="True"/>
+            <xpath expr="//button[@name='get_ai_list_model']" position="attributes">
+                <attribute name="invisible">False</attribute>
             </xpath>
         </field>
-    </record> -->
+    </record>
 </odoo>