RA.Aid/ra_aid/agents_alias.py

12 lines
282 B
Python

from typing import TYPE_CHECKING
from langgraph.graph.graph import CompiledGraph
# Unfortunately need this to avoid Circular Imports
if TYPE_CHECKING:
from ra_aid.agents.ciayn_agent import CiaynAgent
RAgents = CompiledGraph | CiaynAgent
else:
RAgents = CompiledGraph