Update api/models/requests.py

api/models/requests.py

@@ -1,79 +1,95 @@
 """
-Request models para para.AI API v3.0
-Define estrutura de requests para todos os endpoints
+Request models para a API
+CORRIGIDO: model_config para evitar warning de namespace protegido
 """
+from pydantic import BaseModel, Field, ConfigDict
 from typing import Optional, List
-from pydantic import BaseModel, Field, field_validator
+from datetime import datetime
 
 
-# ============================================================================
-# PROCESSING
-# ============================================================================
-
 class ProcessingOptionsRequest(BaseModel):
-    """Opções de processamento."""
-    llm_provider: str = Field(default="groq", description="Provedor LLM (groq/openai/anthropic)")
-    model_type: str = Field(default="balanced", description="Tipo de modelo (fast/balanced/quality)")
-    enable_parallel: bool = Field(default=True, description="Habilitar processamento paralelo")
-    max_workers: int = Field(default=3, ge=1, le=10, description="Número de workers paralelos")
-    save_to_db: bool = Field(default=False, description="Salvar resultados no banco")
-
-    @field_validator("llm_provider")
-    @classmethod
-    def validate_provider(cls, v):
-        allowed = ["groq", "openai", "anthropic"]
-        if v not in allowed:
-            raise ValueError(f"Provider deve ser um de: {allowed}")
-        return v
-
-    @field_validator("model_type")
-    @classmethod
-    def validate_model_type(cls, v):
-        allowed = ["fast", "balanced", "quality"]
-        if v not in allowed:
-            raise ValueError(f"Model type deve ser um de: {allowed}")
-        return v
-
-
-# ============================================================================
-# LLM TESTING
-# ============================================================================
-
-class LLMGenerateRequest(BaseModel):
-    """Request para gerar texto com LLM."""
-    prompt: str = Field(..., description="Prompt para o LLM", min_length=1)
-    provider: str = Field(default="groq", description="Provedor LLM")
-    model: Optional[str] = Field(default=None, description="Model específico (opcional)")
-    temperature: float = Field(default=0.7, ge=0.0, le=2.0, description="Temperatura")
-    max_tokens: int = Field(default=1024, ge=1, le=4096, description="Máximo de tokens")
-
-
-# ============================================================================
-# PROCESSOR TESTING
-# ============================================================================
-
-class ProcessorTestRequest(BaseModel):
-    """Request para testar processador."""
-    processor_name: str = Field(..., description="Nome do processador (1-9 ou nome completo)")
-    ementa: str = Field(..., description="Texto da ementa", min_length=10)
-    integra: str = Field(..., description="Texto da íntegra", min_length=10)
-    tribunal: str = Field(default="TJPR", description="Sigla do tribunal")
-    llm_provider: str = Field(default="groq", description="Provedor LLM")
-
-
-# ============================================================================
-# FILE OPERATIONS
-# ============================================================================
-
-class FileCreateRequest(BaseModel):
-    """Request para criar arquivo."""
-    content: str = Field(..., description="Conteúdo do arquivo")
-    filename: str = Field(..., description="Nome do arquivo")
-    acordao_id: Optional[str] = Field(default=None, description="ID do acórdão relacionado")
-
-
-class FileListRequest(BaseModel):
-    """Request para listar arquivos."""
-    acordao_id: Optional[str] = Field(default=None, description="Filtrar por acórdão")
+    """
+    Opções de processamento de acórdãos.
+
+    CORRIGIDO: Adicionado model_config para desabilitar protected_namespaces
+    """
+    # ✅ Configuração para permitir campos que começam com "model_"
+    model_config = ConfigDict(protected_namespaces=())
+
+    llm_provider: str = Field(
+        default="groq",
+        description="Provedor LLM (groq, openai, anthropic)"
+    )
+
+    model_type: str = Field(
+        default="balanced",
+        description="Tipo de modelo (fast, balanced, quality)"
+    )
+
+    enable_parallel: bool = Field(
+        default=True,
+        description="Processar especialistas em paralelo"
+    )
+
+    max_workers: int = Field(
+        default=3,
+        ge=1,
+        le=10,
+        description="Número de workers paralelos"
+    )
+
+    save_to_db: bool = Field(
+        default=False,
+        description="Salvar resultados no banco de dados"
+    )
+
+    specialist_ids: Optional[List[int]] = Field(
+        default=None,
+        description="IDs dos especialistas a executar (1-9), None = todos"
+    )
+
+
+class AcordaoUploadRequest(BaseModel):
+    """Request para upload de acórdão individual."""
+
+    acordao_id: str = Field(..., description="ID único do acórdão")
+    tribunal: str = Field(..., description="Tribunal (ex: TJPR)")
+    ementa: str = Field(..., description="Ementa do acórdão")
+    integra: str = Field(..., description="Texto integral do acórdão")
+
+    # Campos opcionais
+    numero_processo: Optional[str] = Field(None, description="Número do processo")
+    relator: Optional[str] = Field(None, description="Nome do relator")
+    data_julgamento: Optional[datetime] = Field(None, description="Data do julgamento")
+    classe_processual: Optional[str] = Field(None, description="Classe processual")
+
+
+class BatchProcessingRequest(BaseModel):
+    """Request para processamento em batch."""
+
+    acordaos: List[AcordaoUploadRequest] = Field(
+        ..., 
+        description="Lista de acórdãos a processar"
+    )
+
+    processing_options: ProcessingOptionsRequest = Field(
+        default_factory=ProcessingOptionsRequest,
+        description="Opções de processamento"
+    )
+
+
+class TaskStatusRequest(BaseModel):
+    """Request para consultar status de task."""
+
+    task_id: str = Field(..., description="ID da task")
+
+
+class DatabaseQueryRequest(BaseModel):
+    """Request para consultas ao banco de dados."""
+
+    tribunal: Optional[str] = Field(None, description="Filtrar por tribunal")
+    relator: Optional[str] = Field(None, description="Filtrar por relator")
+    date_from: Optional[datetime] = Field(None, description="Data inicial")
+    date_to: Optional[datetime] = Field(None, description="Data final")
     limit: int = Field(default=100, ge=1, le=1000, description="Limite de resultados")
     offset: int = Field(default=0, ge=0, description="Offset para paginação")