File size: 334 Bytes
a9cc27c
 
 
 
 
 
 
 
 
 
 
 
8d56dc3
1
2
3
4
5
6
7
8
9
10
11
12
13
"""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 = ""