Feat: Moved to langchain based providers

This commit is contained in:
2025-11-17 22:21:21 -05:00
parent e9cf5228dd
commit 958c29b18a
43 changed files with 11903 additions and 291 deletions
+3
View File
@@ -0,0 +1,3 @@
"""Type stubs for langgraph package."""
__all__: list[str] = []
@@ -0,0 +1,3 @@
"""Type stubs for langgraph.checkpoint module."""
__all__: list[str] = []
+8
View File
@@ -0,0 +1,8 @@
"""Type stubs for langgraph.checkpoint.memory module."""
class MemorySaver:
"""In-memory checkpoint saver for state graphs."""
def __init__(self) -> None: ...
__all__ = ["MemorySaver"]
+51
View File
@@ -0,0 +1,51 @@
"""Type stubs for langgraph.graph module."""
from collections.abc import Callable, Iterator
from typing import Any, Literal, TypeVar, overload
_StateT = TypeVar("_StateT")
END: str
class StateGraph:
"""A graph that maintains state across node executions."""
def __init__(self, state_schema: type[Any]) -> None: ...
@overload
def add_node(self, name: str, action: Callable[[Any], Any]) -> StateGraph: ...
@overload
def add_node(self, action: Callable[[Any], Any]) -> StateGraph: ...
def add_edge(self, start: str, end: str) -> StateGraph: ...
def add_conditional_edges(
self,
source: str,
path: Callable[[Any], str],
path_map: dict[str, str],
) -> StateGraph: ...
def set_entry_point(self, node: str) -> StateGraph: ...
def compile(
self,
checkpointer: Any = None,
*,
cache: Any = None,
store: Any = None,
interrupt_before: list[str] | Literal["*"] | None = None,
interrupt_after: list[str] | Literal["*"] | None = None,
debug: bool = False,
name: str | None = None,
) -> CompiledStateGraph: ...
class CompiledStateGraph:
"""A compiled state graph ready for execution."""
def invoke(
self, input: dict[str, Any], config: dict[str, Any] | None = None
) -> dict[str, Any]: ...
async def ainvoke(
self, input: dict[str, Any], config: dict[str, Any] | None = None
) -> dict[str, Any]: ...
def stream(
self, input: dict[str, Any], config: dict[str, Any] | None = None
) -> Iterator[dict[str, Any]]: ...
__all__ = ["END", "CompiledStateGraph", "StateGraph"]