@@ -19,6 +19,7 @@ class AgentState(TypedDict):
refined_queries: List[str]
context: str
iteration_count: int
+ knowledgeType: str
class QueryGenerationAgent: