idun-agent-engine 0.4.0__tar.gz → 0.4.2__tar.gz

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/.gitignore

@@ -3,6 +3,8 @@ __pycache__/
 *.py[cod]
 *$py.class
 
+
+
 .python-version
 # C extensions
 *.so
@@ -128,15 +130,24 @@ celerybeat.pid
 # SageMath parsed files
 *.sage.py
 
-# Environments
+# Environments / local config
 .env
-.venv
+.env.*
+!.env.example
+.venv/
 env/
 venv/
 ENV/
 env.bak/
 venv.bak/
 
+# Node / frontend
+node_modules/
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+
 # Spyder project settings
 .spyderproject
 .spyproject
@@ -177,34 +188,15 @@ cython_debug/
 # Database files
 *.db
 *.db-shm
-*.db-wai
+*.db-wal
 *.db-journal*
 *.sqlite
 *.sqlite3
 agents.db
 checkpoint_async.db
 
-# Python virtual environment
-.venv/
-venv/
-ENV/
-
-# Python cache
-__pycache__/
-*.pyc
-*.pyo
-*.pyd
-
-# Poetry
+# Poetry (unused in this repo)
 poetry.lock
 
-# Environment variables
-.env
-
-# Custom Gitignore for Idun
-*.db-wal
-.DS_Store
-*.DS_Store
-
-# uv lock file
-uv.lock
+# Editor / tool artifacts
+.specstory/

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.4.0
+Version: 0.4.2
 Summary: Python SDK and runtime to serve AI agents with FastAPI, LangGraph, and observability.
 Project-URL: Homepage, https://github.com/geoffreyharrazi/idun-agent-platform
 Project-URL: Repository, https://github.com/geoffreyharrazi/idun-agent-platform
@@ -31,7 +31,7 @@ Requires-Dist: deepagents<1.0.0,>=0.2.8
 Requires-Dist: fastapi<0.116.0,>=0.115.0
 Requires-Dist: google-adk<2.0.0,>=1.19.0
 Requires-Dist: google-cloud-logging<4.0.0,>=3.10.0
-Requires-Dist: guardrails-ai<0.8.0,>=0.7.0
+Requires-Dist: guardrails-ai<0.8.0,>=0.7.2
 Requires-Dist: httpx<0.29.0,>=0.28.1
 Requires-Dist: idun-agent-schema<1.0.0,>=0.3.8
 Requires-Dist: langchain-core<2.0.0,>=1.0.0

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "idun-agent-engine"
-version = "0.4.0"
+version = "0.4.2"
 description = "Python SDK and runtime to serve AI agents with FastAPI, LangGraph, and observability."
 authors = [{ name = "Geoffrey HARRAZI", email = "geoffreyharrazi@gmail.com" }]
 requires-python = ">=3.12,<3.14"
