agentkernel 0.3.2__tar.gz → 0.4.0__tar.gz

{agentkernel-0.3.2 → agentkernel-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.3.2
+Version: 0.4.0
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -22,6 +22,7 @@ Requires-Dist: boto3>=1.41.4 ; extra == 'aws'
 Requires-Dist: azure-functions>=1.24.0 ; extra == 'azure'
 Requires-Dist: azure-data-tables>=12.4.0 ; extra == 'azure'
 Requires-Dist: azure-core>=1.26.0 ; extra == 'azure'
+Requires-Dist: chromadb>=0.4.0 ; extra == 'chromadb'
 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'
@@ -30,7 +31,7 @@ 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: langfuse>=4.2.0 ; extra == 'langfuse'
 Requires-Dist: nest-asyncio>=1.6.0 ; extra == 'langfuse'
 Requires-Dist: langgraph~=1.0.5 ; extra == 'langgraph'
 Requires-Dist: langchain~=1.2.3 ; extra == 'langgraph'
@@ -39,6 +40,7 @@ Requires-Dist: litellm~=1.74.9 ; extra == 'langgraph'
 Requires-Dist: fastmcp>=2.12.4 ; extra == 'mcp'
 Requires-Dist: httpx>=0.27.0 ; extra == 'messenger'
 Requires-Dist: litellm>=1.74.9 ; extra == 'multimodal'
+Requires-Dist: neo4j>=5.0.0 ; extra == 'neo4j'
 Requires-Dist: openai-agents>=0.6.5 ; extra == 'openai'
 Requires-Dist: openinference-instrumentation-openai-agents>=1.4.0 ; extra == 'openai'
 Requires-Dist: openai-guardrails>=0.2.1 ; extra == 'openai'
@@ -46,6 +48,10 @@ 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: botbuilder-core>=4.17.0 ; extra == 'teams'
+Requires-Dist: botbuilder-schema>=4.17.0 ; extra == 'teams'
+Requires-Dist: msal>=1.31.1 ; extra == 'teams'
+Requires-Dist: httpx>=0.27.0 ; extra == 'teams'
 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'
@@ -57,6 +63,7 @@ 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: trino>=0.337.0 ; extra == 'trino'
 Requires-Dist: walledai>=4.9.3 ; extra == 'walledai'
 Requires-Dist: httpx>=0.27.0 ; extra == 'whatsapp'
 Requires-Python: >=3.12, <3.14
@@ -66,6 +73,7 @@ Provides-Extra: api
 Provides-Extra: auth
 Provides-Extra: aws
 Provides-Extra: azure
+Provides-Extra: chromadb
 Provides-Extra: cli
 Provides-Extra: crewai
 Provides-Extra: gmail
@@ -75,12 +83,15 @@ Provides-Extra: langgraph
 Provides-Extra: mcp
 Provides-Extra: messenger
 Provides-Extra: multimodal
+Provides-Extra: neo4j
 Provides-Extra: openai
 Provides-Extra: openllmetry
 Provides-Extra: redis
 Provides-Extra: slack
+Provides-Extra: teams
 Provides-Extra: telegram
 Provides-Extra: test
+Provides-Extra: trino
 Provides-Extra: walledai
 Provides-Extra: whatsapp
 Description-Content-Type: text/markdown
@@ -100,6 +111,7 @@ Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer
 - **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK
 - **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate) or Azure (Functions, Container Apps) with the same code
 - **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB)
+- **Knowledge Bases**: Unified `KnowledgeBase` interface with ChromaDB, Neo4j, and Starburst/Trino backends via `KnowledgeBuilder`
 - **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions), containerized (AWS ECS, Azure Container Apps)
 - **Pluggable Architecture**: Easy to extend with custom framework adapters and cloud providers
 - **MCP Server**: Built-in Model Context Protocol server for exposing agents as MCP tools and exposing any custom tool
