audrius/app-odoo
1
0
Fork 0
mirror of https://github.com/guohuadeng/app-odoo.git synced 2025-02-23 04:11:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

add timeout

Browse Source
This commit is contained in:
ivan deng
2023-04-13 01:15:22 +08:00
parent fbc2453f8f
commit d344bd6499
4 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app_chatgpt/data/ai_robot_data.xml
View File

@@ -1,23 +1,35 @@
<?xml version="1.0" encoding="UTF-8" ?>
<odoo>
<odoo noupdate="1">
<record id="chatgpt_robot" model="ai.robot">
<field name="name">ChatGPT odoo</field>
<field name="provider">openai</field>
<field name="endpoint">https://api.openai.com/v1/chat/completions</field>
<field name="sequence">1</field>
</record>
<record id="chatgpt_robot1" model="ai.robot">
<field name="name">ChatGPT Coding</field>
<field name="provider">openai</field>
<field name="endpoint">https://api.openai.com/v1/chat/completions</field>
<field name="sequence">6</field>
</record>
<record id="chatgpt_robot2" model="ai.robot">
<field name="name">ChatGPT Finance</field>
<field name="provider">openai</field>
<field name="endpoint">https://api.openai.com/v1/chat/completions</field>
<field name="sequence">7</field>
</record>
<record id="chatgpt_robot3" model="ai.robot">
<field name="name">ChatGPT Azure</field>
<field name="provider">azure</field>
<field name="endpoint">https://my.openai.azure.com</field>
<field name="engine">gpt35</field>
<field name="sequence">8</field>
</record>
<record id="chatgpt4_azure" model="ai.robot">
<field name="name">ChatGPT4 Azure</field>
<field name="provider">azure</field>
<field name="endpoint">https://my.openai.azure.com</field>
<field name="engine">gpt4</field>
<field name="sequence">9</field>
</record>
</odoo>

15
app_chatgpt/data/user_partner_data.xml
View File

@@ -62,5 +62,20 @@
<field name="company_ids" eval="[Command.link(ref('base.main_company'))]"/>
<field name="groups_id" eval="[Command.link(ref('base.group_user'))]"/>
</record>
<record id="partner_chatgpt4_azure" model="res.partner">
<field name="name">ChatGPT4 Azure</field>
<field name="image_1920" type="base64" file="app_chatgpt/static/description/chatgpt.png"/>
</record>
<record id="user_chatgpt4_azure" model="res.users">
<field name="login">chatgpt4@sunpop.cn</field>
<field name="email">chatgpt4@sunpop.cn</field>
<field name="password">chatgpt</field>
<field name="partner_id" ref="app_chatgpt.chatgpt4_azure"/>
<field name="gpt_id" ref="app_chatgpt.chatgpt4_azure"/>
<field name="company_id" ref="base.main_company"/>
<field name="company_ids" eval="[Command.link(ref('base.main_company'))]"/>
<field name="groups_id" eval="[Command.link(ref('base.group_user'))]"/>
</record>
</data>
</odoo>

7
app_chatgpt/models/ai_robot.py
View File

@@ -42,6 +42,7 @@ GPT-3 A set of models that can understand and generate natural language
endpoint = fields.Char('End Point', default='https://api.openai.com/v1/chat/completions')
engine = fields.Char('Engine', help='If use Azure, Please input the Model deployment name.')
api_version = fields.Char('API Version', default='2022-12-01')
ai_timeout = fields.Integer('Timeout(seconds)', help="Connect timeout for Ai response", default=120)
sequence = fields.Integer('Sequence', help="Determine the display order", default=10)
sensitive_words = fields.Text('Sensitive Words Plus', help='Sensitive word filtering. Separate keywords with a carriage return.')
is_filtering = fields.Boolean('Filter Sensitive Words', default=False, help='Use base Filter in dir models/lib/sensi_words.txt')
@@ -77,7 +78,7 @@ GPT-3 A set of models that can understand and generate natural language
def get_ai_model_info(self):
self.ensure_one()
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
R_TIMEOUT = 300
R_TIMEOUT = self.ai_timeout or 120
o_url = "https://api.openai.com/v1/models/%s" % self.ai_model
if self.endpoint:
o_url = self.endpoint.replace("/chat/completions", "") + "/models/%s" % self.ai_model
@@ -94,7 +95,7 @@ GPT-3 A set of models that can understand and generate natural language
def get_ai_list_model(self):
self.ensure_one()
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
R_TIMEOUT = 300
R_TIMEOUT = self.ai_timeout or 120
o_url = "https://api.openai.com/v1/models"
if self.endpoint:
o_url = self.endpoint.replace("/chat/completions", "") + "/models"
@@ -110,7 +111,7 @@ GPT-3 A set of models that can understand and generate natural language
def get_openai(self, data, partner_name='odoo', *args):
self.ensure_one()
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
R_TIMEOUT = 300
R_TIMEOUT = self.ai_timeout or 120
o_url = self.endpoint or "https://api.openai.com/v1/chat/completions"
# 以下处理 open ai

1
app_chatgpt/views/ai_robot_views.xml
View File

@@ -46,6 +46,7 @@
<field name="engine"/>
<field name="endpoint"/>
<field name="api_version"/>
<field name="ai_timeout"/>
</group>
<group>
<field name="is_filtering"/>