Update src/state/rag_state.py

src/state/rag_state.py

@@ -1,12 +1,13 @@
-"""RAG state definition for LangGraph"""
-
-from typing import List
-from pydantic import BaseModel
-from langchain.schema import Document
-
-class RAGState(BaseModel):
-    """State object for RAG workflow"""
-    
-    question: str
-    retrieved_docs: List[Document] = []
+"""RAG state definition for LangGraph"""
+
+from typing import List
+from pydantic import BaseModel
+# from langchain.schema import Document
+from langchain_core.documents import Document
+
+class RAGState(BaseModel):
+    """State object for RAG workflow"""
+    
+    question: str
+    retrieved_docs: List[Document] = []
     answer: str = ""