@@ -113,6 +125,14 @@ Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer
 pip install agentkernel
 ```
 
+Install optional knowledge base extras as needed:
+
+```bash
+pip install "agentkernel[chromadb]"
+pip install "agentkernel[neo4j]"
+pip install "agentkernel[trino]"
+```
+
 **Requirements:**
 - Python 3.12+
 
@@ -364,13 +384,19 @@ Supported formats: `.yaml`, `.yml`, `.json`
 
 ### Configuration Options
 
-#### Debug Mode
+#### Logging Configuration
 
-- **Field**: `debug`
-- **Type**: boolean
-- **Default**: `false`
-- **Description**: Enable debug mode across the library
-- **Environment Variable**: `AK_DEBUG`
+- **Field**: `logging.ak.level`
+- **Type**: string
+- **Default**: `WARNING`
+- **Description**: Agent Kernel logger level (INFO, DEBUG, ERROR, WARNING, CRITICAL)
+- **Environment Variable**: `AK_LOGGING__AK__LEVEL`
+
+- **Field**: `logging.system.level`
+- **Type**: string
+- **Default**: `WARNING`
+- **Description**: System/root logger level (INFO, DEBUG, ERROR, WARNING, CRITICAL)
+- **Environment Variable**: `AK_LOGGING__SYSTEM__LEVEL`
 
 #### Session Store
 
@@ -956,7 +982,6 @@ AK_TRACE__TYPE=langfuse  # or openllmetry
 #### config.yaml
 
 ```yaml
-debug: false
 session:
   type: redis
   redis:

{agentkernel-0.3.2 → agentkernel-0.4.0}/README.md

@@ -13,6 +13,7 @@ Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer
 - **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK
 - **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate) or Azure (Functions, Container Apps) with the same code
 - **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB)
+- **Knowledge Bases**: Unified `KnowledgeBase` interface with ChromaDB, Neo4j, and Starburst/Trino backends via `KnowledgeBuilder`
 - **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions), containerized (AWS ECS, Azure Container Apps)
 - **Pluggable Architecture**: Easy to extend with custom framework adapters and cloud providers
 - **MCP Server**: Built-in Model Context Protocol server for exposing agents as MCP tools and exposing any custom tool
@@ -26,6 +27,14 @@ Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer
 pip install agentkernel
 ```
 
+Install optional knowledge base extras as needed:
+
+```bash
+pip install "agentkernel[chromadb]"
+pip install "agentkernel[neo4j]"
+pip install "agentkernel[trino]"
+```
+
 **Requirements:**
 - Python 3.12+
 
@@ -277,13 +286,19 @@ Supported formats: `.yaml`, `.yml`, `.json`
 
 ### Configuration Options
 
-#### Debug Mode
+#### Logging Configuration
 
-- **Field**: `debug`
-- **Type**: boolean
-- **Default**: `false`
-- **Description**: Enable debug mode across the library
-- **Environment Variable**: `AK_DEBUG`
+- **Field**: `logging.ak.level`
+- **Type**: string
+- **Default**: `WARNING`
+- **Description**: Agent Kernel logger level (INFO, DEBUG, ERROR, WARNING, CRITICAL)
+- **Environment Variable**: `AK_LOGGING__AK__LEVEL`
+
+- **Field**: `logging.system.level`
+- **Type**: string
+- **Default**: `WARNING`
+- **Description**: System/root logger level (INFO, DEBUG, ERROR, WARNING, CRITICAL)
+- **Environment Variable**: `AK_LOGGING__SYSTEM__LEVEL`
 
 #### Session Store
 
@@ -869,7 +884,6 @@ AK_TRACE__TYPE=langfuse  # or openllmetry
 #### config.yaml
 
 ```yaml
-debug: false
 session:
   type: redis
   redis:

