agentomatic 0.9.0__tar.gz → 1.0.0__tar.gz

{agentomatic-0.9.0 → agentomatic-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentomatic
-Version: 0.9.0
+Version: 1.0.0
 Summary: Drop agents, not code — zero-code multi-agent API platform framework
 Project-URL: Homepage, https://github.com/UnicoLab/agentomatic
 Project-URL: Documentation, https://unicolab.github.io/agentomatic

{agentomatic-0.9.0 → agentomatic-1.0.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentomatic"
-version = "0.9.0"
+version = "1.0.0"
 description = "Drop agents, not code — zero-code multi-agent API platform framework"
 readme = "README.md"
 license = {text = "MIT"}
@@ -92,6 +92,8 @@ Changelog = "https://github.com/UnicoLab/agentomatic/blob/main/CHANGELOG.md"
 
 
 
+
+
 [dependency-groups]
 dev = [
     "pytest>=8",

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/_version.py

@@ -2,4 +2,4 @@
 
 from __future__ import annotations
 
-__version__ = "0.9.0"
+__version__ = "1.0.0"

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/agents/optimizers.py

@@ -12,12 +12,15 @@ from __future__ import annotations
 
 import itertools
 from collections.abc import Sequence
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from loguru import logger
 
 from .types import AgentDataset, Metric
 
+if TYPE_CHECKING:
+    from agentomatic.optimize.llm_types import LLMSpec
+
 # ---------------------------------------------------------------------------
 # NoOpOptimizer
 # ---------------------------------------------------------------------------
@@ -175,8 +178,8 @@ class PromptFitterBridge:
     def __init__(
         self,
         agent_name: str = "",
-        task_model: str = "ollama/qwen2.5:7b",
-        rewrite_model: str = "openai/gpt-4.1",
+        task_model: LLMSpec = "ollama/qwen2.5:7b",
+        rewrite_model: LLMSpec = "openai/gpt-4.1",
         **kwargs: Any,
     ) -> None:
         self.agent_name = agent_name

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/cli/commands.py

@@ -364,7 +364,11 @@ def init(name: str, agents_dir: str, template: str | None, force: bool) -> None:
 @click.option("--title", default=None, help="Platform title")
 @click.option("--log-level", default="INFO", help="Log level")
 @click.option("--with-ui", "--ui", is_flag=True, help="Enable Chainlit debug UI at /chat")
-@click.option("--studio", is_flag=True, help="Enable Agentomatic Studio debug UI at /studio/ui")
+@click.option(
+    "--studio/--no-studio",
+    default=True,
+    help="Enable Agentomatic Studio debug UI at /studio/ui (default: on)",
+)
 def run(
     agents_dir: str,
     plugins_dir: str,

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/cli/templates.py

@@ -1720,14 +1720,14 @@ def get_template_files(template: str, name: str) -> dict[str, str]:
 
     if template == "basic":
         return {
-            "__init__.py": '"""Agent package."""\\n',
+            "__init__.py": '"""Agent package."""\n',
             "agent.py": _class_agent_py(name, "basic"),
             **common,
         }
 
     elif template == "full":
         return {
-            "__init__.py": '"""Agent package."""\\n',
+            "__init__.py": '"""Agent package."""\n',
             "agent.py": _class_agent_py(name, "full"),
             "config.py": _config_py(name),
             "schemas.py": _schemas_py(name),
@@ -1744,7 +1744,7 @@ def get_template_files(template: str, name: str) -> dict[str, str]:
 
     elif template == "coordinator":
         return {
-            "__init__.py": '"""Agent package."""\\n',
+            "__init__.py": '"""Agent package."""\n',
             "agent.py": _coordinator_agent_py(name),
             "delegation.py": _coordinator_delegation_py(name),
             "security.py": _coordinator_security_py(name),
@@ -1765,7 +1765,7 @@ def get_template_files(template: str, name: str) -> dict[str, str]:
 
     elif template == "rag":
         return {
-            "__init__.py": '"""Agent package."""\\n',
+            "__init__.py": '"""Agent package."""\n',
             "agent.py": _class_agent_py(name, "rag"),
             "config.py": _config_py(name),
             "tools.py": _tools_py(name),
@@ -1774,7 +1774,7 @@ def get_template_files(template: str, name: str) -> dict[str, str]:
 
     elif template == "chatbot":
         return {
-            "__init__.py": '"""Agent package."""\\n',
+            "__init__.py": '"""Agent package."""\n',
             "agent.py": _class_agent_py(name, "chatbot"),
             "config.py": _config_py(name),
             **common,

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/config/settings.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import threading
 from pathlib import Path
 from typing import Any
 
@@ -105,20 +106,25 @@ class PlatformSettings(BaseSettings):
 
 
 _settings: PlatformSettings | None = None
+_settings_lock = threading.Lock()
 
 
 def get_settings() -> PlatformSettings:
     """Get or create the singleton settings instance."""
     global _settings
-    if _settings is None:
-        _settings = PlatformSettings()  # type: ignore[call-arg]
-    return _settings
+    if _settings is not None:
+        return _settings
+    with _settings_lock:
+        if _settings is None:
+            _settings = PlatformSettings()  # type: ignore[call-arg]
+        return _settings
 
 
 def reset_settings() -> None:
     """Reset settings singleton (for testing)."""
     global _settings
-    _settings = None
+    with _settings_lock:
+        _settings = None
 
 
 def load_environment(env_file: str | Path | None = None) -> None:

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/core/platform.py

@@ -790,29 +790,31 @@ class AgentPlatform:
                 app.include_router(studio_router)
                 logger.info("🎨 Studio API mounted at /studio/")
 
-                # Mount the built Studio UI (if available)
-                from agentomatic.studio.serve import is_studio_available, mount_studio_ui
-
-                if is_studio_available():
-                    mount_studio_ui(app)
-
-                    # Convenience redirects: /studio → /studio/ui/
-                    from fastapi.responses import RedirectResponse
-
-                    @app.get("/studio", include_in_schema=False)
-                    @app.get("/studio/", include_in_schema=False)
-                    async def _studio_redirect() -> RedirectResponse:
-                        return RedirectResponse(url="/studio/ui/")
-
-                    logger.info("🎨 Studio UI + redirects mounted")
-                else:
-                    logger.info(
-                        "🎨 Studio API is running but UI assets not bundled. "
-                        "Run the frontend separately or build with: "
-                        "./scripts/build_studio.sh"
-                    )
+                # Mount the built Studio UI (always — shows helpful
+                # error page if assets are missing)
+                from agentomatic.studio.serve import mount_studio_ui
+
+                mount_studio_ui(app)
+
+                # Convenience redirects: /studio → /studio/ui/
+                from fastapi.responses import RedirectResponse
+
+                @app.get("/studio", include_in_schema=False)
+                @app.get("/studio/", include_in_schema=False)
+                async def _studio_redirect() -> RedirectResponse:
+                    return RedirectResponse(url="/studio/ui/")
+
             except Exception as exc:  # noqa: BLE001
                 logger.warning(f"Studio setup failed: {exc}")
+        else:
+            # Studio disabled — mount informative error pages so users
+            # hitting /studio/ui/ get guidance instead of a bare 404.
+            try:
+                from agentomatic.studio.serve import mount_studio_disabled_page
+
+                mount_studio_disabled_page(app)
+            except Exception:  # noqa: BLE001
+                pass  # Non-critical — don't log noise
 
         self._app = app
         return app

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/demo/__init__.py

@@ -1 +1,3 @@
 """Built-in demo agent for testing Agentomatic Studio end-to-end."""
+
+from __future__ import annotations

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/__init__.py

@@ -67,6 +67,14 @@ from agentomatic.optimize.judges import (
     LocalJudgeMetric,
     MultiJudgePanel,
 )
+
+# ── Pluggable LLM type system ────────────────────────────────────────
+from agentomatic.optimize.llm_types import (
+    LLMCallable,
+    LLMSpec,
+    call_llm,
+    call_llm_json,
+)
 from agentomatic.optimize.loop import (
     AVAILABLE_STRATEGIES,
     LoopResult,
@@ -112,6 +120,11 @@ from agentomatic.optimize.synthesizer import (
 )
 
 __all__ = [
+    # ── Pluggable LLM type system ─────────────────────────────────
+    "LLMSpec",
+    "LLMCallable",
+    "call_llm",
+    "call_llm_json",
     # Core — local-first optimization loop
     "PromptOptimizationLoop",
     "LoopResult",

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/failure_analysis.py

@@ -13,10 +13,13 @@ Classes
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from loguru import logger
 
+if TYPE_CHECKING:
+    from agentomatic.optimize.llm_types import LLMSpec
+
 # =====================================================================
 # Failure Clustering
 # =====================================================================
@@ -71,7 +74,7 @@ class FailureClusterer:
 
     def __init__(
         self,
-        model: str = "ollama/qwen2.5:7b",
+        model: LLMSpec = "ollama/qwen2.5:7b",
         max_clusters: int = 8,
         max_examples_per_cluster: int = 3,
     ) -> None:

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/fitter.py

@@ -43,7 +43,7 @@ import json
 import time
 import uuid
 from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from loguru import logger
 
@@ -77,6 +77,9 @@ from agentomatic.optimize.metrics import (
 from agentomatic.optimize.runner import AgentRunner, RunResult
 from agentomatic.optimize.search_space import PromptSearchSpace
 
+if TYPE_CHECKING:
+    from agentomatic.optimize.llm_types import LLMSpec
+
 # ---------------------------------------------------------------------------
 # Constants
 # ---------------------------------------------------------------------------
@@ -164,8 +167,8 @@ class PromptFitter:
         self,
         agent: str,
         base_prompt_version: str = "v1",
-        task_model: str = "ollama/qwen2.5:7b",
-        rewrite_model: str | None = None,
+        task_model: LLMSpec = "ollama/qwen2.5:7b",
+        rewrite_model: LLMSpec | None = None,
         local_judges: list[str] | None = None,
         search_space: PromptSearchSpace | None = None,
         optimizer: str | Any = "gepa_like",

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/fitter_optimizers.py

@@ -51,6 +51,7 @@ from agentomatic.optimize.search_space import PromptSearchSpace
 
 if TYPE_CHECKING:
     from agentomatic.optimize.context import OptimizationContext
+    from agentomatic.optimize.llm_types import LLMSpec
 
 # =====================================================================
 # Perspectives used by MIPRO-like instruction generation
@@ -177,7 +178,7 @@ class RewriteOptimizer(BaseFitterOptimizer):
     """
 
     name: str = "rewrite"
-    model: str = "ollama/qwen2.5:7b"
+    model: LLMSpec = "ollama/qwen2.5:7b"
     max_failures: int = 5
 
     async def propose(
@@ -557,7 +558,7 @@ class MIPROLikeOptimizer(BaseFitterOptimizer):
     """
 
     name: str = "mipro_like"
-    model: str = "ollama/qwen2.5:7b"
+    model: LLMSpec = "ollama/qwen2.5:7b"
     n_instruction_candidates: int = 5
     n_few_shot_candidates: int = 3
     fuse_top_k: int = 2
@@ -858,8 +859,8 @@ class GEPALikeOptimizer(BaseFitterOptimizer):
     """
 
     name: str = "gepa_like"
-    judge_model: str = "ollama/qwen2.5:7b"
-    rewrite_model: str = "ollama/qwen2.5:7b"
+    judge_model: LLMSpec = "ollama/qwen2.5:7b"
+    rewrite_model: LLMSpec = "ollama/qwen2.5:7b"
     n_mutations: int = 3
 
     # Default mutation aspects when fewer categories are found in feedback
@@ -1444,8 +1445,8 @@ def _build_failure_summary(
 
 def resolve_fitter_optimizer(
     name: str | BaseFitterOptimizer,
-    model: str = "ollama/qwen2.5:7b",
-    rewrite_model: str | None = None,
+    model: LLMSpec = "ollama/qwen2.5:7b",
+    rewrite_model: LLMSpec | None = None,
     **kwargs: Any,
 ) -> BaseFitterOptimizer:
     """Resolve a fitter optimizer by name or pass through an existing instance.

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/judges.py

@@ -15,12 +15,15 @@ from __future__ import annotations
 
 import asyncio
 from dataclasses import dataclass, field
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from loguru import logger
 
 from agentomatic.optimize.metrics import BaseMetric, EvalResult, MetricResult
 
+if TYPE_CHECKING:
+    from agentomatic.optimize.llm_types import LLMSpec
+
 # =====================================================================
 # Local SLM Judge
 # =====================================================================
@@ -48,7 +51,7 @@ class LocalJudgeMetric(BaseMetric):
     def __init__(
         self,
         name: str = "local_judge",
-        model: str = "ollama/qwen2.5:7b",
+        model: LLMSpec = "ollama/qwen2.5:7b",
         criteria: str = "Evaluate the quality and correctness of the response.",
         dimensions: list[str] | None = None,
         weight: float = 1.0,

{agentomatic-0.9.0 → agentomatic-1.0.0}/src/agentomatic/optimize/llm_caller.py

@@ -7,6 +7,8 @@ calling Ollama directly, every call goes through
 :class:`LLMCaller` which handles:
 
 * **Provider routing** — ``ollama/``, ``openai/``, ``litellm/`` prefixes.
+* **Callable dispatch** — custom async/sync callables and LangChain
+  models are transparently supported via :data:`LLMSpec`.
 * **Graceful degradation** — failures are logged and an empty string
   (or empty dict for JSON calls) is returned so the caller never crashes.
 * **JSON extraction** — :meth:`LLMCaller.call_with_json` strips
@@ -16,16 +18,23 @@ Example
 -------
 >>> text = await LLMCaller.call("ollama/mistral:7b", "Say hello")
 >>> data = await LLMCaller.call_with_json("openai/gpt-4o-mini", "Return {\"ok\": true}")
+>>>
+>>> # Custom callable also works:
+>>> async def my_llm(prompt, *, system_prompt=None): return "Hello!"
+>>> text = await LLMCaller.call(my_llm, "Say hello")
 """
 
 from __future__ import annotations
 
 import json
 import re
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from loguru import logger
 
+if TYPE_CHECKING:
+    from agentomatic.optimize.llm_types import LLMSpec
+
 # =====================================================================
 # Constants
 # =====================================================================
@@ -102,6 +111,10 @@ class LLMCaller:
 
     All methods are **static / async** so that no instance state is
     required — just call ``await LLMCaller.call(model, prompt)``.
+
+    Accepts both **string model specs** (e.g. ``"ollama/mistral:7b"``) and
+    **custom callables** (async functions, LangChain models, etc.) via the
+    :data:`~agentomatic.optimize.llm_types.LLMSpec` union type.
     """
 
     # -----------------------------------------------------------------
@@ -110,7 +123,7 @@ class LLMCaller:
 
     @staticmethod
     async def call(
-        model: str,
+        model: LLMSpec,
         prompt: str,
         *,
         system_prompt: str | None = None,
@@ -124,8 +137,8 @@ class LLMCaller:
         Parameters
         ----------
         model:
-            Model specification, optionally prefixed with the provider
-            (e.g. ``"ollama/mistral:7b"``, ``"openai/gpt-4o-mini"``).
+            Model specification — a string like ``"ollama/mistral:7b"``
+            or a callable / LangChain model matching :data:`LLMSpec`.
         prompt:
             User prompt text.
         system_prompt:
@@ -145,6 +158,21 @@ class LLMCaller:
         str
             Generated text, or ``""`` on failure.
         """
+        # ── Non-string: delegate to unified callable dispatcher ──
+        if not isinstance(model, str):
+            from agentomatic.optimize.llm_types import call_llm
+
+            return await call_llm(
+                model,
+                prompt,
+                system_prompt=system_prompt,
+                temperature=temperature,
+                max_tokens=max_tokens,
+                json_mode=json_mode,
+                timeout=timeout,
+            )
+
+        # ── String: route to provider backend ────────────────────
         provider, model_name = parse_model_spec(model)
         try:
             if provider == "ollama":
@@ -196,7 +224,7 @@ class LLMCaller:
 
     @staticmethod
     async def call_with_json(
-        model: str,
+        model: LLMSpec,
         prompt: str,
         *,
         system_prompt: str | None = None,
@@ -210,10 +238,12 @@ class LLMCaller:
         If the first attempt fails to parse, the call is retried up to
         *max_retries* times.  Returns an empty ``{}`` on total failure.
 
+        Accepts both string model specs and callables via :data:`LLMSpec`.
+
         Parameters
         ----------
         model:
-            Model specification (see :meth:`call`).
+            Model specification — string or callable (see :meth:`call`).
         prompt:
             User prompt — a JSON-return instruction is appended automatically.
         system_prompt:
@@ -230,9 +260,23 @@ class LLMCaller:
         dict[str, Any]
             Parsed JSON object, or ``{}`` on failure.
         """
+        # For non-string models, delegate to call_llm_json
+        if not isinstance(model, str):
+            from agentomatic.optimize.llm_types import call_llm_json
+
+            return await call_llm_json(
+                model,
+                prompt,
+                system_prompt=system_prompt,
+                temperature=temperature,
+                max_retries=max_retries,
+                timeout=timeout,
+            )
+
         json_instruction = (
             "\n\nIMPORTANT: Reply with ONLY a valid JSON object. "
-            "Do not include any other text, explanation, or markdown formatting."
+            "Do not include any other text, explanation, or "
+            "markdown formatting."
         )
         augmented_prompt = prompt + json_instruction
 
@@ -329,9 +373,10 @@ async def _call_openai(
         kwargs["response_format"] = {"type": "json_object"}
 
     client = openai.AsyncOpenAI(timeout=timeout)
-    response = await client.chat.completions.create(**kwargs)
-    choice = response.choices[0]
-    return (choice.message.content or "").strip()
+    async with client as c:
+        response = await c.chat.completions.create(**kwargs)
+        choice = response.choices[0]
+        return (choice.message.content or "").strip()
 
 
 async def _call_litellm(