add deepseek

app_deepseek/models/ai_robot.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+
+import base64
+import os
+import requests, json
+import logging
+from openai import OpenAI
+from http import HTTPStatus
+from odoo import api, fields, models, modules, tools, _
+from odoo.exceptions import UserError
+
+
+_logger = logging.getLogger(__name__)
+
+
+class AiRobot(models.Model):
+    _inherit = 'ai.robot'
+
+    provider = fields.Selection(selection_add=[('deepseek', 'Deepseek')], ondelete={'deepseek': 'set default'})
+    set_ai_model = fields.Selection(
+        selection_add=[
+            ('deepseek-chat', 'Deepseek Chat'),
+        ],
+        ondelete={'deepseek-chat': 'set default'})
+
+    @api.onchange('provider')
+    def _onchange_provider(self):
+        if self.provider == 'deepseek':
+            if self.ai_model == 'deepseek-chat':
+                self.endpoint = 'https://api.deepseek.com'
+            # 取头像
+            module_path = modules.get_module_path('app_ai', display_warning=False)
+            if module_path:
+                path = modules.check_resource_path(module_path, ('static/description/src/%s.png' % self.provider))
+                if path:
+                    image_file = tools.file_open(path, 'rb')
+                    self.image_avatar = base64.b64encode(image_file.read())
+        return super()._onchange_provider()
+
+    def get_deepseek(self, data, author_id, answer_id, param={}):
+        self.ensure_one()
+        api_key = self.openapi_api_key
+        if not api_key or not self.endpoint:
+            raise UserError(_("Please provide Ai Robot [%s] API Key and Endpoint URL first." % self.name))
+
+        if isinstance(data, list):
+            messages = data
+        else:
+            messages = [{"role": "user", "content": data}]
+
+        url = self.endpoint
+
+        client = OpenAI(api_key=api_key, base_url=self.endpoint)
+        response = client.chat.completions.create(
+            model=self.ai_model,
+            messages=messages
+        )
+        res = response.model_dump()
+        if 'choices' in res:
+            return res
+        else:
+            _logger.warning('=====================deepseek output data: %s' % response.json())
+
+        return _("Response Timeout, please speak again.")
+
+    def get_ai_post(self, res, author_id=False, answer_id=False, param={}):
+        if self.provider == 'deepseek':
+            usage = res['usage']
+            content = res['choices'][0]['message']['content']
+            return content, usage, True
+        else:
+            return super().get_ai_post(res, author_id, answer_id, param)