| 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 | |