alexorlov commited on
Commit
d65b50e
·
verified ·
1 Parent(s): 6368256

Upload app/agent/state.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app/agent/state.py +5 -18
app/agent/state.py CHANGED
@@ -1,28 +1,15 @@
1
- from typing import TypedDict, List, Optional
2
- from app.models.question import Question, Answer
3
  from app.models.checklist import ChecklistItem
4
 
5
 
6
  class AgentState(TypedDict):
7
- # Session info
8
  session_id: str
9
-
10
- # Round tracking
11
- current_round: int # 1, 2, or 3
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