mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
"""Type protocols for LangGraph modules."""
|
|
|
|
from typing import Any, Protocol, runtime_checkable
|
|
|
|
|
|
@runtime_checkable
|
|
class LangGraphMemorySaver(Protocol):
|
|
"""Protocol for LangGraph MemorySaver.
|
|
|
|
Defines the interface for LangGraph's memory persistence mechanism.
|
|
"""
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize the memory saver."""
|
|
...
|
|
|
|
|
|
@runtime_checkable
|
|
class LangGraphCheckPointMemoryModule(Protocol):
|
|
"""Protocol for LangGraph checkpoint memory module.
|
|
|
|
Defines the interface for modules containing memory checkpoint functionality.
|
|
"""
|
|
|
|
MemorySaver: type[LangGraphMemorySaver]
|
|
|
|
|
|
@runtime_checkable
|
|
class LangGraphPrebuiltModule(Protocol):
|
|
"""Protocol for LangGraph prebuilt module.
|
|
|
|
Defines the interface for modules containing prebuilt agent factories.
|
|
"""
|
|
|
|
def create_react_agent(
|
|
self,
|
|
model: Any,
|
|
tools: list[Any],
|
|
checkpointer: Any,
|
|
debug: bool = False,
|
|
**kwargs: Any,
|
|
) -> Any:
|
|
"""Create a ReAct agent with the given configuration.
|
|
|
|
Args:
|
|
model: The language model to use for the agent.
|
|
tools: List of tools available to the agent.
|
|
checkpointer: Memory checkpointer for state persistence.
|
|
debug: Whether to enable debug mode.
|
|
**kwargs: Additional configuration options.
|
|
|
|
Returns:
|
|
The configured ReAct agent instance.
|
|
"""
|
|
...
|