File size: 1,222 Bytes
18f5c57 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
from uuid import UUID
from datetime import datetime
from typing import Optional, List
from pydantic import BaseModel, Field
class Agent(BaseModel):
id: str
name: str
project_id: str
input_variables: Optional[List[str]]
prompt: str
is_template: Optional[bool]
routes: Optional[List[str]]
routes_description: Optional[List[str]]
output_collector: Optional[str]
output_variables: Optional[List[str]]
input_api_endpoints: Optional[List[str]]
output_api_endpoints: Optional[List[str]]
conversation_style: float = Field(..., ge=-9.9, le=9.9)
creativity: float = Field(..., ge=0.0, le=1.0)
fallback_message: str
fallback_threshold: float
class OutputCollector(BaseModel):
id: str
project_id: str
keys: List[str]
descriptions: List[str]
data_types: List[str]
class APIEndpoint(BaseModel):
id: str
project_id: str
name: str
description: Optional[str]
input_variables: Optional[list]
url: str
method: str
headers: Optional[dict]
params: Optional[dict]
request_body: Optional[dict]
html_to_markdown: Optional[bool]
html_tags_to_extract: Optional[list]
response_type: Optional[str]
|