PraisonAI 4.6.82__tar.gz → 4.6.83__tar.gz

{praisonai-4.6.82 → praisonai-4.6.83}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.82
+Version: 4.6.83
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,14 +11,16 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.82
+Requires-Dist: praisonaiagents>=1.6.90
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0
 Requires-Dist: mcp>=1.20.0
-Requires-Dist: typer>=0.9.0
+Requires-Dist: typer>=0.12.0
+Requires-Dist: click>=8.4.2
 Requires-Dist: textual>=0.47.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: toml>=0.10.2
 Provides-Extra: ui
 Requires-Dist: aiui<0.4,>=0.3.121; extra == "ui"
 Provides-Extra: gradio
@@ -93,6 +95,9 @@ Requires-Dist: praisonai-tools>=0.1.0; extra == "autogen-v4"
 Requires-Dist: crewai; extra == "autogen-v4"
 Provides-Extra: acp
 Requires-Dist: agent-client-protocol>=0.7.0; extra == "acp"
+Provides-Extra: flow
+Requires-Dist: langflow>=1.0.0; extra == "flow"
+Requires-Dist: requests>=2.31.0; extra == "flow"
 Provides-Extra: lite
 Provides-Extra: all
 Requires-Dist: fastapi>=0.115.0; extra == "all"
@@ -751,7 +756,7 @@ praisonai agents.yaml
 | Query Rewriter Agent | [Example](examples/python/agents/query-rewriter-agent.py) | [📖](https://docs.praison.ai/docs/agents/query-rewriter) |
 | Native Web Search | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/agents/websearch) |
 | Built-in Search Tools | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/tools/tavily) |
