File size: 539 Bytes
2f235a0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ef83e66
2f235a0
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
from typing import Dict, Any, List
from pydantic import BaseModel


class AgentRequest(BaseModel):
    tenant_id: str
    user_id: str | None
    message: str
    conversation_history: List[Dict[str, str]] = []
    temperature: float = 0.0


class AgentDecision(BaseModel):
    action: str
    tool: str | None
    tool_input: Dict[str, Any] | None
    reason: str | None


class AgentResponse(BaseModel):
    text: str
    decision: AgentDecision
    tool_traces: List[Dict[str, Any]] = []
    reasoning_trace: List[Dict[str, Any]] = []