agentkernel 0.2.8__tar.gz → 0.2.10__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.2.8
+Version: 0.2.10
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -21,6 +21,10 @@ Requires-Dist: boto3>=1.41.4 ; extra == 'aws'
 Requires-Dist: crewai>=0.150.0 ; extra == 'crewai'
 Requires-Dist: openinference-instrumentation-crewai>=0.1.16 ; extra == 'crewai'
 Requires-Dist: openinference-instrumentation-litellm>=0.1.28 ; extra == 'crewai'
+Requires-Dist: google-auth>=2.0.0 ; extra == 'gmail'
+Requires-Dist: google-auth-oauthlib>=1.0.0 ; extra == 'gmail'
+Requires-Dist: google-auth-httplib2>=0.2.0 ; extra == 'gmail'
+Requires-Dist: google-api-python-client>=2.0.0 ; extra == 'gmail'
 Requires-Dist: httpx>=0.27.0 ; extra == 'instagram'
 Requires-Dist: langfuse>=3.9.2 ; extra == 'langfuse'
 Requires-Dist: nest-asyncio>=1.6.0 ; extra == 'langfuse'
@@ -30,11 +34,12 @@ Requires-Dist: langchain-community~=0.3.27 ; extra == 'langgraph'
 Requires-Dist: litellm~=1.74.3 ; extra == 'langgraph'
 Requires-Dist: fastmcp>=2.12.4 ; extra == 'mcp'
 Requires-Dist: httpx>=0.27.0 ; extra == 'messenger'
-Requires-Dist: openai-agents>=0.2.3 ; extra == 'openai'
-Requires-Dist: openinference-instrumentation-openai-agents>=1.3.0 ; extra == 'openai'
+Requires-Dist: openai-agents>=0.6.3 ; extra == 'openai'
+Requires-Dist: openinference-instrumentation-openai-agents>=1.4.0 ; extra == 'openai'
 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'
@@ -42,6 +47,10 @@ Requires-Dist: pytest-cov>=6.2.1 ; extra == 'test'
 Requires-Dist: pytest-html>=4.1.1 ; extra == 'test'
 Requires-Dist: pytest-order>=1.3.0 ; extra == 'test'
 Requires-Dist: rapidfuzz>=3.14.1 ; extra == 'test'
+Requires-Dist: ragas>=0.4.1 ; extra == 'test'
+Requires-Dist: datasets>=2.14.0 ; extra == 'test'
+Requires-Dist: pandas>=2.0.0 ; extra == 'test'
+Requires-Dist: litellm>=1.74.9 ; extra == 'test'
 Requires-Dist: httpx>=0.27.0 ; extra == 'whatsapp'
 Requires-Python: >=3.12
 Provides-Extra: a2a
@@ -50,6 +59,7 @@ Provides-Extra: api
 Provides-Extra: aws
 Provides-Extra: cli
 Provides-Extra: crewai
+Provides-Extra: gmail
 Provides-Extra: instagram
 Provides-Extra: langfuse
 Provides-Extra: langgraph