@@ -32,7 +32,7 @@ classifiers = [
 dependencies = [
     "tavily-python>=0.7.9,<0.8.0",
     "deepagents>=0.2.8,<1.0.0",
-    "guardrails-ai>=0.7.0, <0.8.0",
+    "guardrails-ai>=0.7.2, <0.8.0",
     "fastapi>=0.115.0,<0.116.0",
     "uvicorn>=0.35.0,<0.36.0",
     "langgraph>=1.0.0,<2.0.0",
@@ -50,7 +50,7 @@ dependencies = [
     "arize-phoenix-otel>=0.2.0,<1.0.0",
     "openinference-instrumentation-langchain>=0.1.13,<1.0.0",
     "langchain>=1.0.0,<2.0.0",
-    "textual>=7.3.0,<7.4.0",  
+    "textual>=7.3.0,<7.4.0",
     # Pin <12: 12.x currently depends on placeholder jmespath==99.99.99 and fails in Docker
     "arize-phoenix>=11.22.0,<12.0.0",
     "idun-agent-schema>=0.3.8,<1.0.0",
@@ -98,6 +98,8 @@ dev = [
     "pytest-cov>=5.0.0,<7",
     "types-PyYAML>=6.0.12.20240808,<7",
     "pytest-asyncio>=1.2.0",
+    "pytest-xdist>=3.8.0",
+    "posthog>=7.5.1",
 ]
 
 [tool.hatch.build.targets.sdist]
@@ -155,10 +157,16 @@ select = [
     "UP",  # pyupgrade
     "N",   # pep8-naming
     "SIM", # flake8-simplify
-    "D",   # pydocstyle (docstrings)
 ]
 extend-ignore = [
     "E501",  # line length handled by black
+    # Keep linting actionable without behavior changes across the codebase.
+    "E722",   # bare except
+    "B904",   # raise from within except
+    "SIM105", # prefer contextlib.suppress
+    "SIM117", # combine nested with
+    "SIM118", # key in dict instead of key in dict.keys()
+    "UP038",  # isinstance union style
 ]
 
 [tool.ruff.lint.per-file-ignores]
@@ -180,18 +188,45 @@ pretty = true
 show_error_codes = true
 
 [tool.pytest.ini_options]
-addopts = "-ra -q --strict-markers --disable-warnings"
-testpaths = [
-    "tests",
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+addopts = "-ra -q --strict-markers"
+testpaths = ["tests/unit", "tests/integration"]
+pythonpath = ["src", "tests"]
+filterwarnings = [
+    "ignore::DeprecationWarning",
+    "ignore::PendingDeprecationWarning",
 ]
-pythonpath = [
-    "src",
+markers = [
+    "unit: Unit tests (fast, no external dependencies)",
+    "integration: Integration tests (may require external services)",
+    "slow: Slow tests (marked for optional skipping)",
+    "requires_langfuse: Test requires Langfuse service",
+    "requires_phoenix: Test requires Phoenix service",
+    "requires_postgres: Test requires PostgreSQL",
 ]
 
 [tool.coverage.run]
 branch = true
-source = ["src"]
+source = ["src/idun_agent_engine"]
+omit = [
+    "*/__init__.py",
+    "*/templates/*",
+    "*/_version.py",
+    "*/idun_agent_engine_ui/*",
+    "*/idun_platform_cli/*",
+    "*/mcp/*",  # CHANGED: MCP has Pydantic compatibility issues with coverage
+]
 
 [tool.coverage.report]
 show_missing = true
-skip_covered = true
+skip_covered = false
+fail_under = 0  # Start with 0, increase as coverage grows
+exclude_lines = [
+    "pragma: no cover",
+    "if TYPE_CHECKING:",
+    "raise NotImplementedError",
+    "if __name__ == .__main__.:",
+    "@abstractmethod",
+    "pass",
+]

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/_version.py

@@ -1,3 +1,3 @@
 """Version information for Idun Agent Engine."""
 
-__version__ = "0.4.0"
+__version__ = "0.4.2"

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/agent/adk/adk.py

@@ -5,6 +5,7 @@ import uuid
 from collections.abc import AsyncGenerator
 from typing import Any
 
+from ag_ui_adk import ADKAgent as ADKAGUIAgent
 from google.adk.apps.app import App
 from google.adk.memory import (
     InMemoryMemoryService,
@@ -25,9 +26,8 @@ from idun_agent_schema.engine.adk import (
 )
 from idun_agent_schema.engine.observability_v2 import ObservabilityConfig
 
-from ag_ui_adk import ADKAgent as ADKAGUIAgent
-from idun_agent_engine.agent import base as agent_base
 from idun_agent_engine import observability
+from idun_agent_engine.agent import base as agent_base
 
 
 class AdkAgent(agent_base.BaseAgent):
@@ -123,7 +123,7 @@ class AdkAgent(agent_base.BaseAgent):
         # Observability (provider-agnostic)
         if observability_config:
             handlers, infos = observability.create_observability_handlers(
-                observability_config # type: ignore[arg-type]
+                observability_config  # type: ignore[arg-type]
             )
             self._obs_callbacks = []
             for handler in handlers:
@@ -153,6 +153,7 @@ class AdkAgent(agent_base.BaseAgent):
 
                 if is_langfuse_enabled:
                     import os
+
                     langfuse_pk = os.environ.get("LANGFUSE_PUBLIC_KEY")
                     langfuse_host = os.environ.get("LANGFUSE_BASE_URL")
                     print(f"LANGFUSE_PUBLIC_KEY: {langfuse_pk}")
@@ -171,7 +172,9 @@ class AdkAgent(agent_base.BaseAgent):
                     except Exception as e:
                         print(f"Failed to instrument Google ADK: {e}")
             except Exception as e:
-                print(f"Error checking observability config for ADK instrumentation: {e}")
+                print(
+                    f"Error checking observability config for ADK instrumentation: {e}"
+                )
 
         # Initialize Session Service
         await self._initialize_session_service()

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/agent/haystack/__init__.py

@@ -1,9 +1,7 @@
 """LangGraph agent package."""
 
 from .haystack import HaystackAgent
-from .haystack_model import HaystackAgentConfig
 
 __all__ = [
     "HaystackAgent",
-    "HaystackAgentConfig",
 ]

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/agent/haystack/haystack.py

@@ -10,9 +10,10 @@ from haystack import Pipeline
 from haystack.components.agents import Agent
 from haystack.dataclasses import ChatMessage
 from haystack_integrations.components.connectors.langfuse import LangfuseConnector
+from idun_agent_schema.engine.haystack import HaystackAgentConfig
+from idun_agent_schema.engine.observability_v2 import ObservabilityConfig
 
 from idun_agent_engine.agent.base import BaseAgent
-from idun_agent_schema.engine.haystack import HaystackAgentConfig
 from idun_agent_engine.agent.haystack.utils import _parse_component_definition
 
 logging.basicConfig(
@@ -77,7 +78,9 @@ class HaystackAgent(BaseAgent):
         Raises:
             RuntimeError: If the CopilotKit agent is not yet initialized.
         """
-        raise NotImplementedError("CopilotKit agent instance not supported yet for Haystack agent.")
+        raise NotImplementedError(
+            "CopilotKit agent instance not supported yet for Haystack agent."
+        )
 
     @property
     def configuration(self) -> HaystackAgentConfig:
@@ -146,9 +149,10 @@ class HaystackAgent(BaseAgent):
             # TODO: await persistence haystack
             # TODO OBS block
 
-            # check if config has observability `enabled` or `disabled`, so that we adjust our component to
-            # either add a tracer or not
-            if self._configuration.observability.enabled:
+            if (
+                self._configuration.observability
+                and self._configuration.observability.enabled
+            ):  # FIXED: should also check for enabled
                 self._enable_tracing = True
                 logger.info("Enabling tracing...")
             component: Agent | Pipeline = self._load_component(

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/agent/langgraph/langgraph.py

@@ -1,7 +1,7 @@
 """LangGraph agent adapter implementing the BaseAgent protocol."""
 
-import importlib.util
 import importlib
+import importlib.util
 import uuid
 from collections.abc import AsyncGenerator
 from typing import Any
@@ -9,6 +9,7 @@ from typing import Any
 import aiosqlite
 from ag_ui.core import events as ag_events
 from ag_ui.core import types as ag_types
+from copilotkit import LangGraphAGUIAgent
 from idun_agent_schema.engine.langgraph import (
     InMemoryCheckpointConfig,
     LangGraphAgentConfig,
@@ -24,7 +25,6 @@ from langgraph.graph.state import CompiledStateGraph
 
 from idun_agent_engine import observability
 from idun_agent_engine.agent import base as agent_base
-from copilotkit import LangGraphAGUIAgent
 
 
 class LanggraphAgent(agent_base.BaseAgent):
@@ -202,8 +202,11 @@ class LanggraphAgent(agent_base.BaseAgent):
             self._agent_instance = graph_builder.compile(
                 checkpointer=self._checkpointer, store=self._store
             )
-        elif isinstance(graph_builder, CompiledStateGraph): # TODO: to remove, dirty fix for template deepagent langgraph
-            self._agent_instance = graph_builder
+        elif isinstance(graph_builder, CompiledStateGraph):
+            # TODO: this was made for supporting langgraph's DeepAgent, modernize.
+            raise TypeError(
+                "Expected StateGraph, Got CompiledStateGraph. Make sure not to run `compile` on your agent's graph."
+            )
 
         self._copilotkit_agent_instance = LangGraphAGUIAgent(
             name=self._name,
@@ -212,12 +215,6 @@ class LanggraphAgent(agent_base.BaseAgent):
             config={"callbacks": self._obs_callbacks} if self._obs_callbacks else None,
         )
 
-        self._copilotkit_agent_instance = LangGraphAGUIAgent(
-            name=self._name,
-            description="Agent description", # TODO: add agent description
-            graph=self._agent_instance,
-        )
-
         if self._agent_instance:
             try:
                 self._input_schema = self._agent_instance.input_schema
@@ -252,9 +249,10 @@ class LanggraphAgent(agent_base.BaseAgent):
 
         if self._configuration.checkpointer:
             if isinstance(self._configuration.checkpointer, SqliteCheckpointConfig):
-                self._connection = await aiosqlite.connect(
-                    self._configuration.checkpointer.db_path
+                db_path = self._configuration.checkpointer.db_url.replace(
+                    "sqlite:///", ""
                 )
+                self._connection = await aiosqlite.connect(db_path)
                 self._checkpointer = AsyncSqliteSaver(conn=self._connection)
                 self._infos["checkpointer"] = (
                     self._configuration.checkpointer.model_dump()
@@ -344,7 +342,6 @@ class LanggraphAgent(agent_base.BaseAgent):
     def _validate_graph_builder(
         self, graph_builder: Any, module_path: str, graph_variable_name: str
     ) -> StateGraph:
-        # TODO to remove, dirty fix for template deepagent langgraph
         if not isinstance(graph_builder, StateGraph) and not isinstance(
             graph_builder, CompiledStateGraph
         ):

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/core/config_builder.py

@@ -4,6 +4,7 @@ This module provides a fluent API for building configuration objects using Pydan
 This approach ensures type safety, validation, and consistency with the rest of the codebase.
 """
 
+import os
 from pathlib import Path
 from typing import Any
 
@@ -50,7 +51,6 @@ class ConfigBuilder:
         """Initialize a new configuration builder with default values."""
         self._server_config = ServerConfig()
         self._agent_config: AgentConfig | None = None
-        # TODO: add mcp_servers config
         self._mcp_servers: list[MCPServer] | None = None
         self._observability: list[ObservabilityConfig] | None = None
         self._guardrails: Guardrails | None = None
@@ -259,7 +259,7 @@ class ConfigBuilder:
             mcp_servers=self._mcp_servers,
         )
 
-    def build_dict(self) -> dict[str, Any]:
+    def build_dict(self) -> dict[str, Any]:  # NOT USED
         """Build and return the configuration as a dictionary.
 
         This is a convenience method for backward compatibility.
@@ -268,17 +268,24 @@ class ConfigBuilder:
             Dict[str, Any]: The complete configuration dictionary
         """
         engine_config = self.build()
-        return engine_config.model_dump()
+        return engine_config.model_dump(mode="python")
 
-    def save_to_file(self, file_path: str) -> None:
+    def save_to_file(
+        self, file_path: str
+    ) -> None:  # FIXED: old method doesn't serialize enums correctly
         """Save the configuration to a YAML file.
 
         Args:
             file_path: Path where to save the configuration file
         """
-        config = self.build_dict()
+        engine_model = self.build()
         with open(file_path, "w") as f:
-            yaml.dump(config, f, default_flow_style=False, indent=2)
+            yaml.dump(
+                engine_model.model_dump(mode="json"),
+                f,
+                default_flow_style=False,
+                indent=2,
+            )
 
     async def build_and_initialize_agent(
         self, mcp_registry: Any | None = None
@@ -319,9 +326,10 @@ class ConfigBuilder:
         # Initialize the appropriate agent
         agent_instance = None
         if agent_type == AgentFramework.LANGGRAPH:
-            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
             import os
 
+            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
+
             print("Current directory: ", os.getcwd())  # TODO remove
             try:
                 validated_config = LangGraphAgentConfig.model_validate(agent_config_obj)
@@ -334,10 +342,12 @@ class ConfigBuilder:
             agent_instance = LanggraphAgent()
 
         elif agent_type == AgentFramework.TRANSLATION_AGENT:
-            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
-            from idun_agent_schema.engine.templates import TranslationAgentConfig
             import os
 
+            from idun_agent_schema.engine.templates import TranslationAgentConfig
+
+            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
+
             try:
                 translation_config = TranslationAgentConfig.model_validate(
                     agent_config_obj
@@ -364,10 +374,12 @@ class ConfigBuilder:
             agent_instance = LanggraphAgent()
 
         elif agent_type == AgentFramework.CORRECTION_AGENT:
-            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
-            from idun_agent_schema.engine.templates import CorrectionAgentConfig
             import os
 
+            from idun_agent_schema.engine.templates import CorrectionAgentConfig
+
+            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
+
             try:
                 correction_config = CorrectionAgentConfig.model_validate(
                     agent_config_obj
@@ -391,10 +403,12 @@ class ConfigBuilder:
             agent_instance = LanggraphAgent()
 
         elif agent_type == AgentFramework.DEEP_RESEARCH_AGENT:
-            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
-            from idun_agent_schema.engine.templates import DeepResearchAgentConfig
             import os
 
+            from idun_agent_schema.engine.templates import DeepResearchAgentConfig
+
+            from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
+
             try:
                 deep_research_config = DeepResearchAgentConfig.model_validate(
                     agent_config_obj
@@ -517,6 +531,9 @@ class ConfigBuilder:
     def load_from_file(config_path: str = "config.yaml") -> EngineConfig:
         """Load configuration from a YAML file and return a validated EngineConfig.
 
+        Sets IDUN_CONFIG_PATH environment variable to enable MCP helper functions
+        (get_adk_tools, get_langchain_tools) to automatically discover the config file.
+
         Args:
             config_path: Path to the configuration YAML file
 
@@ -532,6 +549,9 @@ class ConfigBuilder:
             # Resolve relative to the current working directory
             path = Path.cwd() / path
 
+        # Set IDUN_CONFIG_PATH for MCP helpers to discover
+        os.environ["IDUN_CONFIG_PATH"] = str(path)
+
         with open(path) as f:
             config_data = yaml.safe_load(f)
 

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.2}/src/idun_agent_engine/guardrails/guardrails_hub/guardrails_hub.py

@@ -9,26 +9,51 @@ from ..base import BaseGuardrail
 
 def get_guard_instance(name: GuardrailConfigId) -> Guard:
     """Returns a map of guard type -> guard instance."""
-    if name.value == "ban_list":
+    if name == GuardrailConfigId.BAN_LIST:
         from guardrails.hub import BanList
 
         return BanList
 
-    elif name.value == "detect_pii":
+    elif name == GuardrailConfigId.DETECT_PII:
         from guardrails.hub import DetectPII
 
         return DetectPII
 
-    elif name.value == "nsfw":
+    elif name == GuardrailConfigId.NSFW_TEXT:
         from guardrails.hub import NSFWText
 
         return NSFWText
 
-    elif name.value == "competitor_check":
+    elif name == GuardrailConfigId.COMPETITION_CHECK:
         from guardrails.hub import CompetitorCheck
 
         return CompetitorCheck
 
+    elif name == GuardrailConfigId.BIAS_CHECK:
+        from guardrails.hub import BiasCheck
+
+        return BiasCheck
+
+    elif name == GuardrailConfigId.CORRECT_LANGUAGE:
+        from guardrails.hub import ValidLanguage
+
+        return ValidLanguage
+
+    elif name == GuardrailConfigId.GIBBERISH_TEXT:
+        from guardrails.hub import GibberishText
+
+        return GibberishText
+
+    elif name == GuardrailConfigId.TOXIC_LANGUAGE:
+        from guardrails.hub import ToxicLanguage
+
+        return ToxicLanguage
+
+    elif name == GuardrailConfigId.RESTRICT_TO_TOPIC:
+        from guardrails.hub import RestrictToTopic
+
+        return RestrictToTopic
+
     else:
         raise ValueError(f"Guard {name} not found.")
 
@@ -90,11 +115,29 @@ class GuardrailsHubGuard(BaseGuardrail):
                 f"Guard: {self.guard_id} is not yet supported, or does not exist."
             )
 
-        guard_instance_params = self._guardrail_config.guard_params.model_dump()
-        guard_instance = guard(**guard_instance_params)
-        for param, value in guard_instance_params.items():
-            setattr(guard_instance, param, value)
-        return guard_instance
+        if hasattr(self._guardrail_config, "guard_params"):
+            guard_instance_params = self._guardrail_config.guard_params.model_dump()
+        else:
+            config_dict = self._guardrail_config.model_dump()
+            exclude_fields = {"config_id", "api_key", "reject_message", "guard_url"}
+            guard_instance_params = {
+                k: v for k, v in config_dict.items() if k not in exclude_fields
+            }
+
+        try:
+            guard_instance = guard(**guard_instance_params)
+            for param, value in guard_instance_params.items():
+                setattr(guard_instance, param, value)
+            return guard_instance
+        except SystemError:
+            # sentencepiece mutex lock error when loading models in quick succession
+            import time
+
+            time.sleep(0.5)
+            guard_instance = guard(**guard_instance_params)
+            for param, value in guard_instance_params.items():
+                setattr(guard_instance, param, value)
+            return guard_instance
 
     def validate(self, input: str) -> bool:
         """TODO."""