40 lines
818 B
Python
40 lines
818 B
Python
from pydantic import BaseModel
|
|
from typing import Literal, Optional, Dict, Any
|
|
from datetime import datetime
|
|
|
|
class AgentInvocation(BaseModel):
|
|
agent_id: str
|
|
entrypoint: Literal["channel_message", "direct", "cron"] = "channel_message"
|
|
payload: Dict[str, Any]
|
|
|
|
class AgentBlueprint(BaseModel):
|
|
id: str
|
|
name: str
|
|
model: str
|
|
instructions: str
|
|
capabilities: Dict[str, Any] = {}
|
|
tools: list[str] = []
|
|
|
|
class ChannelMessage(BaseModel):
|
|
sender_id: str
|
|
sender_type: Literal["human", "agent"]
|
|
content: str
|
|
created_at: datetime
|
|
|
|
class LLMRequest(BaseModel):
|
|
model: str
|
|
messages: list[Dict[str, str]]
|
|
max_tokens: int = 1000
|
|
temperature: float = 0.7
|
|
|
|
class LLMResponse(BaseModel):
|
|
content: str
|
|
model: str
|
|
usage: Optional[Dict[str, int]] = None
|
|
|
|
|
|
|
|
|
|
|
|
|