agentkernel 0.2.0__tar.gz → 0.2.1__tar.gz

{agentkernel-0.2.0 → agentkernel-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.2.0
+Version: 0.2.1
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <info@yaalalabs.com>
@@ -317,6 +317,11 @@ Configure the REST API server (if using the API module).
   - **Default**: `8000`
   - **Environment Variable**: `AK_API_PORT`
 
+- **Custom Router Prefix**
+  - **Field**: `api.custom_router_prefix`
+  - **Default**: `/custom`
+  - **Environment Variable**: `AK_API_CUSTOM_ROUTER_PREFIX`
+
 - **Enabled Routes**
   - **Field**: `api.enabled_routes.agents`
   - **Default**: `true`

{agentkernel-0.2.0 → agentkernel-0.2.1}/README.md

@@ -274,6 +274,11 @@ Configure the REST API server (if using the API module).
   - **Default**: `8000`
   - **Environment Variable**: `AK_API_PORT`
 
+- **Custom Router Prefix**
+  - **Field**: `api.custom_router_prefix`
+  - **Default**: `/custom`
+  - **Environment Variable**: `AK_API_CUSTOM_ROUTER_PREFIX`
+
 - **Enabled Routes**
   - **Field**: `api.enabled_routes.agents`
   - **Default**: `true`

{agentkernel-0.2.0 → agentkernel-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.2.0"
+version = "0.2.1"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
 requires-python = ">=3.12"

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/adk/adk.py

@@ -6,7 +6,7 @@ from google.adk.runners import Runner
 from google.adk.sessions import InMemorySessionService, Session as ADKSession
 from google.genai import types
 
-from agentkernel.core import Agent as AKBaseAgent, Module, Runner as BaseRunner, Session
+from ..core import Agent as AKBaseAgent, Module, Runner as BaseRunner, Session, Runtime
 
 FRAMEWORK = "adk"
 
@@ -161,5 +161,14 @@ class GoogleADKModule(Module):
         Initializes a Google ADK Module instance.
         :param agents: List of agents in the module.
         """
-        runner = GoogleADKRunner()
-        super().__init__(list(map(lambda agent: GoogleADKAgent(agent.name, runner, agent), agents)))
+        self.runner = GoogleADKRunner()
+        super().__init__(list(map(lambda agent: GoogleADKAgent(agent.name, self.runner, agent), agents)))
+
+    def add(self, agent: BaseAgent):
+        """
+        Adds an agent to the module.
+        :param agent: The agent to add.
+        """
+        ak_agent = GoogleADKAgent(agent.name, self.runner, agent)
+        super().add(ak_agent)
+        Runtime.instance().register(ak_agent)

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/api/restapi.py

@@ -1,7 +1,7 @@
 import logging
 
 import uvicorn
-from fastapi import FastAPI
+from fastapi import FastAPI, APIRouter
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.utils import get_openapi
 
@@ -21,6 +21,7 @@ class RESTAPI:
     Can run any FastAPI app instance or assemble one from routers.
     """
     _log = logging.getLogger("ak.api.restapi")
+    _custom_routers = []
 
     @classmethod
     def _create_app(cls, routers, lifespan=None) -> FastAPI:
@@ -52,6 +53,13 @@ class RESTAPI:
 
         return app
 
+    @classmethod
+    def add(cls, router: APIRouter):
+        cls._log.debug(f"Adding custom router")
+        for route in router.routes:
+            cls._log.debug(f"Route: {route.path} [{route.methods}]")
+        cls._custom_routers.append(router)
+
     @classmethod
     def run(cls):
         """
@@ -75,4 +83,7 @@ class RESTAPI:
             app.mount("/mcp", mcp_app)
         else:
             app = cls._create_app(routers=routers)
+        # Add custom routers
+        for router in cls._custom_routers:
+            app.include_router(router, prefix=AKConfig.get().api.custom_router_prefix)
         uvicorn.run(app=app, host=host, port=port, reload=False)

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/core/config.py

@@ -35,6 +35,7 @@ class _APIConfig(BaseModel):
     host: str = Field(default="0.0.0.0", description="API host")
     port: int = Field(default=8000, description="API port")
     enabled_routes: _RoutesConfig = Field(description="API route flags", default_factory=_RoutesConfig)
+    custom_router_prefix: str = Field(default="/custom", description="Custom router prefix")
 
 
 class _A2AConfig(BaseModel):

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/core/module.py

@@ -1,14 +1,16 @@
+from abc import ABC, abstractmethod
+
 from .base import Agent
 from .runtime import Runtime
 
 
-class Module:
+class Module(ABC):
     """
     Module is the base class for all agent modules.
-    
+
     An agent module is a Python module containing a set of agents built using a supported agent
     framework. Agent Kernel provides an implementation of the Module class for each supported agent
-    framework, allowing the agents to be registered and managed in a framework agnostic manner.
+    framework, allowing the agents to be registered and managed in a framework-agnostic manner.
     """
 
     def __init__(self, agents: list[Agent]):
@@ -26,3 +28,11 @@ class Module:
         Returns the list of agents in the module.
         """
         return self._agents
+
+    @abstractmethod
+    def add(self, agent: Agent):
+        """
+        Adds an agent to the module.
+        :param agent: The agent to add.
+        """
+        self._agents.append(agent)

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/crewai/crewai.py

@@ -5,7 +5,7 @@ from crewai import Agent, Crew, Task
 from crewai.memory.external.external_memory import ExternalMemory
 from crewai.memory.storage.interface import Storage
 
-from ..core import Agent as BaseAgent, Module, Runner, Session
+from ..core import Agent as BaseAgent, Module, Runner, Session, Runtime
 
 FRAMEWORK = "crewai"
 
@@ -171,6 +171,17 @@ class CrewAIModule(Module):
         Initializes a CrewAIModule instance.
         :param agents: List of agents in the module.
         """
-        runner = CrewAIRunner()
+        self.runner = CrewAIRunner()
+        self.crew = agents
         super().__init__(
-            list(map(lambda agent: CrewAIAgent(agent.role, runner, agent, agents), agents)))
+            list(map(lambda agent: CrewAIAgent(agent.role, self.runner, agent, self.crew), self.crew)))
+
+    def add(self, agent: Agent):
+        """
+        Adds an agent to the module.
+        :param agent: The agent to add.
+        """
+        ak_agent = CrewAIAgent(agent.role, self.runner, agent, self.crew)
+        super().add(ak_agent)
+        self.crew.append(agent)
+        Runtime.instance().register(ak_agent)

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/langgraph/langgraph.py

@@ -6,7 +6,7 @@ from langgraph.checkpoint.base import BaseCheckpointSaver, Checkpoint, Checkpoin
 from langgraph.graph.state import CompiledStateGraph
 from pydantic import BaseModel
 
-from ..core import Agent as BaseAgent, Module as BaseModule, Runner as BaseRunner, Session as BaseSession
+from ..core import Agent as BaseAgent, Module as BaseModule, Runner as BaseRunner, Session as BaseSession, Runtime
 
 FRAMEWORK = "langgraph"
 
@@ -294,5 +294,15 @@ class LangGraphModule(BaseModule):
         Initializes a LangGraphModule instance.
         :param agents: List of agents in the module.
         """
-        runner = LangGraphRunner()
-        super().__init__(list(map(lambda agent: LangGraphAgent(name=agent.name, runner=runner, agent=agent), agents)))
+        self.runner = LangGraphRunner()
+        super().__init__(
+            list(map(lambda agent: LangGraphAgent(name=agent.name, runner=self.runner, agent=agent), agents)))
+
+    def add(self, agent: CompiledStateGraph):
+        """
+        Adds an agent to the module.
+        :param agent: The agent to add.
+        """
+        ak_agent = LangGraphAgent(name=agent.name, runner=self.runner, agent=agent)
+        super().add(ak_agent)
+        Runtime.instance().register(ak_agent)

{agentkernel-0.2.0 → agentkernel-0.2.1}/src/agentkernel/openai/openai.py

@@ -3,7 +3,7 @@ from typing import Any, List
 from agents import Agent, Runner
 from agents.memory.session import SessionABC
 
-from ..core import Agent as BaseAgent, Module, Runner as BaseRunner, Session
+from ..core import Agent as BaseAgent, Module, Runner as BaseRunner, Session, Runtime
 
 FRAMEWORK = "openai"
 
@@ -136,7 +136,7 @@ class OpenAIAgent(BaseAgent):
 
 class OpenAIModule(Module):
     """
-    OpenAIModule class provides a module for OpenAI Agent SDK based agents.
+    OpenAIModule class provides a module for OpenAI Agents SDK based agents.
     """
 
     def __init__(self, agents: list[Agent]):
@@ -144,5 +144,14 @@ class OpenAIModule(Module):
         Initializes an OpenAIModule instance.
         :param agents: List of agents in the module.
         """
-        runner = OpenAIRunner()
-        super().__init__(list(map(lambda agent: OpenAIAgent(agent.name, runner, agent), agents)))
+        self.runner = OpenAIRunner()
+        super().__init__(list(map(lambda agent: OpenAIAgent(agent.name, self.runner, agent), agents)))
+
+    def add(self, agent: Agent):
+        """
+        Adds an agent to the module.
+        :param agent: The agent to add.
+        """
+        ak_agent = OpenAIAgent(agent.name, self.runner, agent)
+        super().add(ak_agent)
+        Runtime.instance().register(ak_agent)