@@ -463,6 +473,122 @@ trace:
   type: openllmetry
 ```
 
+#### Test Configuration
+
+Configure test comparison modes for automated testing.
+
+- **Mode**
+  - **Field**: `test.mode`
+  - **Options**: `fuzzy`, `judge`, `fallback`
+  - **Default**: `fallback`
+  - **Description**: Test comparison mode
+  - **Environment Variable**: `AK_TEST__MODE`
+
+- **Judge Model**
+  - **Field**: `test.judge.model`
+  - **Default**: `gpt-4o-mini`
+  - **Description**: LLM model for judge evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__MODEL`
+
+- **Judge Provider**
+  - **Field**: `test.judge.provider`
+  - **Default**: `openai`
+  - **Description**: LLM provider for judge evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__PROVIDER`
+
+- **Judge Embedding Model**
+  - **Field**: `test.judge.embedding_model`
+  - **Default**: `text-embedding-3-small`
+  - **Description**: Embedding model for similarity evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__EMBEDDING_MODEL`
+
+**Test Modes:**
+- `fuzzy`: Uses fuzzy string matching (RapidFuzz)
+- `judge`: Uses LLM-based evaluation (Ragas) for semantic similarity
+- `fallback`: Tries fuzzy first, falls back to judge if fuzzy fails
+
+```yaml
+test:
+  mode: fallback
+  judge:
+    model: gpt-4o-mini
+    provider: openai
+    embedding_model: text-embedding-3-small
+```
+
+#### Messaging Platform Integrations
+
+Configure integrations with messaging platforms.
+
+##### Slack
+
+- **Agent**
+  - **Field**: `slack.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Slack interactions
+  - **Environment Variable**: `AK_SLACK__AGENT`
+
+- **Agent Acknowledgement**
+  - **Field**: `slack.agent_acknowledgement`
+  - **Default**: `""`
+  - **Description**: Acknowledgement message when Slack message is received
+  - **Environment Variable**: `AK_SLACK__AGENT_ACKNOWLEDGEMENT`
+
+##### WhatsApp
+
+- **Agent**
+  - **Field**: `whatsapp.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for WhatsApp interactions
+  - **Environment Variable**: `AK_WHATSAPP__AGENT`
+
+- **Verify Token**, **Access Token**, **App Secret**, **Phone Number ID**, **API Version**
+  - **Environment Variables**: `AK_WHATSAPP__VERIFY_TOKEN`, `AK_WHATSAPP__ACCESS_TOKEN`, `AK_WHATSAPP__APP_SECRET`, `AK_WHATSAPP__PHONE_NUMBER_ID`, `AK_WHATSAPP__API_VERSION`
+
+##### Facebook Messenger
+
+- **Agent**
+  - **Field**: `messenger.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Facebook Messenger interactions
+  - **Environment Variable**: `AK_MESSENGER__AGENT`
+
+- **Verify Token**, **Access Token**, **App Secret**, **API Version**
+  - **Environment Variables**: `AK_MESSENGER__VERIFY_TOKEN`, `AK_MESSENGER__ACCESS_TOKEN`, `AK_MESSENGER__APP_SECRET`, `AK_MESSENGER__API_VERSION`
+
+##### Instagram
+
+- **Agent**
+  - **Field**: `instagram.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Instagram interactions
+  - **Environment Variable**: `AK_INSTAGRAM__AGENT`
+
+- **Instagram Account ID**, **Verify Token**, **Access Token**, **App Secret**, **API Version**
+  - **Environment Variables**: `AK_INSTAGRAM__INSTAGRAM_ACCOUNT_ID`, `AK_INSTAGRAM__VERIFY_TOKEN`, `AK_INSTAGRAM__ACCESS_TOKEN`, `AK_INSTAGRAM__APP_SECRET`, `AK_INSTAGRAM__API_VERSION`
+
+##### Telegram
+
+- **Agent**
+  - **Field**: `telegram.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Telegram interactions
+  - **Environment Variable**: `AK_TELEGRAM__AGENT`
+
+- **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
@@ -487,6 +613,19 @@ export AK_TRACE__TYPE=langfuse  # or openllmetry
 # export LANGFUSE_HOST=https://cloud.langfuse.com
 # For OpenLLMetry:
 # export TRACELOOP_API_KEY=your-api-key
+export AK_TEST__MODE=fallback  # Options: fuzzy, judge, fallback
+export AK_TEST__JUDGE__MODEL=gpt-4o-mini
+export AK_TEST__JUDGE__PROVIDER=openai
+export AK_TEST__JUDGE__EMBEDDING_MODEL=text-embedding-3-small
+# Messaging platforms (optional)
+export AK_SLACK__AGENT=my-agent
+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
@@ -541,6 +680,28 @@ mcp:
 trace:
   enabled: true
   type: langfuse
+test:
+  mode: fallback
+  judge:
+    model: gpt-4o-mini
+    provider: openai
+    embedding_model: text-embedding-3-small
+slack:
+  agent: my-agent
+  agent_acknowledgement: "Processing your request..."
+whatsapp:
+  agent: my-agent
+  agent_acknowledgement: "Processing..."
+messenger:
+  agent: my-agent
+instagram:
+  agent: my-agent
+telegram:
+  agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -578,6 +739,36 @@ trace:
   "trace": {
     "enabled": true,
     "type": "langfuse"
+  },
+  "test": {
+    "mode": "fallback",
+    "judge": {
+      "model": "gpt-4o-mini",
+      "provider": "openai",
+      "embedding_model": "text-embedding-3-small"
+    }
+  },
+  "slack": {
+    "agent": "my-agent",
+    "agent_acknowledgement": "Processing your request..."
+  },
+  "whatsapp": {
+    "agent": "my-agent",
+    "agent_acknowledgement": "Processing..."
+  },
+  "messenger": {
+    "agent": "my-agent"
+  },
+  "instagram": {
+    "agent": "my-agent"
+  },
+  "telegram": {
+    "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

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

@@ -397,6 +397,122 @@ trace:
   type: openllmetry
 ```
 
