bemade-addons/ai_integration/models/ai_model.py

# -*- 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