agentkit-sdk-python 0.7.5__py3-none-any.whl → 0.7.7__py3-none-any.whl

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/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/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.dist-info → agentkit_sdk_python-0.7.7.dist-info}/METADATA

@@ -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.dist-info → agentkit_sdk_python-0.7.7.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agentkit/__init__.py,sha256=l27ZMDslc3VhmmnPZJyrqVvTDoZ0LqhCtM5hw0caHcU,1021
 agentkit/errors.py,sha256=UWlXv0ZsXd_oLTMlxuf4u9-XkTdRYeVZueE--Z3JsHU,1214
-agentkit/version.py,sha256=jdEhJm4AO9pEv0qL-YVddyFb51tMohK9OaZmnNwBxME,653
+agentkit/version.py,sha256=aY0LbH3uWPVe2OXWCa9W7LLbRNa4t_a7XqmwB7dpKlw,653
 agentkit/apps/__init__.py,sha256=-oXTjxV3gejpJ8pffTUcUAXw0LfxKCYZJk-_hIJhtLM,1921
 agentkit/apps/base_app.py,sha256=3hZZExL1wyTGWveJEZZoqXN086MzmkVS_WU1vyulIWg,754
 agentkit/apps/utils.py,sha256=IzimIDmT6FS6PV9MLimPh46mq5zT-EjVmnYPxBdyEq0,1934
@@ -43,8 +43,8 @@ agentkit/client/base_service_client.py,sha256=zpPUidGMDRAMDnSVjrX7BwuOp4BQdAEk0b
 agentkit/frameworks/__init__.py,sha256=cfUdMeixph0xAo6B5qGQBBdm8LeQbWZIl1vmO9cEL3E,971
 agentkit/frameworks/_common.py,sha256=0L0cyBQxtlOdi3GbQ_UgaOoHWwenztlKrxUsyvV3hXw,3859
 agentkit/frameworks/langchain.py,sha256=Tcy2HGYqKmcugDSn3OMaD_6uxhsJtZI3o5E8Rw7UJrg,5947
-agentkit/frameworks/langgraph.py,sha256=maa-igm302-PhMAJB3yMJhRfZMuRgKrs0wmV-uCSlGc,13788
-agentkit/frameworks/migration.py,sha256=cDXu3FqbmCUEx3uGtN-atkC6pnICaEb9FboJdekpBek,2680
+agentkit/frameworks/langgraph.py,sha256=FQ3O9RDYFWXizvsg3-e-rpCpjbnQEWqVfcUQSKUp5TQ,14292
+agentkit/frameworks/migration.py,sha256=j8MxFvAFKkDsTy_e1l5bN-B276AE7rwxLrHuBodNcfc,4385
 agentkit/frameworks/serving/__init__.py,sha256=EaR0jzP6np8v1gyD6CYlaRoIY1KgBaFosqCmRUgMc2E,302
 agentkit/frameworks/serving/fastapi_mount.py,sha256=LV6gJunwd7aU8MFdDYSxeaIUYWvYwMl2Am1J07ddP0A,1958
 agentkit/frameworks/serving/langserve.py,sha256=R2R0g95LHcSFJ1BnOm2cxnULv3wQaZNUdm5Aqd-Oq5s,17238
@@ -241,9 +241,9 @@ agentkit/utils/redact.py,sha256=RAtiggxH74eb1Da6Gg0G5SLTVWRaIE5RqYnCozSZozQ,2167
 agentkit/utils/request.py,sha256=1IGxPS3BSCKk1K1_j862_v34xzLEx04CIvrLVb1MsGQ,1625
 agentkit/utils/template_utils.py,sha256=Qjg9V6dEpjAd_yYezIIPwuDsDeeMmp6XTMn5ZECifNc,6336
 agentkit/utils/ve_sign.py,sha256=JfKig7zni8KvdhgbLJXm3Zy_ttyzKrYGUmBY_82_A4Q,12762
-agentkit_sdk_python-0.7.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agentkit_sdk_python-0.7.5.dist-info/METADATA,sha256=fT7Oz5J_Xywf4oDXInVpG6kdlLOhRrNUVxmlgKTLGWg,21144
-agentkit_sdk_python-0.7.5.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
-agentkit_sdk_python-0.7.5.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
-agentkit_sdk_python-0.7.5.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
-agentkit_sdk_python-0.7.5.dist-info/RECORD,,
+agentkit_sdk_python-0.7.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agentkit_sdk_python-0.7.7.dist-info/METADATA,sha256=A046DqMlx-7fFuSloq8mRpGmwv2EAW-EQwmNkmhxDZE,21144
+agentkit_sdk_python-0.7.7.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+agentkit_sdk_python-0.7.7.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
+agentkit_sdk_python-0.7.7.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
+agentkit_sdk_python-0.7.7.dist-info/RECORD,,