+#### Test Configuration
+
+Configure test comparison modes for automated testing.
+
+- **Mode**
+  - **Field**: `test.mode`
+  - **Options**: `fuzzy`, `judge`, `fallback`
+  - **Default**: `fallback`
+  - **Description**: Test comparison mode
+  - **Environment Variable**: `AK_TEST__MODE`
+
+- **Judge Model**
+  - **Field**: `test.judge.model`
+  - **Default**: `gpt-4o-mini`
+  - **Description**: LLM model for judge evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__MODEL`
+
+- **Judge Provider**
+  - **Field**: `test.judge.provider`
+  - **Default**: `openai`
+  - **Description**: LLM provider for judge evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__PROVIDER`
+
+- **Judge Embedding Model**
+  - **Field**: `test.judge.embedding_model`
+  - **Default**: `text-embedding-3-small`
+  - **Description**: Embedding model for similarity evaluation
+  - **Environment Variable**: `AK_TEST__JUDGE__EMBEDDING_MODEL`
+
+**Test Modes:**
+- `fuzzy`: Uses fuzzy string matching (RapidFuzz)
+- `judge`: Uses LLM-based evaluation (Ragas) for semantic similarity
+- `fallback`: Tries fuzzy first, falls back to judge if fuzzy fails
+
+```yaml
+test:
+  mode: fallback
+  judge:
+    model: gpt-4o-mini
+    provider: openai
+    embedding_model: text-embedding-3-small
+```
+
+#### Messaging Platform Integrations
+
+Configure integrations with messaging platforms.
+
+##### Slack
+
+- **Agent**
+  - **Field**: `slack.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Slack interactions
+  - **Environment Variable**: `AK_SLACK__AGENT`
+
+- **Agent Acknowledgement**
+  - **Field**: `slack.agent_acknowledgement`
+  - **Default**: `""`
+  - **Description**: Acknowledgement message when Slack message is received
+  - **Environment Variable**: `AK_SLACK__AGENT_ACKNOWLEDGEMENT`
+
+##### WhatsApp
+
+- **Agent**
+  - **Field**: `whatsapp.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for WhatsApp interactions
+  - **Environment Variable**: `AK_WHATSAPP__AGENT`
+
+- **Verify Token**, **Access Token**, **App Secret**, **Phone Number ID**, **API Version**
+  - **Environment Variables**: `AK_WHATSAPP__VERIFY_TOKEN`, `AK_WHATSAPP__ACCESS_TOKEN`, `AK_WHATSAPP__APP_SECRET`, `AK_WHATSAPP__PHONE_NUMBER_ID`, `AK_WHATSAPP__API_VERSION`
+
+##### Facebook Messenger
+
+- **Agent**
+  - **Field**: `messenger.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Facebook Messenger interactions
+  - **Environment Variable**: `AK_MESSENGER__AGENT`
+
+- **Verify Token**, **Access Token**, **App Secret**, **API Version**
+  - **Environment Variables**: `AK_MESSENGER__VERIFY_TOKEN`, `AK_MESSENGER__ACCESS_TOKEN`, `AK_MESSENGER__APP_SECRET`, `AK_MESSENGER__API_VERSION`
+
+##### Instagram
+
+- **Agent**
+  - **Field**: `instagram.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Instagram interactions
+  - **Environment Variable**: `AK_INSTAGRAM__AGENT`
+
+- **Instagram Account ID**, **Verify Token**, **Access Token**, **App Secret**, **API Version**
+  - **Environment Variables**: `AK_INSTAGRAM__INSTAGRAM_ACCOUNT_ID`, `AK_INSTAGRAM__VERIFY_TOKEN`, `AK_INSTAGRAM__ACCESS_TOKEN`, `AK_INSTAGRAM__APP_SECRET`, `AK_INSTAGRAM__API_VERSION`
+
+##### Telegram
+
+- **Agent**
+  - **Field**: `telegram.agent`
+  - **Default**: `""`
+  - **Description**: Default agent for Telegram interactions
+  - **Environment Variable**: `AK_TELEGRAM__AGENT`
+
+- **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
@@ -421,6 +537,19 @@ export AK_TRACE__TYPE=langfuse  # or openllmetry
 # export LANGFUSE_HOST=https://cloud.langfuse.com
 # For OpenLLMetry:
 # export TRACELOOP_API_KEY=your-api-key
