Corregir modelos de IA

This commit is contained in:
2025-11-25 06:32:47 -03:00
parent b6c7a9ed63
commit 093256382c
3 changed files with 14 additions and 3 deletions

View File

@@ -1131,11 +1131,21 @@ def create_ai_configuration(
# Desactivar configuraciones anteriores # Desactivar configuraciones anteriores
db.query(models.AIConfiguration).update({"is_active": False}) db.query(models.AIConfiguration).update({"is_active": False})
# Determinar modelo por defecto según el proveedor si no se especifica
model_name = config.model_name
if not model_name:
if config.provider == "openai":
model_name = "gpt-4o"
elif config.provider == "gemini":
model_name = "gemini-2.5-pro"
else:
model_name = "default"
# Crear nueva configuración # Crear nueva configuración
new_config = models.AIConfiguration( new_config = models.AIConfiguration(
provider=config.provider, provider=config.provider,
api_key=config.api_key, api_key=config.api_key,
model_name=config.model_name, model_name=model_name,
is_active=True is_active=True
) )

View File

@@ -180,7 +180,8 @@ class AIConfiguration(Base):
id = Column(Integer, primary_key=True, index=True) id = Column(Integer, primary_key=True, index=True)
provider = Column(String(50), nullable=False) # openai, gemini provider = Column(String(50), nullable=False) # openai, gemini
api_key = Column(Text, nullable=False) api_key = Column(Text, nullable=False)
model_name = Column(String(100), nullable=False) model_name = Column(String(100), nullable=True)
logo_url = Column(Text, nullable=True) # URL del logo configurable
is_active = Column(Boolean, default=True) is_active = Column(Boolean, default=True)
created_at = Column(DateTime(timezone=True), server_default=func.now()) created_at = Column(DateTime(timezone=True), server_default=func.now())
updated_at = Column(DateTime(timezone=True), onupdate=func.now()) updated_at = Column(DateTime(timezone=True), onupdate=func.now())

View File

@@ -216,7 +216,7 @@ class InspectionDetail(Inspection):
class AIConfigurationBase(BaseModel): class AIConfigurationBase(BaseModel):
provider: str # openai, gemini provider: str # openai, gemini
api_key: str api_key: str
model_name: str model_name: Optional[str] = None
class AIConfigurationCreate(AIConfigurationBase): class AIConfigurationCreate(AIConfigurationBase):
pass pass