mirror of
https://github.com/rowboatlabs/rowboat.git
synced 2026-04-25 08:26:22 +02:00
18 lines
383 B
Python
18 lines
383 B
Python
|
from pydantic import BaseModel
|
||
|
|
from typing import Literal, List, Any
|
||
|
|
|
||
|
|
class AgentContext(BaseModel):
|
||
|
|
type: Literal['agent']
|
||
|
|
agentName: str
|
||
|
|
|
||
|
|
class PromptContext(BaseModel):
|
||
|
|
type: Literal['prompt']
|
||
|
|
promptName: str
|
||
|
|
|
||
|
|
class ToolContext(BaseModel):
|
||
|
|
type: Literal['tool']
|
||
|
|
toolName: str
|
||
|
|
|
||
|
|
class ChatContext(BaseModel):
|
||
|
|
type: Literal['chat']
|
||
|
|
messages: List[Any]
|