Update app.py

app.py

@@ -69,6 +69,26 @@ def main():
     ]
 
 
+    class GraphState(TypedDict):
+    """
+    Represents the state of our graph.
+
+    Attributes:
+        question: question
+        generation: LLM generation
+        search: whether to add search
+        documents: list of documents
+        generations_count : generations count
+    """
+
+        question: str
+        generation: str
+        search: str
+        documents: List[str]
+        steps: List[str]
+        generation_count: int
+
+    
     search_type = st.selectbox(
         "Choose search type. Options are [Max marginal relevance search (similarity) , Similarity search (similarity). Default value (similarity)]", 
         options=["mmr", "similarity"], 
@@ -89,6 +109,7 @@ def main():
     )
     
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=350, chunk_overlap=30)
+