-| Unified Web Search | [Example](src/praisonai-agents/examples/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
+| Unified Web Search | [Example](examples/python/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
 | Web Fetch (Anthropic) | [Example](examples/python/agents/web-fetch-agent.py) | [📖](https://docs.praison.ai/docs/features/model-capabilities) |
 
 </details>

{praisonai-4.6.82 → praisonai-4.6.83}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.82
+Version: 4.6.83
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,14 +11,16 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.82
+Requires-Dist: praisonaiagents>=1.6.90
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0
 Requires-Dist: mcp>=1.20.0
-Requires-Dist: typer>=0.9.0
+Requires-Dist: typer>=0.12.0
+Requires-Dist: click>=8.4.2
 Requires-Dist: textual>=0.47.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: toml>=0.10.2
 Provides-Extra: ui
 Requires-Dist: aiui<0.4,>=0.3.121; extra == "ui"
 Provides-Extra: gradio
@@ -93,6 +95,9 @@ Requires-Dist: praisonai-tools>=0.1.0; extra == "autogen-v4"
 Requires-Dist: crewai; extra == "autogen-v4"
 Provides-Extra: acp
 Requires-Dist: agent-client-protocol>=0.7.0; extra == "acp"
+Provides-Extra: flow
+Requires-Dist: langflow>=1.0.0; extra == "flow"
+Requires-Dist: requests>=2.31.0; extra == "flow"
 Provides-Extra: lite
 Provides-Extra: all
 Requires-Dist: fastapi>=0.115.0; extra == "all"
@@ -751,7 +756,7 @@ praisonai agents.yaml
 | Query Rewriter Agent | [Example](examples/python/agents/query-rewriter-agent.py) | [📖](https://docs.praison.ai/docs/agents/query-rewriter) |
 | Native Web Search | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/agents/websearch) |
 | Built-in Search Tools | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/tools/tavily) |
-| Unified Web Search | [Example](src/praisonai-agents/examples/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
+| Unified Web Search | [Example](examples/python/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
 | Web Fetch (Anthropic) | [Example](examples/python/agents/web-fetch-agent.py) | [📖](https://docs.praison.ai/docs/features/model-capabilities) |
 
 </details>

{praisonai-4.6.82 → praisonai-4.6.83}/PraisonAI.egg-info/SOURCES.txt

@@ -61,6 +61,7 @@ praisonai/audit/__init__.py
 praisonai/audit/agent_centric.py
 praisonai/bots/__init__.py
 praisonai/bots/_ack.py
+praisonai/bots/_admission.py
 praisonai/bots/_approval.py
 praisonai/bots/_approval_base.py
 praisonai/bots/_approval_store.py
@@ -82,7 +83,9 @@ praisonai/bots/_ingress.py
 praisonai/bots/_media.py
 praisonai/bots/_metrics.py
 praisonai/bots/_mirror.py
+praisonai/bots/_outbound_media.py
 praisonai/bots/_outbound_messenger.py
+praisonai/bots/_outbound_resilience.py
 praisonai/bots/_outbox.py
 praisonai/bots/_pairing_ui.py
 praisonai/bots/_presentation_approval.py
@@ -351,14 +354,10 @@ praisonai/cli/features/output_style.py
 praisonai/cli/features/package.py
 praisonai/cli/features/performance.py
 praisonai/cli/features/persistence.py
-praisonai/cli/features/plugin_cli.py
-praisonai/cli/features/profile.py
-praisonai/cli/features/profiling.py
 praisonai/cli/features/recipe.py
 praisonai/cli/features/recipe_creator.py
 praisonai/cli/features/recipe_optimizer.py
 praisonai/cli/features/recipe_prompts.py
-praisonai/cli/features/recipes.py
 praisonai/cli/features/registry.py
 praisonai/cli/features/repo_map.py
 praisonai/cli/features/router.py
@@ -368,6 +367,7 @@ praisonai/cli/features/sandbox_executor.py
 praisonai/cli/features/sdk_knowledge.py
 praisonai/cli/features/serve.py
 praisonai/cli/features/session.py
+praisonai/cli/features/session_checkpoints.py
 praisonai/cli/features/skills.py
 praisonai/cli/features/slash_commands.py
 praisonai/cli/features/telemetry.py
@@ -554,11 +554,13 @@ praisonai/framework_adapters/crewai_adapter.py
 praisonai/framework_adapters/praisonai_adapter.py
 praisonai/framework_adapters/registry.py
 praisonai/framework_adapters/validators.py
+praisonai/framework_adapters/workflow_framework.py
 praisonai/gateway/__init__.py
 praisonai/gateway/auth.py
 praisonai/gateway/client.py
 praisonai/gateway/cookie_auth.py
 praisonai/gateway/exec_approval.py
+praisonai/gateway/forensics.py
 praisonai/gateway/gateway_approval.py
 praisonai/gateway/health_monitor.py
 praisonai/gateway/home_channels.py
@@ -789,7 +791,6 @@ praisonai/scheduler/blueprint_catalogue.py
 praisonai/scheduler/condition_gate.py
 praisonai/scheduler/daemon_manager.py
 praisonai/scheduler/deployment.py
-praisonai/scheduler/enhanced_executor.py
 praisonai/scheduler/executor.py
 praisonai/scheduler/run_policy.py
 praisonai/scheduler/shared.py

{praisonai-4.6.82 → praisonai-4.6.83}/PraisonAI.egg-info/requires.txt

@@ -1,14 +1,16 @@
 rich>=13.7
 markdown>=3.5
 pyparsing>=3.0.0
-praisonaiagents>=1.6.82
+praisonaiagents>=1.6.90
 python-dotenv>=0.19.0
 litellm<2,>=1.83.14
 PyYAML>=6.0
 mcp>=1.20.0
-typer>=0.9.0
+typer>=0.12.0
+click>=8.4.2
 textual>=0.47.0
 pydantic>=2.0.0
+toml>=0.10.2
 
 [acp]
 agent-client-protocol>=0.7.0
@@ -141,6 +143,10 @@ pytest-asyncio>=0.21.0
 [dynamodb]
 boto3>=1.35.0
 
+[flow]
+langflow>=1.0.0
+requests>=2.31.0
+
 [gradio]
 gradio>=4.26.0
 

{praisonai-4.6.82 → praisonai-4.6.83}/README.md

@@ -568,7 +568,7 @@ praisonai agents.yaml
 | Query Rewriter Agent | [Example](examples/python/agents/query-rewriter-agent.py) | [📖](https://docs.praison.ai/docs/agents/query-rewriter) |
 | Native Web Search | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/agents/websearch) |
 | Built-in Search Tools | [Example](examples/python/agents/websearch-agent.py) | [📖](https://docs.praison.ai/docs/tools/tavily) |
-| Unified Web Search | [Example](src/praisonai-agents/examples/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
+| Unified Web Search | [Example](examples/python/web_search_example.py) | [📖](https://docs.praison.ai/docs/tools/web-search) |
 | Web Fetch (Anthropic) | [Example](examples/python/agents/web-fetch-agent.py) | [📖](https://docs.praison.ai/docs/features/model-capabilities) |
 
 </details>

{praisonai-4.6.82 → praisonai-4.6.83}/praisonai/_async_bridge.py

@@ -7,11 +7,13 @@ on every call (which is expensive and breaks multi-agent workflows).
 """
 import asyncio
 import atexit
+import contextlib
+import contextvars
 import os
 import threading
 import concurrent.futures
 from concurrent.futures import CancelledError as FutureCancelledError, Future
-from typing import Awaitable, TypeVar
+from typing import Awaitable, Iterator, Optional, TypeVar
 
 T = TypeVar("T")
 
@@ -25,6 +27,13 @@ class AsyncBridge:
         self._loop: asyncio.AbstractEventLoop | None = None
         self._thread: threading.Thread | None = None
         self._lock = threading.Lock()
+        # Permanently-closed flag. Set by ``shutdown(permanent=True)`` for
+        # scope-owned bridges so a leaked context (e.g. an ``asyncio.Task`` that
+        # copied the ContextVar inside a ``scoped_bridge`` block) cannot silently
+        # resurrect an orphaned loop+thread after the scope exits. The shared
+        # default is never marked permanent so its only teardown is the atexit
+        # hook at process exit.
+        self._closed = False
         # Identity of the thread currently holding ``self._lock``. Set under the
         # lock by ``get()``/``submit()`` so ``_spawn_locked()`` can verify the
         # *caller* (not merely *someone*) owns the lock.
@@ -47,6 +56,16 @@ class AsyncBridge:
             raise AssertionError(
                 "_spawn_locked() must be called while holding self._lock"
             )
+        # Refuse to resurrect a permanently-closed (scope-owned) bridge. This
+        # guards the case where a context copied inside ``scoped_bridge`` outlives
+        # the ``with`` block and later tries to run work through the now-shut-down
+        # bridge: instead of silently starting an orphaned loop+thread that no
+        # scope or atexit hook owns, fail loudly.
+        if self._closed:
+            raise RuntimeError(
+                "AsyncBridge has been shut down and cannot be reused; "
+                "this usually means a context outlived its scoped_bridge() block"
+            )
         if self._loop is None or self._loop.is_closed():
             self._loop = asyncio.new_event_loop()
             self._thread = threading.Thread(
@@ -103,9 +122,22 @@ class AsyncBridge:
             fut.cancel()
             raise
 
-    def shutdown(self, timeout: float = 5.0) -> None:
+    def shutdown(self, timeout: float = 5.0, *, permanent: bool = False) -> None:
+        """Tear down the loop+thread.
+
+        Args:
+            timeout: Seconds to wait for in-flight tasks to cancel and the
+                background thread to join.
+            permanent: When ``True`` the bridge is marked closed and may not be
+                reused; a later ``run_sync``/``submit`` raises ``RuntimeError``
+                rather than silently resurrecting an orphaned loop. Used by
+                :func:`scoped_bridge` for the bridges it owns. The shared default
+                is shut down with ``permanent=False`` so it never poisons callers.
+        """
         # Snapshot loop and thread outside lock to avoid holding lock during wait
         with self._lock:
+            if permanent:
+                self._closed = True
             loop, thread = self._loop, self._thread
             if loop is None:
                 return
@@ -149,10 +181,67 @@ class AsyncBridge:
 _BG: AsyncBridge = AsyncBridge()
 
 
+# Context-scoped override. When set (via :func:`scoped_bridge`), callers in that
+# context — and any code they transitively invoke, including async tasks that
+# copy the context — resolve to the scoped bridge instead of the shared default.
+# This lets server/gateway/managed callers bind a per-session loop+thread
+# without rewriting internal modules that import ``run_sync`` directly.
+_bridge_var: contextvars.ContextVar[Optional[AsyncBridge]] = contextvars.ContextVar(
+    "praisonai_async_bridge", default=None
+)
+
+
 def _default_bridge() -> AsyncBridge:
+    """Resolve the active bridge: context-scoped override, else shared default."""
+    scoped = _bridge_var.get()
+    if scoped is not None:
+        return scoped
     return _BG
 
 
+def current_bridge() -> AsyncBridge:
+    """Return the :class:`AsyncBridge` active in the current context.
+
+    Returns the bridge bound by an enclosing :func:`scoped_bridge` block, or the
+    process-wide shared default when no scope is active. Prefer this over reaching
+    for the module-level ``_BG`` so embedders can override the bridge via
+    dependency injection.
+    """
+    return _default_bridge()
+
+
+@contextlib.contextmanager
+def scoped_bridge(bridge: Optional[AsyncBridge] = None) -> Iterator[AsyncBridge]:
+    """Bind a per-scope :class:`AsyncBridge` for the duration of the ``with`` block.
+
+    Server callers (``praisonai serve``, gateway, managed agents) can isolate a
+    session's sync→async work onto its own loop+thread so a stuck coroutine in one
+    session does not park the shared default loop for every other session::
+
+        with scoped_bridge() as bridge:
+            run_sync(some_coro())   # uses ``bridge`` instead of the global default
+            ...
+        # bridge is shut down automatically on exit when created here.
+
+    Args:
+        bridge: An existing bridge to bind. When ``None`` a fresh bridge is created
+            for the scope and shut down on exit. A caller-provided bridge is left
+            untouched (the caller owns its lifecycle).
+    """
+    owns = bridge is None
+    if bridge is None:
+        bridge = AsyncBridge()
+    token = _bridge_var.set(bridge)
+    try:
+        yield bridge
+    finally:
+        _bridge_var.reset(token)
+        if owns:
+            # ``permanent=True`` poisons the bridge so a context that copied the
+            # ContextVar inside this block cannot resurrect it after we exit.
+            bridge.shutdown(permanent=True)
+
+
 def _shutdown_default() -> None:
     """Private process-exit hook: tear down the shared default bridge.
 

{praisonai-4.6.82 → praisonai-4.6.83}/praisonai/agents_generator.py

@@ -708,6 +708,16 @@ class AgentsGenerator:
         # Ensure name is present
         if 'name' not in config:
             config['name'] = config.get('topic', 'Workflow')
+
+        from .framework_adapters.workflow_framework import (
+            framework_from_config,
+            validate_workflow_framework,
+        )
+        effective_framework = (self.framework or framework_from_config(config))
+        validate_workflow_framework(
+            effective_framework,
+            source="agents.yaml workflow section",
+        )
         
         # Pass model from config_list to workflow as default_llm
         if self.config_list and self.config_list[0].get('model'):
@@ -763,6 +773,16 @@ class AgentsGenerator:
         # Ensure name is present (YAMLWorkflowParser handles roles->agents conversion)
         if 'name' not in config:
             config['name'] = config.get('topic', 'Workflow')
+
+        from .framework_adapters.workflow_framework import (
+            framework_from_config,
+            validate_workflow_framework,
+        )
+        effective_framework = (self.framework or framework_from_config(config))
+        validate_workflow_framework(
+            effective_framework,
+            source="agents.yaml workflow section",
+        )
         
         # Pass model from config_list to workflow as default_llm
         if self.config_list and self.config_list[0].get('model'):

{praisonai-4.6.82 → praisonai-4.6.83}/praisonai/auto.py

@@ -25,6 +25,21 @@ T = TypeVar('T')
 # =============================================================================
 
 from ._lazy_cache import lazy_get
+from ._async_bridge import run_sync
+import inspect as _inspect
+
+
+async def _maybe_await(value):
+    """Await ``value`` if it is awaitable, otherwise return it unchanged.
+
+    Lets a single coroutine drive both async provider calls (which return
+    awaitables) and sync provider calls (which return plain results) without
+    duplicating the surrounding fallback ladder.
+    """
+    if _inspect.isawaitable(value):
+        return await value
+    return value
+
 
 # Module-level cache for lazily constructed Pydantic workflow models.
 # Populated on first call to _get_workflow_models() / _get_job_workflow_models().
@@ -460,36 +475,63 @@ class BaseAutoGenerator:
         return False
 
     
-    def _structured_completion(self, response_model: Type[T], messages: List[Dict], **kwargs) -> T:
+    def _get_async_openai_client(self):
+        """Get or create the async OpenAI client for this instance.
+
+        Lazy-init AND snapshot the client under the same lock so the returned
+        local reference survives a concurrent aclose() that nulls the attribute,
+        closing the shutdown-during-request race (mirrors the sync path).
         """
-        Make a structured LLM completion with provider fallback.
-        
-        Priority:
-        1. LiteLLM (if available) - supports 100+ LLM providers
-        2. OpenAI SDK (fallback) - uses beta.chat.completions.parse
-        
-        Args:
-            response_model: Pydantic model class for structured output
-            messages: List of message dicts for the LLM
-            **kwargs: Additional arguments passed to the LLM
-            
-        Returns:
-            Instance of response_model with parsed response
-            
-        Raises:
-            ImportError: If neither litellm nor openai is installed
+        with self._client_lock:
+            client = self._async_openai_client
+            if client is None:
+                try:
+                    from openai import AsyncOpenAI
+                except ImportError as e:
+                    raise ImportError("Install with: pip install openai") from e
+                cfg = self.config_list[0]
+                client = self._async_openai_client = AsyncOpenAI(
+                    api_key=cfg.get("api_key") or os.environ.get("OPENAI_API_KEY"),
+                    base_url=cfg.get("base_url"),
+                )
+            return client
+
+    async def _completion_impl(
+        self,
+        response_model: Type[T],
+        messages: List[Dict],
+        *,
+        is_async: bool,
+        **kwargs,
+    ) -> T:
+        """Single source of truth for the structured-completion fallback ladder.
+
+        Drives the same LiteLLM -> OpenAI ladder for both the sync and async
+        entry points; the ONLY divergence is whether the provider call is
+        awaited (async) or run directly (sync), and which OpenAI client cache is
+        used. This collapses the previously byte-for-byte duplicated sync/async
+        paths into one place so the ladder can only ever be changed once.
         """
         model_name = self.config_list[0]['model']
-        
+
         # Try LiteLLM first (preferred - supports 100+ providers)
         if is_available("litellm"):
             try:
                 litellm = _get_litellm()
-                response = litellm.completion(
+                call = litellm.acompletion if is_async else litellm.completion
+                call_kwargs = dict(
                     model=model_name,
                     messages=messages,
                     response_format=response_model,
-                    **kwargs
+                    **kwargs,
+                )
+                # The async path awaits the provider coroutine directly. The sync
+                # path offloads the blocking provider call via asyncio.to_thread
+                # so it never blocks the bridge event loop this coroutine runs on.
+                response = (
+                    await _maybe_await(call(**call_kwargs))
+                    if is_async
+                    else await asyncio.to_thread(lambda: call(**call_kwargs))
                 )
                 content = response.choices[0].message.content
                 return response_model.model_validate_json(content)
@@ -504,97 +546,114 @@ class BaseAutoGenerator:
                     "LiteLLM structured completion failed (%s); "
                     "falling back to OpenAI SDK.", e
                 )
-        
+
         # Fallback to OpenAI SDK (uses beta.chat.completions.parse)
         if is_available("openai"):
-            client = self._get_openai_client()
-            response = client.beta.chat.completions.parse(
+            client = (
+                self._get_async_openai_client() if is_async
+                else self._get_openai_client()
+            )
+            parse = client.beta.chat.completions.parse
+            parse_kwargs = dict(
                 model=model_name,
                 messages=messages,
                 response_format=response_model,
-                **kwargs
+                **kwargs,
+            )
+            # As above: await directly when async, otherwise offload the blocking
+            # sync parse() so the bridge event loop is never blocked.
+            response = (
+                await _maybe_await(parse(**parse_kwargs))
+                if is_async
+                else await asyncio.to_thread(lambda: parse(**parse_kwargs))
             )
             return response.choices[0].message.parsed
-        
+
         # Neither available - raise helpful error
         raise ImportError(
             "Structured output requires either litellm or openai. "
             "Install with: pip install litellm  OR  pip install openai"
         )
-    
+
+    def _structured_completion(self, response_model: Type[T], messages: List[Dict], **kwargs) -> T:
+        """
+        Make a structured LLM completion with provider fallback.
+
+        Priority:
+        1. LiteLLM (if available) - supports 100+ LLM providers
+        2. OpenAI SDK (fallback) - uses beta.chat.completions.parse
+
+        Args:
+            response_model: Pydantic model class for structured output
+            messages: List of message dicts for the LLM
+            **kwargs: Additional arguments passed to the LLM
+
+        Returns:
+            Instance of response_model with parsed response
+
+        Raises:
+            ImportError: If neither litellm nor openai is installed
+        """
+        # Backward compatibility: the historical sync path called the blocking
+        # provider clients directly, so it worked even when invoked from inside a
+        # running event loop (FastAPI handler, Jupyter, async test). run_sync()
+        # rejects that case, so when a loop is already running we drive the sync
+        # ladder on a dedicated worker thread (with its own loop) instead of
+        # raising. The provider calls still execute off the caller's loop.
+        try:
+            asyncio.get_running_loop()
+        except RuntimeError:
+            running_loop = False
+        else:
+            running_loop = True
+
+        if not running_loop:
+            return run_sync(
+                self._completion_impl(response_model, messages, is_async=False, **kwargs)
+            )
+
+        result: List[Any] = []
+        error: List[BaseException] = []
+
+        def _runner() -> None:
+            try:
+                result.append(
+                    asyncio.run(
+                        self._completion_impl(
+                            response_model, messages, is_async=False, **kwargs
+                        )
+                    )
+                )
+            except BaseException as e:  # noqa: BLE001 - re-raised on caller thread
+                error.append(e)
+
+        thread = threading.Thread(target=_runner, name="praisonai-sync-completion")
+        thread.start()
+        thread.join()
+        if error:
+            raise error[0]
+        return result[0]
+
     async def _astructured_completion(self, response_model: Type[T], messages: List[Dict], **kwargs) -> T:
         """
         Make an async structured LLM completion with provider fallback.
-        
+
         Priority:
         1. LiteLLM async (if available) - supports 100+ LLM providers
         2. OpenAI AsyncSDK (fallback) - uses beta.chat.completions.parse
-        
+
         Args:
             response_model: Pydantic model class for structured output
             messages: List of message dicts for the LLM
             **kwargs: Additional arguments passed to the LLM
-            
+
         Returns:
             Instance of response_model with parsed response
-            
+
         Raises:
             ImportError: If neither litellm nor openai is installed
         """
-        model_name = self.config_list[0]['model']
-        
-        # Try LiteLLM async first (preferred - supports 100+ providers)
-        if is_available("litellm"):
-            try:
-                litellm = _get_litellm()
-                response = await litellm.acompletion(
-                    model=model_name,
-                    messages=messages,
-                    response_format=response_model,
-                    **kwargs
-                )
-                content = response.choices[0].message.content
-                return response_model.model_validate_json(content)
-            except Exception as e:
-                # Graceful execution-level fallback (see _structured_completion).
-                if not is_available("openai"):
-                    raise
-                logger.warning(
-                    "LiteLLM async structured completion failed (%s); "
-                    "falling back to OpenAI SDK.", e
-                )
-        
-        # Fallback to OpenAI AsyncSDK (uses beta.chat.completions.parse)
-        if is_available("openai"):
-            # Lazy-init AND snapshot the client under the same lock. The local
-            # reference survives a concurrent aclose() that nulls the attribute,
-            # closing the shutdown-during-request race.
-            with self._client_lock:
-                if self._async_openai_client is None:
-                    try:
-                        from openai import AsyncOpenAI
-                    except ImportError as e:
-                        raise ImportError("Install with: pip install openai") from e
-                    cfg = self.config_list[0]
-                    self._async_openai_client = AsyncOpenAI(
-                        api_key=cfg.get("api_key") or os.environ.get("OPENAI_API_KEY"),
-                        base_url=cfg.get("base_url"),
-                    )
-                client = self._async_openai_client
-            
-            response = await client.beta.chat.completions.parse(
-                model=model_name,
-                messages=messages,
-                response_format=response_model,
-                **kwargs
-            )
-            return response.choices[0].message.parsed
-        
-        # Neither available - raise helpful error
-        raise ImportError(
-            "Structured output requires either litellm or openai. "
-            "Install with: pip install litellm  OR  pip install openai"
-        )
+        return await self._completion_impl(response_model, messages, is_async=True, **kwargs)
     
     @staticmethod
     def get_available_tools() -> List[str]:

{praisonai-4.6.82 → praisonai-4.6.83}/praisonai/bots/__init__.py

@@ -31,6 +31,7 @@ if TYPE_CHECKING:
     from ._approval_store import ApprovalStore
     from ._delivery import DurableDelivery, deliver_with_retry
     from ._durable_adapter import DurableAdapterMixin
+    from ._outbound_resilience import OutboundResilienceMixin
     from ._outbound_messenger import BotOutboundMessenger
     from ._correlation import (
         correlation_id_from,
@@ -140,6 +141,9 @@ def __getattr__(name: str):
     if name == "DurableAdapterMixin":
         from ._durable_adapter import DurableAdapterMixin
         return DurableAdapterMixin
+    if name == "OutboundResilienceMixin":
+        from ._outbound_resilience import OutboundResilienceMixin
+        return OutboundResilienceMixin
     if name == "BotOutboundMessenger":
         from ._outbound_messenger import BotOutboundMessenger
         return BotOutboundMessenger
@@ -173,6 +177,7 @@ __all__ = [
     "ApprovalStore",
     "DurableDelivery", "deliver_with_retry", "deliver_chunked",
     "DurableAdapterMixin",
+    "OutboundResilienceMixin",
     "BotOutboundMessenger",
     # Correlation + metrics (gateway message-flow observability)
     "correlation_id_from", "current_correlation_id", "new_correlation_id",