demo2 / src /state /rag_state.py
Dinesh310's picture
Update src/state/rag_state.py
a9cc27c verified
raw
history blame contribute delete
334 Bytes
"""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 = ""