agentkit-sdk-python 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

agentkit/apps/agent_server_app/agent_server_app.py

@@ -24,6 +24,7 @@ from fastapi.responses import StreamingResponse
 from google.adk.a2a.utils.agent_to_a2a import to_a2a
 from google.adk.agents.base_agent import BaseAgent
 from google.adk.agents.run_config import RunConfig, StreamingMode
+from google.adk.apps.app import App
 from google.adk.artifacts.in_memory_artifact_service import (
     InMemoryArtifactService,
 )
@@ -42,8 +43,8 @@ from google.adk.utils.context_utils import Aclosing
 from google.genai import types
 from opentelemetry import trace
 from veadk import Agent
-from veadk.memory.short_term_memory import ShortTermMemory
 from veadk.runner import Runner
+from veadk.memory.short_term_memory import ShortTermMemory
 
 from agentkit.apps.agent_server_app.middleware import (
     AgentkitTelemetryHTTPMiddleware,
@@ -55,27 +56,37 @@ logger = logging.getLogger(__name__)
 
 
 class AgentKitAgentLoader(BaseAgentLoader):
-    def __init__(self, agent: BaseAgent) -> None:
+    def __init__(self, agent_or_app: BaseAgent | App) -> None:
         super().__init__()
 
-        self.agent = agent
+        self.agent_or_app = agent_or_app
+        if isinstance(agent_or_app, App):
+            self.root_agent = agent_or_app.root_agent
+            self.app_name = agent_or_app.name or self.root_agent.name
+        else:
+            self.root_agent = agent_or_app
+            self.app_name = agent_or_app.name
 
     @override
-    def load_agent(self, agent_name: str) -> BaseAgent:
-        return self.agent
+    def load_agent(self, agent_name: str) -> BaseAgent | App:
+        if agent_name != self.app_name:
+            raise ValueError(
+                f"Unknown agent '{agent_name}'. Expected '{self.app_name}'."
+            )
+        return self.agent_or_app
 
     @override
     def list_agents(self) -> list[str]:
-        return [self.agent.name]
+        return [self.app_name]
 
     @override
     def list_agents_detailed(self) -> list[dict[str, Any]]:
-        name = self.agent.name
-        description = getattr(self.agent, "description", "") or ""
+        name = self.app_name
+        description = getattr(self.root_agent, "description", "") or ""
         return [
             {
                 "name": name,
-                "root_agent_name": name,
+                "root_agent_name": self.root_agent.name,
                 "description": description,
                 "language": "python",
             }
@@ -85,11 +96,25 @@ class AgentKitAgentLoader(BaseAgentLoader):
 class AgentkitAgentServerApp(BaseAgentkitApp):
     def __init__(
         self,
-        agent: BaseAgent,
-        short_term_memory: BaseSessionService | ShortTermMemory,
+        agent: BaseAgent | App | None = None,
+        short_term_memory: BaseSessionService | ShortTermMemory | None = None,
+        *,
+        app: App | None = None,
     ) -> None:
         super().__init__()
 
+        if short_term_memory is None:
+            raise TypeError("short_term_memory is required.")
+
+        if app is not None and agent is not None:
+            raise TypeError("Only one of 'agent' or 'app' can be provided.")
+
+        entry = app if app is not None else agent
+        if entry is None:
+            raise TypeError("Either 'agent' or 'app' must be provided.")
+
+        root_agent = entry.root_agent if isinstance(entry, App) else entry
+
         _artifact_service = InMemoryArtifactService()
         _credential_service = InMemoryCredentialService()
 
@@ -97,12 +122,12 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
         _eval_set_results_manager = LocalEvalSetResultsManager(agents_dir=".")
 
         self.server = AdkWebServer(
-            agent_loader=AgentKitAgentLoader(agent),
+            agent_loader=AgentKitAgentLoader(entry),
             session_service=short_term_memory
             if isinstance(short_term_memory, BaseSessionService)
             else short_term_memory.session_service,
-            memory_service=agent.long_term_memory
-            if isinstance(agent, Agent) and agent.long_term_memory
+            memory_service=root_agent.long_term_memory
+            if isinstance(root_agent, Agent) and root_agent.long_term_memory
             else InMemoryMemoryService(),
             artifact_service=_artifact_service,
             credential_service=_credential_service,
@@ -111,8 +136,8 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
             agents_dir=".",
         )
 
-        runner = Runner(agent=agent)
-        _a2a_server_app = to_a2a(agent=agent, runner=runner)
+        runner = Runner(agent=root_agent)
+        _a2a_server_app = to_a2a(agent=root_agent, runner=runner)
 
         @asynccontextmanager
         async def lifespan(app: FastAPI):

agentkit/version.py

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

{agentkit_sdk_python-0.4.3.dist-info → agentkit_sdk_python-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.4.3
+Version: 0.4.4
 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.4.3.dist-info → agentkit_sdk_python-0.4.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 agentkit/__init__.py,sha256=l27ZMDslc3VhmmnPZJyrqVvTDoZ0LqhCtM5hw0caHcU,1021
-agentkit/version.py,sha256=WWpi0MzdPmp_MuiTNwGjRNGOPRdyCUzfNHz9bivjHf4,653
+agentkit/version.py,sha256=gzPlMiiIfnq-nXw3IJopMiKzEmgsppA5t46q-oQRpKY,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
@@ -7,7 +7,7 @@ agentkit/apps/a2a_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7
 agentkit/apps/a2a_app/a2a_app.py,sha256=mFuPFuqRlGdQ2VIjQq2_gtQKWwnkwSjSmIMRnammWJU,7605
 agentkit/apps/a2a_app/telemetry.py,sha256=vR8tf7EHLor62cxs5PBx4mrfHeOaV1a7Xryqlu8nGCU,4290
 agentkit/apps/agent_server_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
-agentkit/apps/agent_server_app/agent_server_app.py,sha256=G0RaXGHOMvdlsbU3aUjSWqqEPvQj55hVM6JCjJlaazc,8895
+agentkit/apps/agent_server_app/agent_server_app.py,sha256=9PQWdhNK5A7d8gYvtHPDqs_lRzPw03DLVbjclFuBfyk,9975
 agentkit/apps/agent_server_app/middleware.py,sha256=2gLCRN-JYjYsTQzHqRBU69S2OI8gD5AdbrFeoGzDcM0,2962
 agentkit/apps/agent_server_app/telemetry.py,sha256=js3byvtdPr_anbrptSHtIS3R9MtSPqtKLkVC9kwL7LU,4160
 agentkit/apps/mcp_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
@@ -169,9 +169,9 @@ agentkit/utils/misc.py,sha256=SEAk8e9DGfDm1bZSVbd5AM9EJKyoiIZeLzhg7ddC3w0,3437
 agentkit/utils/request.py,sha256=IVoat3EavR9rQ_fXi0eA2rZPCJ9lVZAXXn7uIIX6bY4,1614
 agentkit/utils/template_utils.py,sha256=Qjg9V6dEpjAd_yYezIIPwuDsDeeMmp6XTMn5ZECifNc,6336
 agentkit/utils/ve_sign.py,sha256=GC5xvcKXNZC-T0j3Klr8TYPItgWcURr7N7lVLPYnldE,8579
-agentkit_sdk_python-0.4.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agentkit_sdk_python-0.4.3.dist-info/METADATA,sha256=RSa1uVC9HcN2SQH6MZf99b218EkrZ9WbNdpei1tDNHY,19384
-agentkit_sdk_python-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agentkit_sdk_python-0.4.3.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
-agentkit_sdk_python-0.4.3.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
-agentkit_sdk_python-0.4.3.dist-info/RECORD,,
+agentkit_sdk_python-0.4.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agentkit_sdk_python-0.4.4.dist-info/METADATA,sha256=tqk9nvlnE4GI3h9HCDT_taOSBkmne7ex4ZzFxfrFD7s,19384
+agentkit_sdk_python-0.4.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+agentkit_sdk_python-0.4.4.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
+agentkit_sdk_python-0.4.4.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
+agentkit_sdk_python-0.4.4.dist-info/RECORD,,

{agentkit_sdk_python-0.4.3.dist-info → agentkit_sdk_python-0.4.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any