agentkernel 0.2.9__tar.gz → 0.2.10__tar.gz

{agentkernel-0.2.9 → agentkernel-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.2.9
+Version: 0.2.10
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -39,6 +39,7 @@ Requires-Dist: openinference-instrumentation-openai-agents>=1.4.0 ; extra == 'op
 Requires-Dist: traceloop-sdk>=0.48.0 ; extra == 'openllmetry'
 Requires-Dist: redis>=7.1.0 ; extra == 'redis'
 Requires-Dist: slack-bolt==1.22.0 ; extra == 'slack'
+Requires-Dist: httpx>=0.27.0 ; extra == 'slack'
 Requires-Dist: httpx>=0.27.0 ; extra == 'telegram'
 Requires-Dist: pytest>=8.4.1 ; extra == 'test'
 Requires-Dist: pytest-asyncio>=1.2.0 ; extra == 'test'
@@ -577,6 +578,17 @@ Configure integrations with messaging platforms.
 - **Bot Token**, **Webhook Secret**, **API Version**
   - **Environment Variables**: `AK_TELEGRAM__BOT_TOKEN`, `AK_TELEGRAM__WEBHOOK_SECRET`, `AK_TELEGRAM__API_VERSION`
 
+##### Gmail
+
+- **Agent**
+  - **Field**: `gmail.agent`
+  - **Default**: `"general"`
+  - **Description**: Default agent for Gmail interactions
+  - **Environment Variable**: `AK_GMAIL__AGENT`
+
+- **Client ID**, **Client Secret**, **Token File**, **Poll Interval**, **Label Filter**
+  - **Environment Variables**: `AK_GMAIL__CLIENT_ID`, `AK_GMAIL__CLIENT_SECRET`, `AK_GMAIL__TOKEN_FILE`, `AK_GMAIL__POLL_INTERVAL`, `AK_GMAIL__LABEL_FILTER`
+
 ### Configuration Examples
 
 #### Environment Variables
@@ -611,6 +623,9 @@ export AK_WHATSAPP__AGENT=my-agent
 export AK_MESSENGER__AGENT=my-agent
 export AK_INSTAGRAM__AGENT=my-agent
 export AK_TELEGRAM__AGENT=my-agent
+export AK_GMAIL__AGENT=my-agent
+export AK_GMAIL__CLIENT_ID=your-google-client-id
+export AK_GMAIL__CLIENT_SECRET=your-google-client-secret
 ```
 
 #### .env File
@@ -683,6 +698,10 @@ instagram:
   agent: my-agent
 telegram:
   agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -745,6 +764,11 @@ telegram:
   },
   "telegram": {
     "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

{agentkernel-0.2.9 → agentkernel-0.2.10}/README.md

@@ -502,6 +502,17 @@ Configure integrations with messaging platforms.
 - **Bot Token**, **Webhook Secret**, **API Version**
   - **Environment Variables**: `AK_TELEGRAM__BOT_TOKEN`, `AK_TELEGRAM__WEBHOOK_SECRET`, `AK_TELEGRAM__API_VERSION`
 
+##### Gmail
+
+- **Agent**
+  - **Field**: `gmail.agent`
+  - **Default**: `"general"`
+  - **Description**: Default agent for Gmail interactions
+  - **Environment Variable**: `AK_GMAIL__AGENT`
+
+- **Client ID**, **Client Secret**, **Token File**, **Poll Interval**, **Label Filter**
+  - **Environment Variables**: `AK_GMAIL__CLIENT_ID`, `AK_GMAIL__CLIENT_SECRET`, `AK_GMAIL__TOKEN_FILE`, `AK_GMAIL__POLL_INTERVAL`, `AK_GMAIL__LABEL_FILTER`
+
 ### Configuration Examples
 
 #### Environment Variables
@@ -536,6 +547,9 @@ export AK_WHATSAPP__AGENT=my-agent
 export AK_MESSENGER__AGENT=my-agent
 export AK_INSTAGRAM__AGENT=my-agent
 export AK_TELEGRAM__AGENT=my-agent
+export AK_GMAIL__AGENT=my-agent
+export AK_GMAIL__CLIENT_ID=your-google-client-id
+export AK_GMAIL__CLIENT_SECRET=your-google-client-secret
 ```
 
 #### .env File
@@ -608,6 +622,10 @@ instagram:
   agent: my-agent
 telegram:
   agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -670,6 +688,11 @@ telegram:
   },
   "telegram": {
     "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

{agentkernel-0.2.9 → agentkernel-0.2.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.2.9"
+version = "0.2.10"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -63,7 +63,8 @@ api = [
     "gunicorn>=23.0.0"
 ]
 slack = [
-    "slack-bolt==1.22.0"
+    "slack-bolt==1.22.0",
+    "httpx>=0.27.0"
 ]
 whatsapp = [
     "httpx>=0.27.0"

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/api/a2a/a2a.py

@@ -9,8 +9,9 @@ from a2a.types import AgentCard, InternalError, UnsupportedOperationError
 from a2a.utils import new_agent_text_message
 from a2a.utils.errors import ServerError
 
-from ...core import Agent, AgentService, GlobalRuntime
+from ...core import Agent, AgentService
 from ...core.config import AKConfig
+from ...core.runtime import Runtime
 
 
 class A2A:
@@ -69,7 +70,7 @@ class A2A:
             return
         if not AKConfig.get().a2a.enabled:
             return
-        agents: dict[str, Agent] = GlobalRuntime.instance().agents()
+        agents: dict[str, Agent] = Runtime.current().agents()
         for name, agent in agents.items():
             whitelisted = AKConfig.get().a2a.agents == ["*"] or name in AKConfig.get().a2a.agents
             if not whitelisted:

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/api/handler.py

@@ -17,7 +17,8 @@ from agentkernel.core.model import (
     AgentRequestText,
 )
 
-from ..core import AgentService, GlobalRuntime
+from ..core import AgentService, Config
+from ..core.runtime import Runtime
 
 
 class RESTRequestHandler(ABC):
@@ -37,7 +38,8 @@ class RESTRequestHandler(ABC):
 
         @router.get("/agents")
         def list_agents():
-            return {"agents": list(GlobalRuntime.instance().agents().keys())}
+            from ..core.runtime import Runtime
+            return {"agents": list(Runtime.current().agents().keys())}
 
         """
         pass
@@ -55,6 +57,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
 
     def __init__(self):
         self._log = logging.getLogger("ak.api.agent")
+        self._max_file_size = Config.get().api.max_file_size
 
     class FileData(BaseModel):
         """Represents a file attachment"""
@@ -92,7 +95,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
 
         @router.get("/agents")
         def list_agents():
-            return {"agents": list(GlobalRuntime.instance().agents().keys())}
+            return {"agents": list(Runtime.current().agents().keys())}
 
         @router.post("/run")
         async def run(body: AgentRESTRequestHandler.RunRequest):
@@ -224,6 +227,15 @@ class AgentRESTRequestHandler(RESTRequestHandler):
                     self._log.debug(f"Processing uploaded file: {file.filename}")
                     # Read file content
                     content = await file.read()
+
+                    # Validate file size
+                    file_size = len(content)
+                    if file_size > self._max_file_size:
+                        raise ValueError(
+                            f"File {file.filename} exceeds maximum size of {self._max_file_size / (1024 * 1024):.2f} MB "
+                            f"(size: {file_size / (1024 * 1024):.2f} MB)"
+                        )
+
                     # Encode to base64
                     file_data_base64 = base64.b64encode(content).decode("utf-8")
 
@@ -245,6 +257,15 @@ class AgentRESTRequestHandler(RESTRequestHandler):
                     self._log.debug(f"Processing uploaded image: {image.filename}")
                     # Read image content
                     content = await image.read()
+
+                    # Validate image size
+                    image_size = len(content)
+                    if image_size > self._max_file_size:
+                        raise ValueError(
+                            f"Image {image.filename} exceeds maximum size of {self._max_file_size / (1024 * 1024):.2f} MB "
+                            f"(size: {image_size / (1024 * 1024):.2f} MB)"
+                        )
+
                     # Encode to base64
                     image_data_base64 = base64.b64encode(content).decode("utf-8")
 

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/api/mcp/akmcp.py

@@ -4,8 +4,9 @@ from typing import Any
 from fastmcp import Context, FastMCP
 from fastmcp.server.http import StarletteWithLifespan
 
-from ...core import Agent, AgentService, GlobalRuntime
+from ...core import Agent, AgentService
 from ...core.config import AKConfig
+from ...core.runtime import Runtime
 
 
 class MCP:
@@ -60,7 +61,7 @@ class MCP:
         if cls._fastmcp is None:
             cls._fastmcp = FastMCP("Agent Kernel FastMCP Instance")
         if AKConfig.get().mcp.expose_agents:
-            agents: dict[str, Agent] = GlobalRuntime.instance().agents()
+            agents: dict[str, Agent] = Runtime.current().agents()
             for name, agent in agents.items():
                 whitelisted = AKConfig.get().mcp.agents == ["*"] or name in AKConfig.get().mcp.agents
                 if not whitelisted:

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/core/__init__.py

@@ -24,7 +24,7 @@ from .model import (
 )
 from .config import AKConfig as Config
 from .module import Module
-from .runtime import GlobalRuntime, Runtime
+from .runtime import Runtime, AuxiliaryCache
 from .service import AgentService
-from .hooks import Prehook, Posthook
+from .hooks import PreHook, PostHook
 from .util.key_value_cache import KeyValueCache

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/core/base.py

@@ -3,6 +3,7 @@ import logging
 from abc import ABC, abstractmethod
 from typing import Any, List
 
+from ..core.hooks import PostHook, PreHook
 from .config import AKConfig
 from .model import AgentReply, AgentRequest
 from .util.key_value_cache import KeyValueCache
@@ -197,6 +198,8 @@ class Agent(ABC):
         """
         self._name = name
         self._runner = runner
+        self._pre_hooks = []
+        self._post_hooks = []
 
     @property
     def name(self) -> str:
@@ -212,6 +215,42 @@ class Agent(ABC):
         """
         return self._runner
 
+    @property
+    def pre_hooks(self) -> list[PreHook]:
+        """
+        Returns the list of pre-execution hooks registered for the agent.
+        """
+        return self._pre_hooks
+
+    @property
+    def post_hooks(self) -> list[PostHook]:
+        """
+        Returns the list of post-execution hooks registered for the agent.
+        """
+        return self._post_hooks
+
+    def attach_pre_hooks(self, hooks: list[PreHook]):
+        """
+        Attaches pre-execution hooks to the agent.
+        Duplicate hook objects are ignored to prevent multiple registrations
+        of the same hook.
+        :param hooks: List of pre-execution hooks to attach.
+        """
+        for hook in hooks:
+            if hook not in self._pre_hooks:
+                self._pre_hooks.append(hook)
+
+    def attach_post_hooks(self, hooks: list[PostHook]):
+        """
+        Attaches post-execution hooks to the agent.
+        Duplicate hook objects are ignored to prevent multiple registrations
+        of the same hook.
+        :param hooks: List of post-execution hooks to attach.
+        """
+        for hook in hooks:
+            if hook not in self._post_hooks:
+                self._post_hooks.append(hook)
+
     @staticmethod
     def _generate_a2a_card(agent_name: str, description: str, skills: List):
         """

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/core/config.py

@@ -52,6 +52,7 @@ class _APIConfig(BaseModel):
     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")
+    max_file_size: int = Field(default=2097152, description="Maximum file size in bytes (default: 2 MB)")
 
 
 class _A2AConfig(BaseModel):

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/core/hooks.py

@@ -1,20 +1,25 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .base import Agent, Session
 
-from .base import Agent, Session
 from .model import AgentReply, AgentRequest
 
 """
-Prehook and Posthook classes define the interface for hooks that can be executed before and after an agent's execution respectively.
+PreHook and PostHook classes define the interface for hooks that can be executed before and after an agent's execution respectively.
 These hooks allow for modification of prompts before execution and replies after execution, enabling functionalities such as context injection, validation, moderation, logging, and analytics. 
 
 Currently, they will get only called for the initial execution of an agent when a user prompt is provided. It's unable to hook into agent-to-agent calls within a workflow. This will be a future enhancement.
 """
 
 
-class Prehook(ABC):
+class PreHook(ABC):
     @abstractmethod
     async def on_run(
-        self, session: Session, agent: Agent, requests: list[AgentRequest]
+        self, session: "Session", agent: "Agent", requests: list[AgentRequest]
     ) -> list[AgentRequest] | AgentReply:
         """
         Hook method called before an agent starts executing a prompt. These hooks can modify the prompt or halt execution.
@@ -42,10 +47,10 @@ class Prehook(ABC):
         raise NotImplementedError
 
 
-class Posthook(ABC):
+class PostHook(ABC):
     @abstractmethod
     async def on_run(
-        self, session: Session, requests: list[AgentRequest], agent: Agent, agent_reply: AgentReply
+        self, session: "Session", requests: list[AgentRequest], agent: "Agent", agent_reply: AgentReply
     ) -> AgentReply:
         """
         Hook method called after an agent finishes executing a prompt. These hooks can modify the agent's reply. Some use cases:

{agentkernel-0.2.9 → agentkernel-0.2.10}/src/agentkernel/core/module.py

@@ -1,8 +1,9 @@
 from abc import ABC, abstractmethod
 from typing import Any, List
 
+from ..core.hooks import PostHook, PreHook
 from .base import Agent
-from .runtime import ModuleLoader
+from .runtime import Runtime
 
 
 class Module(ABC):
@@ -32,9 +33,20 @@ class Module(ABC):
         Unloads and deregisters all agents in the module
         """
         for agent in self._agents:
-            ModuleLoader.runtime().deregister(agent)
+            Runtime.current().deregister(agent)
         self._agents.clear()
 
+    def get_agent(self, name: str) -> Agent | None:
+        """
+        Returns an agent by name from the module.
+        :param name: The name of the agent to retrieve.
+        :return: The agent with the specified name, or None if not found.
+        """
+        for agent in self._agents:
+            if agent.name == name:
+                return agent
+        return None
+
     @abstractmethod
     def _wrap(self, agent: Any, agents: List[Any]) -> Agent:
         """
@@ -45,7 +57,7 @@ class Module(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def load(self, agents: list[Any]):
+    def load(self, agents: list[Any]) -> "Module":
         """
         Loads and registers all agents to the runtime by replacing the current agents.
         :param agents: List of agents to load.
@@ -55,10 +67,29 @@ class Module(ABC):
         for agent in agents:
             try:
                 wrapped = self._wrap(agent, agents)
-                ModuleLoader.runtime().register(wrapped)
+                Runtime.current().register(wrapped)
                 registered.append(wrapped)
             except Exception:
                 self._agents = registered
                 self.unload()
                 raise
         self._agents = registered
+        return self
+
+    @abstractmethod
+    def pre_hook(self, agent: Any, hooks: list[PreHook]) -> "Module":
+        """
+        Attaches pre-execution hooks to the agent.
+        :param agent: The agent to attach hooks to.
+        :param hooks: List of pre-execution hooks to attach.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def post_hook(self, agent: Any, hooks: list[PostHook]) -> "Module":
+        """
+        Attaches post-execution hooks to the agent.
+        :param agent: The agent to attach hooks to.
+        :param hooks: List of post-execution hooks to attach.
+        """
+        raise NotImplementedError