Spaces:
Running
on
T4
Running
on
T4
File size: 772 Bytes
d2beadd |
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 |
from pydantic import BaseModel
# Planner: QueryAnalysis
class QueryAnalysis(BaseModel):
consice_summary: str
required_skills: str
relevant_tools: str
additional_considerations: str
def __str__(self):
return f"""
Consice Summary: {self.consice_summary}
Required Skills:
{self.required_skills}
Relevant Tools:
{self.relevant_tools}
Additional Considerations:
{self.additional_considerations}
"""
# Planner: NextStep
class NextStep(BaseModel):
justification: str
context: str
sub_goal: str
tool_name: str
# Executor: MemoryVerification
class MemoryVerification(BaseModel):
analysis: str
stop_signal: bool
# Executor: ToolCommand
class ToolCommand(BaseModel):
analysis: str
explanation: str
command: str
|