+export AK_TEST__MODE=fallback  # Options: fuzzy, judge, fallback
+export AK_TEST__JUDGE__MODEL=gpt-4o-mini
+export AK_TEST__JUDGE__PROVIDER=openai
+export AK_TEST__JUDGE__EMBEDDING_MODEL=text-embedding-3-small
+# Messaging platforms (optional)
+export AK_SLACK__AGENT=my-agent
+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
@@ -475,6 +604,28 @@ mcp:
 trace:
   enabled: true
   type: langfuse
+test:
+  mode: fallback
+  judge:
+    model: gpt-4o-mini
+    provider: openai
+    embedding_model: text-embedding-3-small
+slack:
+  agent: my-agent
+  agent_acknowledgement: "Processing your request..."
+whatsapp:
+  agent: my-agent
+  agent_acknowledgement: "Processing..."
+messenger:
+  agent: my-agent
+instagram:
+  agent: my-agent
+telegram:
+  agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -512,6 +663,36 @@ trace:
   "trace": {
     "enabled": true,
     "type": "langfuse"
+  },
+  "test": {
+    "mode": "fallback",
+    "judge": {
+      "model": "gpt-4o-mini",
+      "provider": "openai",
+      "embedding_model": "text-embedding-3-small"
+    }
+  },
+  "slack": {
+    "agent": "my-agent",
+    "agent_acknowledgement": "Processing your request..."
+  },
+  "whatsapp": {
+    "agent": "my-agent",
+    "agent_acknowledgement": "Processing..."
+  },
+  "messenger": {
+    "agent": "my-agent"
+  },
+  "instagram": {
+    "agent": "my-agent"
+  },
+  "telegram": {
+    "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

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

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.2.8"
+version = "0.2.10"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -49,8 +49,8 @@ redis = [
     "redis>=7.1.0",
 ]
 openai = [
-    "openai-agents>=0.2.3",
-    "openinference-instrumentation-openai-agents>=1.3.0",
+    "openai-agents>=0.6.3",
+    "openinference-instrumentation-openai-agents>=1.4.0",
 ]
 adk = [
     "google-adk>=1.14.1",
@@ -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"
@@ -77,6 +78,12 @@ instagram = [
 telegram = [
     "httpx>=0.27.0"
 ]
+gmail = [
+    "google-auth>=2.0.0",
+    "google-auth-oauthlib>=1.0.0",
+    "google-auth-httplib2>=0.2.0",
+    "google-api-python-client>=2.0.0",
+]
 test = [
     "pytest>=8.4.1",
     "pytest-asyncio>=1.2.0",
@@ -84,6 +91,10 @@ test = [
     "pytest-html>=4.1.1",
     "pytest-order>=1.3.0",
     "rapidfuzz>=3.14.1",
+    "ragas>=0.4.1",
+    "datasets>=2.14.0",
+    "pandas>=2.0.0",
+    "litellm>=1.74.9",
 ]
 a2a = [
     "a2a-sdk[http-server]>=0.3.6",

{agentkernel-0.2.8 → 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.8 → 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.8 → 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.8 → 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.8 → 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):
         """