fixing Class Initiation

app.py

@@ -14,6 +14,13 @@ import re
 
 load_dotenv()
 
+class DocumentState(TypedDict):
+    documents: list[dict]
+    summaries: list[str]
+    search_results: list[dict]
+    search_query: str
+    needs_search: bool
+
 def get_llm_response(prompt):
     url = "http://192.168.181.215:8081/llms"
     headers = {"Content-Type": "application/json"}
@@ -149,13 +156,6 @@ Response:"""
     state['summaries'] = [llm_response['response']]
     return state
 
-class DocumentState(TypedDict):
-    documents: list[dict]
-    summaries: list[str]
-    search_results: list[dict]
-    search_query: str
-    needs_search: bool
-
 def document_extractor_agent(state: DocumentState, pdf_path: str) -> DocumentState:
     """Extracts documents from a PDF file."""
     try: