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

{flowstate_sdk-0.1.11/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.11
+Version: 0.1.13
 Summary: SDK for Agentic AI workflow management with Flowstate
 Author: Flowstate
 License: Nobody can use this

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

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

{flowstate_sdk-0.1.11 → 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.11/src/flowstate_sdk/langchain/callback_handler.py → flowstate_sdk-0.1.13/src/flowstate_sdk/langchain/telemetry.py

@@ -252,6 +252,31 @@ def _is_tool_call_message(message: Any, gen_info: Optional[Dict[str, Any]]) -> b
     return False
 
 
+def _estimate_token_count(text: Optional[str], model: Optional[str]) -> Optional[int]:
+    if not text:
+        return 0
+    try:
+        import tiktoken  # type: ignore
+    except Exception:
+        tiktoken = None
+
+    if tiktoken is not None:
+        try:
+            try:
+                enc = tiktoken.encoding_for_model(model or "")
+            except Exception:
+                enc = tiktoken.get_encoding("cl100k_base")
+            return len(enc.encode(text))
+        except Exception:
+            return None
+
+    if os.getenv("FLOWSTATE_APPROX_TOKENS"):
+        approx = (len(text) + 3) // 4
+        return approx if approx > 0 else 0
+
+    return None
+
+
 class FlowstateCallbackHandler(BaseCallbackHandler):
     def __init__(self, provider: str, model: str) -> None:
         self.provider = provider
@@ -341,6 +366,25 @@ class FlowstateCallbackHandler(BaseCallbackHandler):
         if finish_reason is None and _is_tool_call_message(ai_message_chunk, gen_info):
             return
 
+        llm_output = getattr(response, "llm_output", None)
+        resolved_model = self.model
+        if isinstance(llm_output, dict):
+            for key in ("model_name", "model", "model_id"):
+                val = llm_output.get(key)
+                if isinstance(val, str) and val.strip():
+                    resolved_model = val
+                    break
+        resolved_model = _normalize_model_name(resolved_model)
+
+        # Extract output text robustly (chat message content or plain .text)
+        output_text = ""
+        if response.generations and response.generations[0]:
+            generation = response.generations[0][0]
+            if getattr(generation, "message", None) is not None:
+                output_text = _content_to_text(generation.message.content)
+            else:
+                output_text = generation.text or ""
+
         usage_candidates: List[Dict[str, Any]] = []
         gen_response_metadata = getattr(generation_chunk, "response_metadata", None)
         if isinstance(gen_response_metadata, dict):
@@ -360,20 +404,15 @@ class FlowstateCallbackHandler(BaseCallbackHandler):
                 usage_candidates.append(additional_kwargs)
         if isinstance(gen_info, dict) and gen_info:
             usage_candidates.append(gen_info)
-        llm_output = getattr(response, "llm_output", None)
         if isinstance(llm_output, dict):
             usage_candidates.append(llm_output)
 
         input_tokens, output_tokens = _extract_token_usage(usage_candidates)
+        if input_tokens is None:
+            input_tokens = _estimate_token_count(self._input_str, resolved_model)
+        if output_tokens is None:
+            output_tokens = _estimate_token_count(output_text, resolved_model)
 
-        resolved_model = self.model
-        if isinstance(llm_output, dict):
-            for key in ("model_name", "model", "model_id"):
-                val = llm_output.get(key)
-                if isinstance(val, str) and val.strip():
-                    resolved_model = val
-                    break
-        resolved_model = _normalize_model_name(resolved_model)
         if os.getenv("FLOWSTATE_DEBUG_LLM_USAGE"):
             try:
                 import json
@@ -410,15 +449,6 @@ class FlowstateCallbackHandler(BaseCallbackHandler):
             if cost_value != 0.0:
                 cost_usd = cost_value
 
-        # Extract output text robustly (chat message content or plain .text)
-        output_text = ""
-        if response.generations and response.generations[0]:
-            generation = response.generations[0][0]
-            if getattr(generation, "message", None) is not None:
-                output_text = _content_to_text(generation.message.content)
-            else:
-                output_text = generation.text or ""
-
         provider_metrics = ProviderMetrics(
             run_id=context.current_run.get(),
             provider=self.provider,

{flowstate_sdk-0.1.11 → 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.11 → 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.11
+Version: 0.1.13
 Summary: SDK for Agentic AI workflow management with Flowstate
 Author: Flowstate
 License: Nobody can use this

{flowstate_sdk-0.1.11 → 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