diff --git a/ai_automation/models/ai_connection.py b/ai_automation/models/ai_connection.py index 0419c58d..e46b72d2 100644 --- a/ai_automation/models/ai_connection.py +++ b/ai_automation/models/ai_connection.py @@ -18,10 +18,14 @@ class AiConnection(models.Model): url = fields.Char(groups="base.group_system") model = fields.Char(groups="base.group_system") - def _run(self, prompt, tools=None, record=None): - return getattr(self, f"_run_{self.kind}")(prompt, tools=tools, record=record) + def _run(self, prompt, tools=None, record=None, system_prompt=None): + return getattr(self, f"_run_{self.kind}")( + prompt, tools=tools, record=record, system_prompt=system_prompt + ) - def _run_ollama(self, prompt, tools=None, messages=None, record=None): + def _run_ollama( + self, prompt, tools=None, messages=None, record=None, system_prompt=None + ): tool_definition = [] for tool in tools or []: definition = tool._get_tool_definition() @@ -40,6 +44,8 @@ def _run_ollama(self, prompt, tools=None, messages=None, record=None): ollama_client = ollama.Client(**self._get_ollama_client_parameters()) if messages is None: messages = [] + if system_prompt: + messages.append({"role": "system", "content": system_prompt}) messages.append({"role": "user", "content": prompt}) while True: response = ollama_client.chat( diff --git a/ai_automation_openai/README.rst b/ai_automation_openai/README.rst new file mode 100644 index 00000000..8f169905 --- /dev/null +++ b/ai_automation_openai/README.rst @@ -0,0 +1,86 @@ +.. image:: https://odoo-community.org/readme-banner-image + :target: https://odoo-community.org/get-involved?utm_source=readme + :alt: Odoo Community Association + +==================== +Ai Automation Openai +==================== + +.. + !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! + !! This file is generated by oca-gen-addon-readme !! + !! changes will be overwritten. !! + !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! + !! source digest: sha256:303903fc80f9498717d9c21c5644906b55096f06290677850780afa57400cefd + !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! + +.. |badge1| image:: https://img.shields.io/badge/maturity-Beta-yellow.png + :target: https://odoo-community.org/page/development-status + :alt: Beta +.. |badge2| image:: https://img.shields.io/badge/license-AGPL--3-blue.png + :target: http://www.gnu.org/licenses/agpl-3.0-standalone.html + :alt: License: AGPL-3 +.. |badge3| image:: https://img.shields.io/badge/github-OCA%2Fai-lightgray.png?logo=github + :target: https://github.com/OCA/ai/tree/16.0/ai_automation_openai + :alt: OCA/ai +.. |badge4| image:: https://img.shields.io/badge/weblate-Translate%20me-F47D42.png + :target: https://translation.odoo-community.org/projects/ai-16-0/ai-16-0-ai_automation_openai + :alt: Translate me on Weblate +.. |badge5| image:: https://img.shields.io/badge/runboat-Try%20me-875A7B.png + :target: https://runboat.odoo-community.org/builds?repo=OCA/ai&target_branch=16.0 + :alt: Try me on Runboat + +|badge1| |badge2| |badge3| |badge4| |badge5| + +This module adds the option to link using OpenAI API (by using Openai +Library). + +It could be used to link with OpenAI directly or other systems that use +the same API, like LiteLLM. + +**Table of contents** + +.. contents:: + :local: + +Bug Tracker +=========== + +Bugs are tracked on `GitHub Issues `_. +In case of trouble, please check there if your issue has already been reported. +If you spotted it first, help us to smash it by providing a detailed and welcomed +`feedback `_. + +Do not contact contributors directly about support or help with technical issues. + +Credits +======= + +Authors +------- + +* Dixmit + +Contributors +------------ + +- `Dixmit `__ + + - Enric Tobella + +Maintainers +----------- + +This module is maintained by the OCA. + +.. image:: https://odoo-community.org/logo.png + :alt: Odoo Community Association + :target: https://odoo-community.org + +OCA, or the Odoo Community Association, is a nonprofit organization whose +mission is to support the collaborative development of Odoo features and +promote its widespread use. + +This module is part of the `OCA/ai `_ project on GitHub. + +You are welcome to contribute. To learn how please visit https://odoo-community.org/page/Contribute. diff --git a/ai_automation_openai/__init__.py b/ai_automation_openai/__init__.py new file mode 100644 index 00000000..0650744f --- /dev/null +++ b/ai_automation_openai/__init__.py @@ -0,0 +1 @@ +from . import models diff --git a/ai_automation_openai/__manifest__.py b/ai_automation_openai/__manifest__.py new file mode 100644 index 00000000..bf43d9cb --- /dev/null +++ b/ai_automation_openai/__manifest__.py @@ -0,0 +1,23 @@ +# Copyright 2026 Dixmit +# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). + +{ + "name": "Ai Automation Openai", + "summary": """Integrate automation with OpenAI/LiteVLLM Api""", + "version": "16.0.1.0.0", + "license": "AGPL-3", + "author": "Dixmit,Odoo Community Association (OCA)", + "website": "https://github.com/OCA/ai", + "depends": [ + "ai_automation", + ], + "data": [ + "views/ai_connection.xml", + ], + "demo": [], + "external_dependencies": { + "python": [ + "openai", + ], + }, +} diff --git a/ai_automation_openai/models/__init__.py b/ai_automation_openai/models/__init__.py new file mode 100644 index 00000000..9f35a7aa --- /dev/null +++ b/ai_automation_openai/models/__init__.py @@ -0,0 +1 @@ +from . import ai_connection diff --git a/ai_automation_openai/models/ai_connection.py b/ai_automation_openai/models/ai_connection.py new file mode 100644 index 00000000..3e78e892 --- /dev/null +++ b/ai_automation_openai/models/ai_connection.py @@ -0,0 +1,79 @@ +# Copyright 2026 Dixmit +# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). + +import json + +import openai + +from odoo import fields, models + + +class AiConnection(models.Model): + _inherit = "ai.connection" + + kind = fields.Selection( + selection_add=[("openai", "OpenAI")], ondelete={"openai": "cascade"} + ) + openai_api_key = fields.Char(groups="base.group_system") + temperature = fields.Float(default=0, groups="base.group_system") + + def _run_openai( + self, prompt, tools=None, messages=None, record=None, system_prompt=None + ): + tool_definition = [] + for tool in tools or []: + definition = tool._get_tool_definition() + input_schema = definition["inputSchema"] + input_schema["additionalProperties"] = False + tool_definition.append( + { + "type": "function", + "function": { + "name": definition["name"], + "description": definition["description"], + "parameters": input_schema, + }, + } + ) + openai_client = openai.OpenAI(**self.sudo()._get_openai_client_parameters()) + if messages is None: + messages = [] + if system_prompt: + messages.append({"role": "system", "content": system_prompt}) + messages.append({"role": "user", "content": prompt}) + while True: + response = openai_client.chat.completions.create( + model=self.model, + messages=messages, + temperature=self.sudo().temperature, + tools=tool_definition or None, + ) + response_message = response.choices[0].message + tool_calls = response_message.tool_calls + if not tool_calls: + return response_message.content + messages.append(response_message) + for call in tool_calls: + tool = tools.filtered(lambda t, call=call: t.name == call.function.name) + if not tool: + continue + tool_output = tool._execute_tool( + record=record, **json.loads(call.function.arguments) + ) + if isinstance(tool_output, dict): + tool_output = json.dumps(tool_output) + messages.append( + { + "role": "tool", + "tool_call_id": call.id, + "content": tool_output, + } + ) + + def _get_openai_client_parameters(self): + params = {} + if self.url: + params["base_url"] = self.url + if self.openai_api_key: + params["api_key"] = self.openai_api_key + return params diff --git a/ai_automation_openai/readme/CONTRIBUTORS.md b/ai_automation_openai/readme/CONTRIBUTORS.md new file mode 100644 index 00000000..2c066ba7 --- /dev/null +++ b/ai_automation_openai/readme/CONTRIBUTORS.md @@ -0,0 +1,2 @@ +- [Dixmit](https://www.dixmit.com) + - Enric Tobella diff --git a/ai_automation_openai/readme/DESCRIPTION.md b/ai_automation_openai/readme/DESCRIPTION.md new file mode 100644 index 00000000..1c46c6d8 --- /dev/null +++ b/ai_automation_openai/readme/DESCRIPTION.md @@ -0,0 +1,3 @@ +This module adds the option to link using OpenAI API (by using Openai Library). + +It could be used to link with OpenAI directly or other systems that use the same API, like LiteLLM. \ No newline at end of file diff --git a/ai_automation_openai/static/description/icon.png b/ai_automation_openai/static/description/icon.png new file mode 100644 index 00000000..3a0328b5 Binary files /dev/null and b/ai_automation_openai/static/description/icon.png differ diff --git a/ai_automation_openai/static/description/index.html b/ai_automation_openai/static/description/index.html new file mode 100644 index 00000000..afb59256 --- /dev/null +++ b/ai_automation_openai/static/description/index.html @@ -0,0 +1,435 @@ + + + + + +README.rst + + + +
+ + + +Odoo Community Association + +
+

Ai Automation Openai

+ +

Beta License: AGPL-3 OCA/ai Translate me on Weblate Try me on Runboat

+

This module adds the option to link using OpenAI API (by using Openai +Library).

+

It could be used to link with OpenAI directly or other systems that use +the same API, like LiteLLM.

+

Table of contents

+ +
+

Bug Tracker

+

Bugs are tracked on GitHub Issues. +In case of trouble, please check there if your issue has already been reported. +If you spotted it first, help us to smash it by providing a detailed and welcomed +feedback.

+

Do not contact contributors directly about support or help with technical issues.

+
+
+

Credits

+
+

Authors

+
    +
  • Dixmit
    • +
+
+
+

Contributors

+ +
+
+

Maintainers

+

This module is maintained by the OCA.

+ +Odoo Community Association + +

OCA, or the Odoo Community Association, is a nonprofit organization whose +mission is to support the collaborative development of Odoo features and +promote its widespread use.

+

This module is part of the OCA/ai project on GitHub.

+

You are welcome to contribute. To learn how please visit https://odoo-community.org/page/Contribute.

+
+
+
+
+ + diff --git a/ai_automation_openai/tests/__init__.py b/ai_automation_openai/tests/__init__.py new file mode 100644 index 00000000..41213cb0 --- /dev/null +++ b/ai_automation_openai/tests/__init__.py @@ -0,0 +1 @@ +from . import test_openai diff --git a/ai_automation_openai/tests/test_openai.py b/ai_automation_openai/tests/test_openai.py new file mode 100644 index 00000000..0b4e1409 --- /dev/null +++ b/ai_automation_openai/tests/test_openai.py @@ -0,0 +1,105 @@ +# Copyright 2026 Dixmit +# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). + +from unittest import mock + +from openai.types.chat import ChatCompletion, ChatCompletionMessage +from openai.types.chat.chat_completion import Choice +from openai.types.chat.chat_completion_message_tool_call import ( + ChatCompletionMessageToolCall, + Function, +) + +from odoo.tests.common import TransactionCase + + +class OpenAIClient: + def __init__(self, chat_messages=None): + self.chat_messages = chat_messages or [] + self.current_message = -1 + self.calls = [] + + @property + def chat(self): + return self + + @property + def completions(self): + return self + + def create(self, *args, **kwargs): # pylint: disable=W8106 + self.current_message += 1 + self.calls.append((args, kwargs)) + return ChatCompletion( + id="chatcmpl-123", + object="chat.completion", + created=1677652288, + model="gpt-3.5-turbo-0301", + choices=[ + Choice( + index=0, + logprobs=None, + finish_reason="stop", + message=ChatCompletionMessage( + id=self.current_message, + **self.chat_messages[self.current_message] + ), + ) + ], + ) + + +class TestOpenai(TransactionCase): + @classmethod + def setUpClass(cls): + super().setUpClass() + cls.connection = cls.env["ai.connection"].create( + { + "name": "My Connection", + "url": "http://my_service:11434", + "model": "my_openai_model", + "kind": "openai", + } + ) + cls.action = cls.env["ir.actions.server"].create( + { + "name": "Test Action", + "model_id": cls.env.ref("base.model_res_partner").id, + "state": "ai_oca", + "ai_connection_id": cls.connection.id, + "ai_tool_ids": [(4, cls.env.ref("ai_tool.current_date").id)], + "ai_prompt": "What is the current date?", + } + ) + cls.partner = cls.env["res.partner"].create({"name": "Test Partner"}) + + def test_execute_action_with_tools(self): + client = OpenAIClient( + [ + { + "role": "assistant", + "content": "", + "tool_calls": [ + ChatCompletionMessageToolCall( + id="1", + type="function", + function=Function( + name="get_date", + arguments="{}", # must be a JSON string, not a dict + ), + ) + ], + }, + { + "role": "assistant", + "content": "Thanks it is 2024-01-01", + }, + ] + ) + with mock.patch("openai.OpenAI", return_value=client): + messages = self.partner.message_ids + self.action.with_context( + active_id=self.partner.id, active_model="res.partner" + ).run() + self.assertEqual(client.current_message, 1) + self.assertEqual(messages, self.partner.message_ids) diff --git a/ai_automation_openai/views/ai_connection.xml b/ai_automation_openai/views/ai_connection.xml new file mode 100644 index 00000000..ba9bd2d7 --- /dev/null +++ b/ai_automation_openai/views/ai_connection.xml @@ -0,0 +1,29 @@ + + + + + + ai.connection + + + + + + + + + + + + + + + + + + diff --git a/requirements.txt b/requirements.txt index 4630c4e4..eecdf228 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ # generated from manifests external_dependencies ollama +openai diff --git a/setup/ai_automation_openai/odoo/addons/ai_automation_openai b/setup/ai_automation_openai/odoo/addons/ai_automation_openai new file mode 120000 index 00000000..e092d190 --- /dev/null +++ b/setup/ai_automation_openai/odoo/addons/ai_automation_openai @@ -0,0 +1 @@ +../../../../ai_automation_openai \ No newline at end of file diff --git a/setup/ai_automation_openai/setup.py b/setup/ai_automation_openai/setup.py new file mode 100644 index 00000000..28c57bb6 --- /dev/null +++ b/setup/ai_automation_openai/setup.py @@ -0,0 +1,6 @@ +import setuptools + +setuptools.setup( + setup_requires=['setuptools-odoo'], + odoo_addon=True, +)