You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.6 KiB
96 lines
3.6 KiB
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2025-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Cybrosys Techno Solutions(<https://www.cybrosys.com>)
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU AFFERO GENERAL PUBLIC LICENSE (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
from odoo import fields, models, _
|
|
import requests
|
|
|
|
|
|
def notification_response(title, notification_type, message):
|
|
"""Method notification_response returns the notification data according
|
|
to each conditions while checking the connection to ChatGPT."""
|
|
notification_data = {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'display_notification',
|
|
'params': {
|
|
'title': _(title),
|
|
'type': notification_type,
|
|
'message': message,
|
|
'sticky': False,
|
|
}
|
|
}
|
|
return notification_data
|
|
|
|
|
|
class ResConfigSettings(models.TransientModel):
|
|
"""This class extends the base 'res.config.settings' model to include
|
|
additional fields for configuring ChatGPT integration settings."""
|
|
_inherit = "res.config.settings"
|
|
|
|
api_key = fields.Char(string="API Key", help="Provide the API key here",
|
|
config_parameter="odoo_chatgpt_connector.api_key")
|
|
|
|
def action_test_api(self):
|
|
"""Method action_test_api to test the connection from Odoo to ChatGPT"""
|
|
url = "https://api.openai.com/v1/chat/completions"
|
|
payload = {
|
|
'model': 'gpt-3.5-turbo',
|
|
'messages': [
|
|
{"role": "user", "content": ""}
|
|
]
|
|
}
|
|
api_key = self.env['ir.config_parameter'].sudo().get_param(
|
|
'odoo_chatgpt_connector.api_key')
|
|
if not api_key:
|
|
notification = notification_response(
|
|
title="No API key!",
|
|
notification_type="danger",
|
|
message="Please provide an API Key"
|
|
)
|
|
return notification
|
|
|
|
headers = {
|
|
'Authorization': f"Bearer {api_key}",
|
|
'Content-Type': "application/json"
|
|
}
|
|
try:
|
|
response = requests.post(url, headers=headers, json=payload, timeout=10)
|
|
response.raise_for_status() # Raise error for bad HTTP status
|
|
notification = notification_response(
|
|
title="Connection Successful!",
|
|
notification_type="success",
|
|
message="Successfully connected to ChatGPT"
|
|
)
|
|
return notification
|
|
|
|
except requests.exceptions.Timeout:
|
|
return notification_response(
|
|
title="Connection Timeout!",
|
|
notification_type="danger",
|
|
message="The request to ChatGPT timed out"
|
|
)
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
return notification_response(
|
|
title="Connection Error!",
|
|
notification_type="danger",
|
|
message="Error connecting to ChatGPT: API Key may be invalid"
|
|
)
|
|
|
|
|