agent_loop.py 446 B

1234567891011121314151617181920212223
  1. from pydantic import BaseModel
  2. class AgentLoop(BaseModel):
  3. position: int = 1
  4. thought: str = None
  5. tool_name: str = None
  6. tool_input: str = None
  7. tool_output: str = None
  8. prompt: str = None
  9. prompt_tokens: int = 0
  10. completion: str = None
  11. completion_tokens: int = 0
  12. latency: float = None
  13. status: str = 'llm_started'
  14. completed: bool = False
  15. started_at: float = None
  16. completed_at: float = None