File size: 895 Bytes
db17bc0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from typing import List, TypedDict
from langchain_core.documents.base import Document
class GraphState(TypedDict):
"""
Represents the state of our adaptive RAG graph.
Attributes:
question (str): Original user question
generation (str, optional): LLM generated answer
documents (List[Document], optional): Retrieved or searched documents
"""
question: str
generation: str | None
documents: List[Document]
class ResearchGraphState(TypedDict):
"""
Represents the state of our adaptive RAG graph.
Attributes:
question (str): Original user question
generation (str, optional): LLM generated answer
documents (List[Document], optional): Retrieved or searched documents
"""
company: str
industry: str | None
research_results: str | None
use_cases: str | None
search_queries: str | None
|