Spaces:
Sleeping
Sleeping
Upload app/agent/state.py with huggingface_hub
Browse files- app/agent/state.py +5 -18
app/agent/state.py
CHANGED
|
@@ -1,28 +1,15 @@
|
|
| 1 |
-
from typing import TypedDict, List
|
| 2 |
-
from app.models.question import
|
| 3 |
from app.models.checklist import ChecklistItem
|
| 4 |
|
| 5 |
|
| 6 |
class AgentState(TypedDict):
|
| 7 |
-
# Session info
|
| 8 |
session_id: str
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
max_rounds: int # 3
|
| 13 |
-
|
| 14 |
-
# Questions & Answers
|
| 15 |
-
current_questions: List[Question]
|
| 16 |
all_answers: List[Answer]
|
| 17 |
-
pending_transcripts: List[str]
|
| 18 |
-
|
| 19 |
-
# Analysis
|
| 20 |
round_summaries: List[str]
|
| 21 |
-
|
| 22 |
-
# Final output
|
| 23 |
checklist_items: List[ChecklistItem]
|
| 24 |
markdown_content: str
|
| 25 |
-
|
| 26 |
-
# Control flow
|
| 27 |
is_complete: bool
|
| 28 |
-
waiting_for_answers: bool
|
|
|
|
| 1 |
+
from typing import TypedDict, List
|
| 2 |
+
from app.models.question import QuestionOut, Answer
|
| 3 |
from app.models.checklist import ChecklistItem
|
| 4 |
|
| 5 |
|
| 6 |
class AgentState(TypedDict):
|
|
|
|
| 7 |
session_id: str
|
| 8 |
+
current_round: int
|
| 9 |
+
max_rounds: int
|
| 10 |
+
current_questions: List[QuestionOut]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
all_answers: List[Answer]
|
|
|
|
|
|
|
|
|
|
| 12 |
round_summaries: List[str]
|
|
|
|
|
|
|
| 13 |
checklist_items: List[ChecklistItem]
|
| 14 |
markdown_content: str
|
|
|
|
|
|
|
| 15 |
is_complete: bool
|
|
|