{agentkernel-0.3.2 → agentkernel-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.3.2"
+version = "0.4.0"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
 requires-python = ">=3.12,<3.14"
@@ -25,7 +25,7 @@ auth = [
     "PyJWT>=2.0.0",
 ]
 langfuse = [
-    "langfuse>=3.9.2",
+    "langfuse>=4.2.0",
     "nest-asyncio>=1.6.0",
 ]
 openllmetry = [
@@ -57,6 +57,17 @@ azure = [
 redis = [
     "redis>=7.1.0",
 ]
+
+neo4j = [
+    "neo4j>=5.0.0",
+]
+chromadb = [
+    "chromadb>=0.4.0",
+]
+trino = [
+    "trino>=0.337.0",
+]
+
 openai = [
     "openai-agents>=0.6.5",
     "openinference-instrumentation-openai-agents>=1.4.0",
@@ -91,6 +102,12 @@ instagram = [
 telegram = [
     "httpx>=0.27.0"
 ]
+teams = [
+    "botbuilder-core>=4.17.0",
+    "botbuilder-schema>=4.17.0",
+    "msal>=1.31.1",
+    "httpx>=0.27.0",
+]
 gmail = [
     "google-auth>=2.0.0",
     "google-auth-oauthlib>=1.0.0",
@@ -150,4 +167,3 @@ target-version = ["py312"]
 
 [tool.pytest.ini_options]
 addopts = "--cov=src --cov-report=term --cov-report=html --html=report.html"
-

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/__init__.py

@@ -19,3 +19,8 @@ except importlib.metadata.PackageNotFoundError:
     __version__ = "0.1.0"
 
 from .core import *
+
+# Auto-configure logging on import if not already configured (handled inside the "configure_from_config()" function)
+from .core.logger import AKLogger
+
+AKLogger.configure_from_config()

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/api/a2a/a2a.py

@@ -31,7 +31,7 @@ class A2A:
     """
     Card built flag
     """
-    _log = logging.getLogger(__name__)
+    _log = logging.getLogger("ak.a2a")
 
     class Executor(AgentExecutor):
 

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/api/http.py

@@ -9,12 +9,6 @@ from ..auth import AuthValidator, ValidationContext
 from ..core.config import AKConfig
 from .handler import AgentRESTRequestHandler, RESTRequestHandler
 
-logging.basicConfig(
-    level=logging.DEBUG,
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-    force=True,
-)
-
 
 class RESTAPI:
     """

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/cli/cli.py

@@ -3,17 +3,14 @@ import logging
 import readline  # Enables line editing and history features for input() in the CLI
 
 from ..core import AgentService
+from ..core.config import AKConfig
 
-# Configure logger only to print agent kernel logs
-ak_logger = logging.getLogger("ak")
-ak_logger.setLevel(logging.INFO)
-ak_logger.propagate = False
+ak_cli_logger = logging.getLogger("ak.cli")
 
-if not ak_logger.handlers:
+if not ak_cli_logger.handlers:
     handler = logging.StreamHandler()
-    handler.setLevel(logging.INFO)
     handler.setFormatter(logging.Formatter("\033[36m(kernel) >> %(message)s\033[0m"))
-    ak_logger.addHandler(handler)
+    ak_cli_logger.addHandler(handler)
 
 
 class CLI:
@@ -101,7 +98,7 @@ class CLI:
                 raise
             except Exception as e:
                 self._print(f"Error: {e}")
-                ak_logger.error("Exception in CLI run loop", exc_info=True)
+                ak_cli_logger.error("Exception in CLI run loop", exc_info=True)
 
     @classmethod
     def main(cls):

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/config.py

@@ -238,6 +238,19 @@ class _QueuesConfig(BaseModel):
     output: _OutputQueueConfig = Field(default_factory=_OutputQueueConfig, description="Output SQS queue configuration for async execution mode")
 
 
+class _LogLevelConfig(BaseModel):
+    level: Optional[str] = Field(
+        default=None,
+        pattern="^(INFO|DEBUG|ERROR|WARNING|CRITICAL)$",
+        description="Log level. Options: INFO, DEBUG, ERROR, WARNING, CRITICAL",
+    )
+
+
+class _LoggingConfig(BaseModel):
+    ak: _LogLevelConfig = Field(description="Agent Kernel logging configuration", default_factory=_LogLevelConfig)
+    system: _LogLevelConfig = Field(description="System logging configuration", default_factory=_LogLevelConfig)
+
+
 class _ExecutionConfig(BaseModel):
     mode: Optional[ExecutionMode] = Field(
         default=None,
@@ -252,7 +265,6 @@ class _ExecutionConfig(BaseModel):
 
 
 class AKConfig(YamlBaseSettingsModified):
-    debug: bool = Field(default=False, description="Enable debug mode")
     session: _SessionStoreConfig = Field(
         description="Agent session / memory related configurations",
         default_factory=_SessionStoreConfig,
@@ -275,6 +287,7 @@ class AKConfig(YamlBaseSettingsModified):
     test: _TestConfig = Field(description="Test related configurations", default_factory=_TestConfig)
     guardrail: _GuardrailConfig = Field(description="Guardrail related configurations", default_factory=_GuardrailConfig)
     execution: _ExecutionConfig = Field(description="Execution mode and queue related configurations", default_factory=_ExecutionConfig)
+    logging: _LoggingConfig = Field(description="Logging related configurations", default_factory=_LoggingConfig)
     library_version: str = Field(default=_get_ak_version(), description="Library version")
 
     @classmethod

agentkernel-0.4.0/src/agentkernel/core/logger.py

@@ -0,0 +1,99 @@
+import logging
+
+from .config import AKConfig
+
+
+class AKLogger:
+    """Agent Kernel logger configuration utility."""
+
+    _initialized = False
+
+    @staticmethod
+    def resolve_level(level: str | int) -> int:
+        """Resolve log level from string or int.
+
+        :param level: Log level as string or int.
+        :return: Resolved log level as int.
+        """
+        if isinstance(level, int):
+            return level
+
+        level_name = str(level).upper()
+
+        get_mapping = getattr(logging, "getLevelNamesMapping", None)
+        if callable(get_mapping):
+            return get_mapping().get(level_name, logging.WARNING)
+
+        return logging.WARNING
+
+    @staticmethod
+    def attach_stream_handler(logger: logging.Logger, attach_formatter=False) -> None:
+        """Attach a stream handler to the logger.
+
+        :param logger: Logger instance to attach handler to.
+        :param attach_formatter: Whether to attach a default formatter.
+        :return: None.
+        """
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+            handler.close()
+        handler = logging.StreamHandler()
+        if attach_formatter:
+            handler.setFormatter(logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s"))
+        logger.addHandler(handler)
+
+    @staticmethod
+    def set_handlers_level(logger: logging.Logger, level: int) -> None:
+        """Set the level for all handlers on a logger.
+
+        :param logger: Logger instance to set handler levels for.
+        :param level: Log level to set.
+        :return: None.
+        """
+        for handler in logger.handlers:
+            handler.setLevel(level)
+
+    @staticmethod
+    def set_ak_log_level(level: str):
+        """Set the log level for the AK logger.
+
+        :param level: Log level as string.
+        :return: None.
+        """
+        resolved_level = AKLogger.resolve_level(level)
+        ak_logger = logging.getLogger("ak")
+        ak_logger.setLevel(resolved_level)
+        ak_logger.propagate = False  # stopping the propagation here so that it won't propagate to the root, so loggers like "ak.api" with propagate=True (propagate is True by default, which is the case here) won't propagate to the root, propagation will stop at this level ("ak" level)
+        AKLogger.attach_stream_handler(logger=ak_logger, attach_formatter=True)
+        AKLogger.set_handlers_level(logger=ak_logger, level=resolved_level)
+
+    @staticmethod
+    def set_system_log_level(level: str):
+        """Set the log level for the system root logger.
+
+        :param level: Log level as string.
+        :return: None.
+        """
+        resolved_level = AKLogger.resolve_level(level)
+        root_logger = logging.getLogger()
+        root_logger.setLevel(resolved_level)
+        root_logger.propagate = False  # the root logger has nowhere to propagate to because it is the root
+        AKLogger.attach_stream_handler(logger=root_logger, attach_formatter=True)
+        AKLogger.set_handlers_level(logger=root_logger, level=resolved_level)
+
+    @classmethod
+    def configure_from_config(cls):
+        """Configure loggers from AKConfig settings.
+
+        :return: None.
+        """
+        if cls._initialized:
+            return
+        config = AKConfig.get()
+        ak_level = config.logging.ak.level
+        system_level = config.logging.system.level
+        if system_level:
+            cls.set_system_log_level(system_level)
+        if ak_level:
+            cls.set_ak_log_level(ak_level)
+        cls._initialized = True

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/storage/dynamodb.py

@@ -24,7 +24,7 @@ class DynamoDBAttachmentDriver:
     raw DynamoDB attachment operations.
     """
 
-    _log = logging.getLogger("ak.core.multimodal.storage.dynamodb.driver")
+    _log = logging.getLogger("ak.multimodal.storage.dynamodb.driver")
 
     def __init__(self, table_name: str, ttl: int):
         """

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/storage/in_memory.py

@@ -16,7 +16,7 @@ class InMemoryAttachmentStore(AttachmentStore):
     # Shared across all instances — survives across multiple instantiations per request
     _attachments: ClassVar[dict[str, dict]] = {}  # "session_id:attachment_id" -> attachment
     _index: ClassVar[dict[str, list[str]]] = {}  # session_id -> [attachment_ids]
-    _log = logging.getLogger("ak.core.multimodal.storage.inmemory")
+    _log = logging.getLogger("ak.multimodal.storage.inmemory")
 
     def __init__(self, session_id: str):
         """

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/storage/redis.py

@@ -19,7 +19,7 @@ class RedisAttachmentDriver:
     raw Redis attachment operations.
     """
 
-    _log = logging.getLogger("ak.core.multimodal.storage.redis.driver")
+    _log = logging.getLogger("ak.multimodal.storage.redis.driver")
     _redis_client = None
 
     def __init__(self, url: str, ttl: int, prefix: str):

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/storage/session_cache.py

@@ -23,7 +23,7 @@ class SessionNonVolatileCacheAttachmentStore(AttachmentStore):
     simple but causes session size to grow with each attachment.
     """
 
-    _log = logging.getLogger("ak.core.multimodal.storage.session_cache")
+    _log = logging.getLogger("ak.multimodal.storage.session_cache")
 
     def __init__(self, session_id: str):
         """

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/storage/storage_manager.py

@@ -13,7 +13,7 @@ from .base import (
     AttachmentStore,
 )
 
-_log = logging.getLogger("ak.core.multimodal.storage")
+_log = logging.getLogger("ak.multimodal.storage")
 
 
 class AttachmentStorageManager:

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/core/multimodal/tools.py

@@ -11,7 +11,7 @@ from ..model import SystemTool
 from ..tool import ToolContext
 from .storage import AttachmentStorageManager
 
-_log = logging.getLogger(__name__)
+_log = logging.getLogger("ak.multimodal.tools")
 
 
 def _analyze_attachments(attachment_ids: list[str], prompt: str) -> str:

agentkernel-0.4.0/src/agentkernel/core/util/error_util.py

@@ -0,0 +1,77 @@
+class ErrorCategory:
+    RATE_LIMIT = "rate_limit"
+    SERVER = "server"
+    CONNECTION = "connection"
+    AUTH = "auth"
+    NOT_FOUND = "not_found"
+    UNKNOWN = "unknown"
+
+
+class ErrorUtil:
+    @staticmethod
+    def _get_status_code(error: Exception) -> int | None:
+        """Extracts HTTP status code from error attributes if they exist."""
+        for attr in ["status_code", "code", "status"]:
+            val = getattr(error, attr, None)
+            if isinstance(val, int):
+                return val
+            if isinstance(val, str) and val.isdigit():
+                return int(val)
+        return None
+
+    @staticmethod
+    def _classify(error: Exception, status_code: int | None) -> str:
+        """Categorizes the error based on code, name, or message content."""
+        error_name = error.__class__.__name__.lower()
+        message = str(error).lower()
+
+        if status_code == 429:
+            return ErrorCategory.RATE_LIMIT
+
+        if status_code in [500, 502, 503, 504]:
+            return ErrorCategory.SERVER
+
+        if "timeout" in error_name or "connection" in error_name:
+            return ErrorCategory.CONNECTION
+
+        if "auth" in error_name or "unauthorized" in error_name:
+            return ErrorCategory.AUTH
+
+        if any(k in error_name for k in ["notfound", "invalid", "badrequest"]):
+            return ErrorCategory.NOT_FOUND
+
+        if any(t in message for t in ["overloaded", "temporarily unavailable", "high demand"]):
+            return ErrorCategory.SERVER
+
+        return ErrorCategory.UNKNOWN
+
+    @staticmethod
+    def _format_error_message(error: Exception) -> str:
+        """Returns a clean string for the UI based on the error category."""
+        status_code = ErrorUtil._get_status_code(error)
+        category = ErrorUtil._classify(error, status_code)
+
+        if category == ErrorCategory.RATE_LIMIT:
+            return "Error: Too many requests. Please try again later."
+
+        if category == ErrorCategory.SERVER:
+            return "Error: The service is temporarily unavailable. Please try again."
+
+        if category == ErrorCategory.CONNECTION:
+            return "Error: Could not connect to the model provider. Please check your internet."
+
+        if category == ErrorCategory.AUTH:
+            return "Error: Invalid API configuration or credentials."
+
+        if category == ErrorCategory.NOT_FOUND:
+            return "Error: Invalid model or resource not found. Please check your configuration."
+
+        message = " ".join(str(error).split()).strip()
+        if message:
+            return f"Error: {message}"
+
+        return f"An unexpected error occurred: {error.__class__.__name__}"
+
+
+def user_facing_error_message(error: Exception) -> str:
+    return ErrorUtil._format_error_message(error)

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/aws/core/response_store/handler.py

@@ -53,7 +53,7 @@ class ResponseDBHandler:
         response_store_type: ResponseDBHandler.Type = ResponseDBHandler.Type.from_str(response_store_config.type)
 
         # Check for Redis configuration
-        if response_store_type == ResponseDBHandler.Type.REDIS:
+        if response_store_type == ResponseDBHandler.Type.REDIS and response_store_config.redis is not None:
             from .redis import RedisResponseStore
 
             redis_config = response_store_config.redis
@@ -65,7 +65,7 @@ class ResponseDBHandler:
             )
 
         # Check for DynamoDB configuration
-        elif response_store_type == ResponseDBHandler.Type.DYNAMODB:
+        elif response_store_type == ResponseDBHandler.Type.DYNAMODB and response_store_config.dynamodb is not None:
             from .dynamodb import DynamoDBResponseStore
 
             dynamodb_config = response_store_config.dynamodb

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/aws/serverless/akagentrunner.py

@@ -87,15 +87,6 @@ class ServerlessAgentRunner(LambdaSQSConsumer):
         """
         cls._log.info(f"Processing message: {record}")
         body = cls._parse_body(record)
-        session_id = body.get("session_id")
-        message_group_id = SQSHandler.get_message_system_attributes(record).get("MessageGroupId")
-        if session_id != message_group_id:
-            cls._log.info(f"Session ID mismatch: message body session_id {session_id} does not match MessageGroupId {message_group_id}")
-            error_message_body = cls._construct_error_message_body(error_msg="Session ID mismatch")
-            record_attributes = cls._get_record_attributes(raw_queue_message=record)
-            cls._send_to_output_queue(message_body=error_message_body, record_attributes=record_attributes)
-            cls._log.info(f"Sent Session ID Mismatch message to Output Queue: '{SQSHandler.get_output_queue_url()}'")
-            return
         _, agent_response = cls._get_chat_service().process_chat_request(body=body)
         cls._log.info(f"Chat service response: '{agent_response}'")
         record_attributes = cls._get_record_attributes(raw_queue_message=record)

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/aws/serverless/aklambda.py

@@ -5,12 +5,6 @@ import logging
 import os
 from typing import Any, Callable, Dict, Optional, Tuple
 
-logging.basicConfig(
-    level=logging.DEBUG,
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-    force=True,
-)
-
 
 class LambdaRouter:
     """

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/aws/serverless/akresponsehandler.py

@@ -32,10 +32,10 @@ class ResponseHandler(LambdaSQSConsumer):
         :param body: Optional message body payload. If not provided, uses record["body"]
         :return: Message dictionary for storage
         """
-        session_id = SQSHandler.get_message_system_attributes(record).get("MessageGroupId")
         message_body = body if body is not None else record.get("body")
         if isinstance(message_body, str):
             message_body = json.loads(message_body)
+        session_id = message_body.get("session_id")
 
         message_attributes = SQSHandler.get_message_custom_attributes(record)
         request_id = message_attributes.get("request_id")

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/azure/akfunction.py

@@ -15,7 +15,13 @@ class AzureFunctions:
 
     _log = logging.getLogger("ak.azure.functions")
     _log.setLevel(logging.DEBUG)
-    _chat_service = ChatService()
+    _chat_service = None
+
+    @classmethod
+    def _get_chat_service(cls) -> ChatService:
+        if cls._chat_service is None:
+            cls._chat_service = ChatService()
+        return cls._chat_service
 
     @classmethod
     def handler(cls, req: func.HttpRequest) -> func.HttpResponse:

{agentkernel-0.3.2 → agentkernel-0.4.0}/src/agentkernel/deployment/common/response_store.py

@@ -10,7 +10,7 @@ class ResponseStore(ABC):
     Abstract base class for response message storage systems.
     """
 
-    _log = __import__("logging").getLogger(__name__)
+    _log = __import__("logging").getLogger("ak.deployment.response_store")
 
     @abstractmethod
     def add_message(self, message: Dict) -> None: