File size: 554 Bytes
30ee88a 0fc97a4 30ee88a 0fc97a4 30ee88a 0fc97a4 30ee88a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from typing import TypedDict, Annotated, Sequence, Optional, List
from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage, ToolMessage
from langgraph.graph.message import add_messages
from pydantic import BaseModel, Field
from enum import Enum
class RAG_method(str, Enum):
RAG = "RAG"
WEBSEARCH = "WEBSEARCH"
GENERAL = "GENERAL"
class AgentState(TypedDict):
messages: Annotated[Sequence[AnyMessage], add_messages]
rag_method: RAG_method
context: Optional[str]
if __name__ == "__main__":
pass |