agentforge / schemas.py
Tahasaif3's picture
'code'
fd1a435
raw
history blame contribute delete
745 Bytes
from pydantic import BaseModel, Field
from typing import List
class ValidationResult(BaseModel):
is_valid: bool = Field(default=False, description="If API key is valid for model")
error: str = Field(default="", description="Validation error if any")
class ExtractionResult(BaseModel):
business_type: str = Field(default="", description="Detected business type (1-3 words)")
core_actions: List[str] = Field(default_factory=list, description="List of 2-4 key actions")
required_tools: List[str] = Field(default_factory=list, description="Snake_case tool names")
tone: str = Field(default="friendly", description="friendly | professional | casual")
error: str = Field(default="", description="Extraction error if any")