93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
from odoo import models, fields, api, _
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class AIModel(models.Model):
|
|
_name = 'ai.model'
|
|
_description = 'AI Model'
|
|
_order = 'sequence, name'
|
|
_check_company = False # Disable automatic company checks
|
|
|
|
@api.model
|
|
def _has_provider_modules(self):
|
|
"""Check if any AI provider modules are installed."""
|
|
modules = ['ollama_ai_integration', 'chatgpt_ai_integration']
|
|
return any(self.env['ir.module.module'].search([('name', 'in', modules), ('state', '=', 'installed')]))
|
|
|
|
@api.model
|
|
def default_get(self, fields_list):
|
|
"""Override default_get to prevent creation if no provider modules are installed."""
|
|
if not self._has_provider_modules():
|
|
raise UserError(_('No AI provider modules are installed. Please install at least one provider module (e.g., Ollama or ChatGPT) before creating an AI model.'))
|
|
return super().default_get(fields_list)
|
|
|
|
active = fields.Boolean(
|
|
string='Active',
|
|
default=True,
|
|
help='Whether this model is active and available for use')
|
|
|
|
name = fields.Char(
|
|
string='Name',
|
|
required=True,
|
|
help='Name of the AI model'
|
|
)
|
|
|
|
identifier = fields.Char(
|
|
string='Identifier',
|
|
required=True,
|
|
help='Technical identifier of the model (e.g., gpt-3.5-turbo, mistral-7b)'
|
|
)
|
|
|
|
provider_instance_id = fields.Many2one(
|
|
'ai.provider.instance',
|
|
string='Provider Instance',
|
|
required=True,
|
|
ondelete='cascade',
|
|
help='Provider instance this model belongs to'
|
|
)
|
|
|
|
provider_type = fields.Selection(
|
|
related='provider_instance_id.provider_type',
|
|
string='Provider Type',
|
|
store=True,
|
|
readonly=True,
|
|
help='Type of AI provider'
|
|
)
|
|
|
|
description = fields.Text(
|
|
string='Description',
|
|
help='Description of the model and its capabilities'
|
|
)
|
|
|
|
sequence = fields.Integer(
|
|
string='Sequence',
|
|
default=10,
|
|
help='Sequence for ordering models in lists and dropdowns'
|
|
)
|
|
|
|
is_active = fields.Boolean(
|
|
string='Model Active',
|
|
default=True,
|
|
help='Whether this model is currently active and available for use'
|
|
)
|
|
|
|
context_window = fields.Integer(
|
|
string='Context Window',
|
|
default=2048,
|
|
help='Maximum number of tokens in the context window'
|
|
)
|
|
|
|
_sql_constraints = [
|
|
('unique_identifier_provider',
|
|
'unique(identifier, provider_instance_id)',
|
|
'The model identifier must be unique per provider instance!')
|
|
]
|
|
|
|
def name_get(self):
|
|
"""Custom name_get to include provider instance in display name."""
|
|
result = []
|
|
for model in self:
|
|
name = f"{model.name} ({model.provider_instance_id.name})"
|
|
result.append((model.id, name))
|
|
return result
|