flowstate-sdk 0.1.12__tar.gz → 0.1.13__tar.gz

{flowstate_sdk-0.1.12/src/flowstate_sdk.egg-info → flowstate_sdk-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowstate_sdk
-Version: 0.1.12
+Version: 0.1.13
 Summary: SDK for Agentic AI workflow management with Flowstate
 Author: Flowstate
 License: Nobody can use this

{flowstate_sdk-0.1.12 → flowstate_sdk-0.1.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "flowstate_sdk"
-version = "0.1.12"
+version = "0.1.13"
 description = "SDK for Agentic AI workflow management with Flowstate"
 authors = [{name="Flowstate"}]
 requires-python = ">=3.9"

{flowstate_sdk-0.1.12 → flowstate_sdk-0.1.13}/src/flowstate_sdk/context.py

@@ -7,3 +7,6 @@ run_stack: ContextVar[List[str]] = ContextVar("run_stack", default=[])
 parent_children_mappings: ContextVar[Dict[str, Set[str]]] = ContextVar(
     "parent_children_mappings", default=defaultdict(set)
 )
+is_replay: ContextVar[bool] = ContextVar("is_replay", default=False)
+replay_counter: ContextVar[int] = ContextVar("replay_counter", default=0)
+replay_step_list: ContextVar[List[str]] = ContextVar("replay_step_list", default=[])

flowstate_sdk-0.1.13/src/flowstate_sdk/langchain/__init__.py

@@ -0,0 +1,15 @@
+from flowstate_sdk.langchain.callback_handler import FlowstateCallbackHandler
+from flowstate_sdk.langchain.chat_models import (
+    FlowstateChatAnthropic,
+    FlowstateChatClaude,
+    FlowstateChatGoogle,
+    FlowstateChatOpenAI,
+)
+
+__all__ = [
+    "FlowstateCallbackHandler",
+    "FlowstateChatOpenAI",
+    "FlowstateChatAnthropic",
+    "FlowstateChatClaude",
+    "FlowstateChatGoogle",
+]

flowstate_sdk-0.1.13/src/flowstate_sdk/langchain/callback_handler.py

@@ -0,0 +1,3 @@
+from flowstate_sdk.langchain.telemetry import FlowstateCallbackHandler
+
+__all__ = ["FlowstateCallbackHandler"]

flowstate_sdk-0.1.13/src/flowstate_sdk/langchain/chat_models.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+from typing import Any
+
+from flowstate_sdk.langchain.flowstate_chat_base import FlowstateChatModelMixin
+
+try:
+    from langchain_openai import ChatOpenAI
+except Exception:  # pragma: no cover - optional dependency
+    ChatOpenAI = None
+
+try:
+    from langchain_anthropic import ChatAnthropic
+except Exception:  # pragma: no cover - optional dependency
+    ChatAnthropic = None
+
+try:
+    from langchain_google_genai import ChatGoogleGenerativeAI
+except Exception:  # pragma: no cover - optional dependency
+    ChatGoogleGenerativeAI = None
+
+
+def _missing_dependency(name: str, package: str) -> None:
+    raise ImportError(
+        f"{name} requires optional dependency '{package}'. "
+        f"Install it to use this Flowstate wrapper."
+    )
+
+
+if ChatOpenAI is not None:
+
+    class FlowstateChatOpenAI(FlowstateChatModelMixin, ChatOpenAI):
+        provider = "openai"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            kwargs = self._flowstate_init_kwargs(kwargs)
+            super().__init__(*args, **kwargs)
+
+else:
+
+    class FlowstateChatOpenAI(FlowstateChatModelMixin):  # type: ignore[misc]
+        provider = "openai"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            _missing_dependency("FlowstateChatOpenAI", "langchain-openai")
+
+
+if ChatAnthropic is not None:
+
+    class FlowstateChatAnthropic(FlowstateChatModelMixin, ChatAnthropic):
+        provider = "anthropic"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            kwargs = self._flowstate_init_kwargs(kwargs)
+            super().__init__(*args, **kwargs)
+
+else:
+
+    class FlowstateChatAnthropic(FlowstateChatModelMixin):  # type: ignore[misc]
+        provider = "anthropic"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            _missing_dependency("FlowstateChatAnthropic", "langchain-anthropic")
+
+
+if ChatGoogleGenerativeAI is not None:
+
+    class FlowstateChatGoogle(FlowstateChatModelMixin, ChatGoogleGenerativeAI):
+        provider = "google"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            kwargs = self._flowstate_init_kwargs(kwargs)
+            super().__init__(*args, **kwargs)
+
+else:
+
+    class FlowstateChatGoogle(FlowstateChatModelMixin):  # type: ignore[misc]
+        provider = "google"
+
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            _missing_dependency("FlowstateChatGoogle", "langchain-google-genai")
+
+
+FlowstateChatClaude = FlowstateChatAnthropic
+
+__all__ = [
+    "FlowstateChatOpenAI",
+    "FlowstateChatAnthropic",
+    "FlowstateChatClaude",
+    "FlowstateChatGoogle",
+]

flowstate_sdk-0.1.13/src/flowstate_sdk/langchain/flowstate_chat_base.py

@@ -0,0 +1,51 @@
+from __future__ import annotations
+
+from typing import Any, Dict, Iterable, List, Optional
+
+from flowstate_sdk.langchain.telemetry import FlowstateCallbackHandler
+
+
+def _extract_model_name(kwargs: Dict[str, Any]) -> Optional[str]:
+    for key in ("model", "model_name", "model_id"):
+        val = kwargs.get(key)
+        if isinstance(val, str) and val.strip():
+            return val
+    return None
+
+
+def _normalize_callbacks(callbacks: Any) -> List[Any]:
+    if callbacks is None:
+        return []
+    if isinstance(callbacks, list):
+        return list(callbacks)
+    if isinstance(callbacks, tuple):
+        return list(callbacks)
+    return [callbacks]
+
+
+def _has_flowstate_callback(callbacks: Iterable[Any]) -> bool:
+    for cb in callbacks:
+        if isinstance(cb, FlowstateCallbackHandler):
+            return True
+    return False
+
+
+def _merge_callbacks(existing: Any, additional: List[Any]) -> List[Any]:
+    existing_list = _normalize_callbacks(existing)
+    if _has_flowstate_callback(existing_list):
+        return existing_list
+    return existing_list + list(additional)
+
+
+class FlowstateChatModelMixin:
+    provider: str = ""
+
+    @classmethod
+    def _flowstate_build_callbacks(cls, kwargs: Dict[str, Any]) -> List[Any]:
+        model = _extract_model_name(kwargs) or "unknown"
+        handler = FlowstateCallbackHandler(cls.provider, model)
+        return _merge_callbacks(kwargs.pop("callbacks", None), [handler])
+
+    def _flowstate_init_kwargs(self, kwargs: Dict[str, Any]) -> Dict[str, Any]:
+        kwargs["callbacks"] = self._flowstate_build_callbacks(kwargs)
+        return kwargs

{flowstate_sdk-0.1.12 → flowstate_sdk-0.1.13}/src/flowstate_sdk/workflow.py

@@ -1,6 +1,7 @@
 import uuid
 from collections import defaultdict
 from datetime import datetime
+import os
 
 from . import context
 from .sdk_client import SDKClient
@@ -40,6 +41,12 @@ class Workflow:
         }
         self.client.emit(workflow_payload)
 
+        if os.getenv("IS_FLOWSTATE_REPLAY"):
+            replay_list = self.client.retrieve_replay_plan(os.getenv("FLOWSTATE_REPLAY_PARENT_ID"))
+            context.replay_counter.set(0)
+            context.replay_step_list.set(replay_list)
+            context.is_replay.set(True)
+
         run_payload = RunRecord(
             run_id=str(self.run_id),
             workflow_id=str(self.workflow_id),

{flowstate_sdk-0.1.12 → flowstate_sdk-0.1.13/src/flowstate_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowstate_sdk
-Version: 0.1.12
+Version: 0.1.13
 Summary: SDK for Agentic AI workflow management with Flowstate
 Author: Flowstate
 License: Nobody can use this

{flowstate_sdk-0.1.12 → flowstate_sdk-0.1.13}/src/flowstate_sdk.egg-info/SOURCES.txt

@@ -17,4 +17,8 @@ src/flowstate_sdk.egg-info/SOURCES.txt
 src/flowstate_sdk.egg-info/dependency_links.txt
 src/flowstate_sdk.egg-info/requires.txt
 src/flowstate_sdk.egg-info/top_level.txt
-src/flowstate_sdk/langchain/callback_handler.py
+src/flowstate_sdk/langchain/__init__.py
+src/flowstate_sdk/langchain/callback_handler.py
+src/flowstate_sdk/langchain/chat_models.py
+src/flowstate_sdk/langchain/flowstate_chat_base.py
+src/flowstate_sdk/langchain/telemetry.py