agentkit-sdk-python 0.7.5__tar.gz → 0.7.7__tar.gz

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.7.5
+Version: 0.7.7
 Summary: Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure.
 Author-email: Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>, Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>
 License:                                  Apache License

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/agentkit/frameworks/langgraph.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from collections.abc import AsyncGenerator
+import hashlib
 import inspect
 import json
 from typing import Any
@@ -36,6 +37,7 @@ from agentkit.frameworks._common import (
 
 
 LANGGRAPH_PENDING_INTERRUPT_STATE_KEY = "agentkit:langgraph:pending_interrupt"
+_OUTPUT_FIELD_KEYS = ("answer", "output", "final", "response", "text", "messages")
 
 
 def _method_kwargs(method: Any, kwargs: dict[str, Any]) -> dict[str, Any]:
@@ -95,7 +97,7 @@ def _update_output_text(data: Any) -> str:
     if not isinstance(data, dict):
         return chunk_to_text(data)
 
-    for key in ("answer", "output", "text", "content", "messages"):
+    for key in _OUTPUT_FIELD_KEYS:
         if key in data:
             text = chunk_to_text(data[key])
             if text:
@@ -105,7 +107,7 @@ def _update_output_text(data: Any) -> str:
     for value in data.values():
         if not isinstance(value, dict):
             continue
-        for key in ("answer", "output", "text", "content", "messages"):
+        for key in _OUTPUT_FIELD_KEYS:
             if key in value:
                 text = chunk_to_text(value[key])
                 if text:
@@ -141,7 +143,15 @@ class LangGraphAgentkitBridge(BaseAgent):
         session_id = getattr(session, "id", None)
         if not session_id:
             return None
-        return {"configurable": {"thread_id": session_id}}
+        app_name = getattr(session, "app_name", None)
+        user_id = getattr(session, "user_id", None)
+        if app_name and user_id:
+            raw_identity = f"{app_name}\0{user_id}\0{session_id}"
+            digest = hashlib.sha256(raw_identity.encode("utf-8")).hexdigest()
+            thread_id = f"agentkit:{digest}"
+        else:
+            thread_id = session_id
+        return {"configurable": {"thread_id": thread_id}}
 
     def _pending_interrupt(self, ctx: InvocationContext) -> Any | None:
         session = getattr(ctx, "session", None)
@@ -218,32 +228,37 @@ class LangGraphAgentkitBridge(BaseAgent):
         stream_fn = getattr(self._graph, "stream", None)
         if not callable(stream_fn):
             return None
-        try:
-            return stream_fn(
-                payload,
-                **_method_kwargs(
-                    stream_fn,
-                    {
-                        "config": config,
-                        "stream_mode": ["messages", "updates"],
-                        "version": "v2",
-                    },
-                ),
-            )
-        except TypeError:
-            try:
-                return stream_fn(
-                    payload,
-                    **_method_kwargs(
-                        stream_fn,
-                        {
-                            "config": config,
-                            "stream_mode": ["messages", "updates"],
-                        },
-                    ),
-                )
-            except TypeError:
-                return stream_fn(payload, **_method_kwargs(stream_fn, {"config": config}))
+        attempts = [
+            {
+                "config": config,
+                "stream_mode": ["messages", "updates"],
+                "version": "v2",
+            },
+            {
+                "config": config,
+                "stream_mode": ["messages", "updates"],
+            },
+            {"config": config},
+        ]
+
+        def iterate_attempts():
+            last_error: TypeError | None = None
+            for kwargs in attempts:
+                emitted = False
+                try:
+                    stream = stream_fn(payload, **_method_kwargs(stream_fn, kwargs))
+                    for item in stream:
+                        emitted = True
+                        yield item
+                    return
+                except TypeError as exc:
+                    if emitted:
+                        raise
+                    last_error = exc
+            if last_error is not None:
+                raise last_error
+
+        return iterate_attempts()
 
     async def _stream_graph(self, payload: Any, config: dict[str, Any] | None, *, prefer_sync: bool = False):
         if prefer_sync:
@@ -255,35 +270,33 @@ class LangGraphAgentkitBridge(BaseAgent):
 
         astream = getattr(self._graph, "astream", None)
         if callable(astream):
-            try:
-                stream = astream(
-                    payload,
-                    **_method_kwargs(
-                        astream,
-                        {
-                            "config": config,
-                            "stream_mode": ["messages", "updates"],
-                            "version": "v2",
-                        },
-                    ),
-                )
-            except TypeError:
+            attempts = [
+                {
+                    "config": config,
+                    "stream_mode": ["messages", "updates"],
+                    "version": "v2",
+                },
+                {
+                    "config": config,
+                    "stream_mode": ["messages", "updates"],
+                },
+                {"config": config},
+            ]
+            last_error: TypeError | None = None
+            for kwargs in attempts:
+                emitted = False
                 try:
-                    stream = astream(
-                        payload,
-                        **_method_kwargs(
-                            astream,
-                            {
-                                "config": config,
-                                "stream_mode": ["messages", "updates"],
-                            },
-                        ),
-                    )
-                except TypeError:
-                    stream = astream(payload, **_method_kwargs(astream, {"config": config}))
-            async for item in stream:
-                yield item
-            return
+                    stream = astream(payload, **_method_kwargs(astream, kwargs))
+                    async for item in stream:
+                        emitted = True
+                        yield item
+                    return
+                except TypeError as exc:
+                    if emitted:
+                        raise
+                    last_error = exc
+            if last_error is not None:
+                raise last_error
 
         stream = self._sync_stream(payload, config)
         if stream is not None:
@@ -313,6 +326,8 @@ class LangGraphAgentkitBridge(BaseAgent):
             elif isinstance(item, dict) and isinstance(item.get("type"), str):
                 mode = item["type"]
                 data = item.get("data")
+            elif isinstance(item, dict):
+                mode = "updates"
 
             if mode == "messages":
                 message = data[0] if isinstance(data, tuple) and data else data

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/agentkit/frameworks/migration.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 import importlib
 import importlib.util
+import inspect
 from pathlib import Path
 import sys
 from types import ModuleType
@@ -55,6 +56,50 @@ def _resolve_object(module: ModuleType, object_path: str) -> Any:
     return target
 
 
+def _call_zero_arg_factory(target: Any, object_path: str) -> Any:
+    if not callable(target):
+        raise TypeError(
+            f"Entry object {object_path!r} was marked as a factory, "
+            "but the loaded object is not callable."
+        )
+
+    try:
+        signature = inspect.signature(target)
+    except (TypeError, ValueError) as exc:
+        raise TypeError(
+            f"Entry factory {object_path!r} has no inspectable signature. "
+            "Expose a zero-argument factory or a constructed agent object."
+        ) from exc
+
+    required_params = [
+        name
+        for name, param in signature.parameters.items()
+        if param.default is inspect.Parameter.empty
+        and param.kind
+        in {
+            inspect.Parameter.POSITIONAL_ONLY,
+            inspect.Parameter.POSITIONAL_OR_KEYWORD,
+            inspect.Parameter.KEYWORD_ONLY,
+        }
+    ]
+    if required_params:
+        formatted = ", ".join(required_params)
+        raise TypeError(
+            f"Entry factory {object_path!r} requires arguments: {formatted}. "
+            "agentkit migrate can only call zero-argument factories; expose a "
+            "thin zero-argument entry object for migration."
+        )
+
+    result = target()
+    if inspect.isawaitable(result):
+        raise TypeError(
+            f"Entry factory {object_path!r} returned an awaitable object. "
+            "Async factories are not supported by generated migration apps; "
+            "expose a synchronous factory or a constructed agent object."
+        )
+    return result
+
+
 def load_entry_object(
     *,
     file: str,
@@ -63,6 +108,7 @@ def load_entry_object(
     project_root: str | Path = ".",
     base_dir: str | Path | None = None,
     import_name: str = "agentkit_migrated_entry",
+    call_factory: bool = False,
 ) -> Any:
     """Load an object from a migrated project's original entry reference.
 
@@ -83,4 +129,7 @@ def load_entry_object(
         entry_path = (base_path / file).resolve()
         loaded_module = _load_module_from_file(entry_path, import_name)
 
-    return _resolve_object(loaded_module, object_path)
+    target = _resolve_object(loaded_module, object_path)
+    if call_factory:
+        return _call_zero_arg_factory(target, object_path)
+    return target

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/agentkit/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.7.5"
+VERSION = "0.7.7"

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/agentkit_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.7.5
+Version: 0.7.7
 Summary: Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure.
 Author-email: Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>, Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>
 License:                                  Apache License

{agentkit_sdk_python-0.7.5 → agentkit_sdk_python-0.7.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentkit-sdk-python"
-version = "0.7.5"
+version = "0.7.7"
 description = "Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure."
 readme = "README.md"
 requires-python = ">=3.10"