PraisonAI 4.7.8__tar.gz → 4.7.10__tar.gz

{praisonai-4.7.8 → praisonai-4.7.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.7.8
+Version: 4.7.10
 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,14 @@ 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.7.6
-Requires-Dist: praisonai-code>=0.1.5
-Requires-Dist: praisonai-bot>=0.1.5
-Requires-Dist: praisonai-train>=0.1.5
-Requires-Dist: praisonai-browser>=0.1.5
-Requires-Dist: praisonai-mcp>=0.1.7
-Requires-Dist: praisonai-sandbox>=0.1.5
-Requires-Dist: praisonai-deploy>=0.1.5
+Requires-Dist: praisonaiagents>=1.7.8
+Requires-Dist: praisonai-code>=0.1.7
+Requires-Dist: praisonai-bot>=0.1.7
+Requires-Dist: praisonai-train>=0.1.7
+Requires-Dist: praisonai-browser>=0.1.7
+Requires-Dist: praisonai-mcp>=0.1.9
+Requires-Dist: praisonai-sandbox>=0.1.7
+Requires-Dist: praisonai-deploy>=0.1.7
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0
@@ -140,6 +140,8 @@ 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"
+Requires-Dist: a2a-sdk[http-server]>=1.0.0; extra == "flow"
+Requires-Dist: fastapi-pagination>=0.12.34; extra == "flow"
 Provides-Extra: lite
 Provides-Extra: all
 Requires-Dist: fastapi>=0.115.0; extra == "all"

{praisonai-4.7.8 → praisonai-4.7.10}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.7.8
+Version: 4.7.10
 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,14 @@ 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.7.6
-Requires-Dist: praisonai-code>=0.1.5
-Requires-Dist: praisonai-bot>=0.1.5
-Requires-Dist: praisonai-train>=0.1.5
-Requires-Dist: praisonai-browser>=0.1.5
-Requires-Dist: praisonai-mcp>=0.1.7
-Requires-Dist: praisonai-sandbox>=0.1.5
-Requires-Dist: praisonai-deploy>=0.1.5
+Requires-Dist: praisonaiagents>=1.7.8
+Requires-Dist: praisonai-code>=0.1.7
+Requires-Dist: praisonai-bot>=0.1.7
+Requires-Dist: praisonai-train>=0.1.7
+Requires-Dist: praisonai-browser>=0.1.7
+Requires-Dist: praisonai-mcp>=0.1.9
+Requires-Dist: praisonai-sandbox>=0.1.7
+Requires-Dist: praisonai-deploy>=0.1.7
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<2,>=1.83.14
 Requires-Dist: PyYAML>=6.0
@@ -140,6 +140,8 @@ 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"
+Requires-Dist: a2a-sdk[http-server]>=1.0.0; extra == "flow"
+Requires-Dist: fastapi-pagination>=0.12.34; extra == "flow"
 Provides-Extra: lite
 Provides-Extra: all
 Requires-Dist: fastapi>=0.115.0; extra == "all"

{praisonai-4.7.8 → praisonai-4.7.10}/PraisonAI.egg-info/SOURCES.txt

@@ -314,6 +314,7 @@ praisonai/integration/pages/bot_health.py
 praisonai/integration/pages/workflow_runs.py
 praisonai/integrations/__init__.py
 praisonai/integrations/_cli_loaders.py
+praisonai/integrations/_managed_base.py
 praisonai/integrations/_session_info.py
 praisonai/integrations/_tool_aliases.py
 praisonai/integrations/backend_registry.py

{praisonai-4.7.8 → praisonai-4.7.10}/PraisonAI.egg-info/requires.txt

@@ -1,14 +1,14 @@
 rich>=13.7
 markdown>=3.5
 pyparsing>=3.0.0
-praisonaiagents>=1.7.6
-praisonai-code>=0.1.5
-praisonai-bot>=0.1.5
-praisonai-train>=0.1.5
-praisonai-browser>=0.1.5
-praisonai-mcp>=0.1.7
-praisonai-sandbox>=0.1.5
-praisonai-deploy>=0.1.5
+praisonaiagents>=1.7.8
+praisonai-code>=0.1.7
+praisonai-bot>=0.1.7
+praisonai-train>=0.1.7
+praisonai-browser>=0.1.7
+praisonai-mcp>=0.1.9
+praisonai-sandbox>=0.1.7
+praisonai-deploy>=0.1.7
 python-dotenv>=0.19.0
 litellm<2,>=1.83.14
 PyYAML>=6.0
@@ -153,6 +153,8 @@ praisonai-sandbox[e2b]
 [flow]
 langflow>=1.0.0
 requests>=2.31.0
+a2a-sdk[http-server]>=1.0.0
+fastapi-pagination>=0.12.34
 
 [gateway]
 praisonai-bot[gateway]

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/_entrypoint.py

@@ -100,6 +100,14 @@ def run(agent_file: str,
     """
     from .agents_generator import AgentsGenerator
 
+    # Wire the built-in readers/retrievers/rerankers into the core-SDK
+    # registries so YAML/CLI declarations (e.g. `retriever: fusion`,
+    # `reranker: llm`) resolve. Idempotent + thread-safe, and register-only-if-
+    # absent: a custom adapter already registered under a built-in name (e.g. a
+    # multi-tenant host's own `fusion`) is preserved, never overwritten.
+    from .adapters import register_default_adapters
+    register_default_adapters()
+
     adapter, config_list = _resolve_run_inputs(framework)
     _apply_model_override(config_list, kwargs)
     cli_config = _merge_cli_config(cli_config, kwargs)
@@ -134,6 +142,14 @@ async def arun(agent_file: str,
 
     from .agents_generator import AgentsGenerator
 
+    # Wire the built-in readers/retrievers/rerankers into the core-SDK
+    # registries so YAML/CLI declarations resolve. Run off the event loop since
+    # registration performs lazy imports. Idempotent + thread-safe, and
+    # register-only-if-absent: a custom adapter already registered under a
+    # built-in name is preserved, never overwritten.
+    from .adapters import register_default_adapters
+    await asyncio.to_thread(register_default_adapters)
+
     # Resolve framework + build config_list off the caller's event loop so the
     # synchronous credential/config-file I/O does not block it. This is the
     # reason arun exists: a FastAPI handler awaiting arun must not stall the loop.

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/adapters/readers.py

@@ -389,25 +389,35 @@ def register_default_readers():
     from praisonaiagents.knowledge.readers import get_reader_registry
     
     registry = get_reader_registry()
-    
+
+    # Preserve any existing entry: an application (or a multi-tenant host) may
+    # have already registered a custom reader under a built-in name. Wiring the
+    # wrapper defaults must never silently replace it, so register only names
+    # that are not already present.
+    existing = set(registry.list_readers())
+
+    def _register(name, factory, extensions=None):
+        if name not in existing:
+            registry.register(name, factory, extensions)
+
     # Register text reader
-    registry.register("text", TextReader, ["txt", "text", "log"])
-    
+    _register("text", TextReader, ["txt", "text", "log"])
+
     # Register markitdown reader
-    registry.register("markitdown", MarkItDownReader, MarkItDownReader.supported_extensions)
-    
+    _register("markitdown", MarkItDownReader, MarkItDownReader.supported_extensions)
+
     # Register directory reader
-    registry.register("directory", DirectoryReader)
-    
+    _register("directory", DirectoryReader)
+
     # Register glob reader
-    registry.register("glob", GlobReader)
-    
+    _register("glob", GlobReader)
+
     # Register URL reader
-    registry.register("url", URLReader)
-    
+    _register("url", URLReader)
+
     # Register auto reader
-    registry.register("auto", AutoReader)
-    
+    _register("auto", AutoReader)
+
     logger.debug("Registered default readers")
 
 

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/adapters/rerankers.py

@@ -324,18 +324,25 @@ def register_default_rerankers():
     from praisonaiagents.knowledge.rerankers import get_reranker_registry
     
     registry = get_reranker_registry()
-    
+
+    # Preserve any existing entry: an application (or a multi-tenant host) may
+    # have already registered a custom reranker under a built-in name. Wiring
+    # the wrapper defaults must never silently replace it, so register only
+    # names that are not already present.
+    existing = set(registry.list_rerankers())
+
     # LLM reranker is always available
-    registry.register("llm", LLMReranker)
-    
+    if "llm" not in existing:
+        registry.register("llm", LLMReranker)
+
     # Cross-encoder if available
-    if _check_sentence_transformers():
+    if "cross_encoder" not in existing and _check_sentence_transformers():
         registry.register("cross_encoder", CrossEncoderReranker)
-    
+
     # Cohere if available
-    if _check_cohere():
+    if "cohere" not in existing and _check_cohere():
         registry.register("cohere", CohereReranker)
-    
+
     logger.debug("Registered default rerankers")
 
 

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/adapters/retrievers.py

@@ -595,13 +595,22 @@ def register_default_retrievers():
     from praisonaiagents.knowledge.retrieval import get_retriever_registry
     
     registry = get_retriever_registry()
-    
-    registry.register("basic", BasicRetriever)
-    registry.register("fusion", FusionRetriever)
-    registry.register("recursive", RecursiveRetriever)
-    registry.register("auto_merge", AutoMergeRetriever)
-    registry.register("hybrid", HybridRetriever)
-    
+
+    # Preserve any existing entry: an application (or a multi-tenant host) may
+    # have already registered a custom retriever under a built-in name. Wiring
+    # the wrapper defaults must never silently replace it, so register only
+    # names that are not already present.
+    existing = set(registry.list_retrievers())
+    for name, factory in (
+        ("basic", BasicRetriever),
+        ("fusion", FusionRetriever),
+        ("recursive", RecursiveRetriever),
+        ("auto_merge", AutoMergeRetriever),
+        ("hybrid", HybridRetriever),
+    ):
+        if name not in existing:
+            registry.register(name, factory)
+
     logger.debug("Registered default retrievers")
 
 

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/agents_generator.py

@@ -844,8 +844,11 @@ class AgentsGenerator:
         # Build tools dictionary using shared logic
         tools_dict = self._build_tools_dict(config)
         
-        # Select framework and resolve adapter variant
-        framework_name = self.framework or config.get('framework', 'praisonai')
+        # Select framework and resolve adapter variant. Registry is the single
+        # source of truth for the default, so no literal framework name here.
+        framework_name = self._adapter_registry.resolve_or_default(
+            self.framework or config.get('framework')
+        )
         adapter = self._select_framework(framework_name, config)
         
         # Validate framework availability through the injected registry so a
@@ -1376,10 +1379,13 @@ class AgentsGenerator:
         if self._is_workflow_yaml(config):
             # Bracket the workflow run in the same observability session the
             # sequential/hierarchical path uses so AgentOps init/finalize fires
-            # for workflow YAMLs too. Workflow YAMLs are native-only, so tag
-            # them 'praisonai'.
+            # for workflow YAMLs too. Tag with the config-resolved framework
+            # (registry default when omitted) rather than a hardcoded name.
             self._validate_workflow_cli_capabilities()
-            with observability_session("praisonai"):
+            workflow_label = self._adapter_registry.resolve_or_default(
+                self.framework or config.get('framework')
+            ).lower()
+            with observability_session(workflow_label):
                 return self._run_yaml_workflow(config)
 
         # Use shared preparation logic
@@ -1432,10 +1438,14 @@ class AgentsGenerator:
         if self._is_workflow_yaml(config):
             # Bracket the async workflow run in the same observability session
             # the sequential/hierarchical path uses so AgentOps init/finalize
-            # fires for workflow YAMLs too. Workflow YAMLs are native-only, so
-            # tag them 'praisonai'.
+            # fires for workflow YAMLs too. Tag with the config-resolved
+            # framework (registry default when omitted) rather than a hardcoded
+            # name.
             self._validate_workflow_cli_capabilities()
-            with observability_session("praisonai"):
+            workflow_label = self._adapter_registry.resolve_or_default(
+                self.framework or config.get('framework')
+            ).lower()
+            with observability_session(workflow_label):
                 return await self._arun_yaml_workflow(config)
 
         # Use shared preparation logic (off the event loop to avoid blocking imports)
@@ -1493,23 +1503,35 @@ class AgentsGenerator:
         )
         # Validate the YAML-declared framework first so a non-native workflow
         # YAML (e.g. framework: crewai) can't slip through just because the
-        # generator instance still holds the default 'praisonai'.
-        workflow_framework = framework_from_config(config)
+        # generator instance still holds the default 'praisonai'. Resolve
+        # through the injected registry (not the process default) so the
+        # build-time framework agrees with the observability label and a
+        # tenant-scoped adapter is honoured. Fall back to the process default
+        # when the instance was built without an injected registry.
+        registry = getattr(self, "_adapter_registry", None) or _get_default_adapter_registry()
+        workflow_framework = framework_from_config(
+            config, registry=registry
+        )
         validate_workflow_framework(
             workflow_framework,
             source="agents.yaml workflow section",
+            registry=registry,
         )
         if self.framework:
             validate_workflow_framework(
                 self.framework,
                 source="AgentsGenerator framework",
+                registry=registry,
             )
 
         # Parity with the sequential/hierarchical path: reject an incompatible
         # cli_backend in a workflow YAML at build time instead of accepting it
         # silently. Workflow YAMLs are native-only, so validate against
         # 'praisonai'.
-        self._validate_cli_backend_compatibility(config, workflow_framework or 'praisonai')
+        self._validate_cli_backend_compatibility(
+            config,
+            registry.resolve_or_default(workflow_framework),
+        )
 
         # The framework-level check above passes for 'praisonai' because the
         # sequential/hierarchical adapter does support cli_backend — but

{praisonai-4.7.8 → praisonai-4.7.10}/praisonai/api/agent_invoke.py

@@ -443,6 +443,42 @@ def _apply_agent_config(agent: Any, agent_config: Optional[Dict[str, Any]]) -> N
             )
 
 
+async def _safe_close_agent(agent: Any) -> None:
+    """Best-effort teardown of a per-request agent clone.
+
+    ``resolve_session_agent`` returns a fresh clone for every invoke of a
+    session-isolatable ``Agent`` (so concurrent callers never share
+    ``chat_history``). Those clones lazily spin up LLM client pools,
+    tool-executor threads and MCP subprocesses that
+    ``Agent.close()``/``aclose()`` release — without this teardown each request
+    leaks them into the long-lived server process.
+
+    Only a *clone* is closed. Ownership is derived from the agent itself via
+    :func:`_supports_session_isolation`: an isolatable agent was cloned for this
+    request and is safe to close, whereas a non-isolatable one (plain mock /
+    lightweight callable) is the shared registry instance and must NOT be closed
+    or the next request would run against a torn-down agent. Deriving ownership
+    from the resolved object — rather than comparing it against a second,
+    independently-timed registry lookup — is race-free even when the registry
+    entry is concurrently replaced.
+
+    Errors are swallowed (logged) so cleanup never masks a successful response.
+    """
+    if agent is None or not _supports_session_isolation(agent):
+        return
+    try:
+        aclose = getattr(agent, "aclose", None)
+        if inspect.iscoroutinefunction(aclose):
+            await aclose()
+        else:
+            close = getattr(agent, "close", None)
+            if callable(close):
+                import asyncio
+                await asyncio.to_thread(close)
+    except Exception:
+        logger.exception("Agent cleanup failed")
+
+
 def _supports_async_start(agent: Any) -> bool:
     """Return True if agent.astart is a coroutine function."""
     astart = getattr(agent, "astart", None)
@@ -511,52 +547,60 @@ if FASTAPI_AVAILABLE and APIRouter is not None:
                 status_code=404,
                 detail=f"Agent '{agent_id}' not found"
             )
-        
-        # Apply per-request agent_config overrides to the isolated clone, or
-        # reject unsupported keys with a 400 so the caller isn't silently
-        # running a different config than they asked for.
-        try:
-            _apply_agent_config(agent, request.agent_config)
-        except AgentConfigError as e:
-            raise HTTPException(status_code=400, detail=str(e))
 
         try:
-            # Session ID echoed back to the caller
-            session_id = request.session_id or "default"
-            
-            # Invoke agent (handle both sync and async agents)
-            if _supports_async_start(agent):
-                # Async agent
-                result = await agent.astart(request.message)
-            elif _supports_sync_start(agent):
-                # Sync agent - run in a worker thread to avoid blocking the event
-                # loop. ``asyncio.to_thread`` is the correct primitive inside a
-                # running request coroutine (``get_event_loop()`` is deprecated
-                # there since Python 3.10).
-                import asyncio
-                result = await asyncio.to_thread(agent.start, request.message)
-            else:
-                raise AttributeError(f"Agent {agent_id} must provide start() or async astart()")
-            
-            logger.info(f"Agent {agent_id} invoked successfully")
-            
-            return AgentInvokeResponse(
-                result=str(result),
-                session_id=session_id,
-                status="success",
-                metadata={
-                    "agent_id": agent_id,
-                    "message_length": len(request.message),
-                    "response_length": len(str(result))
-                }
-            )
-            
-        except Exception as e:
-            logger.error(f"Agent {agent_id} invocation failed: {e}")
-            raise HTTPException(
-                status_code=500,
-                detail=f"Agent execution failed: {str(e)}"
-            )
+            # Apply per-request agent_config overrides to the isolated clone, or
+            # reject unsupported keys with a 400 so the caller isn't silently
+            # running a different config than they asked for.
+            try:
+                _apply_agent_config(agent, request.agent_config)
+            except AgentConfigError as e:
+                raise HTTPException(status_code=400, detail=str(e))
+
+            try:
+                # Session ID echoed back to the caller
+                session_id = request.session_id or "default"
+
+                # Invoke agent (handle both sync and async agents)
+                if _supports_async_start(agent):
+                    # Async agent
+                    result = await agent.astart(request.message)
+                elif _supports_sync_start(agent):
+                    # Sync agent - run in a worker thread to avoid blocking the event
+                    # loop. ``asyncio.to_thread`` is the correct primitive inside a
+                    # running request coroutine (``get_event_loop()`` is deprecated
+                    # there since Python 3.10).
+                    import asyncio
+                    result = await asyncio.to_thread(agent.start, request.message)
+                else:
+                    raise AttributeError(f"Agent {agent_id} must provide start() or async astart()")
+
+                logger.info(f"Agent {agent_id} invoked successfully")
+
+                return AgentInvokeResponse(
+                    result=str(result),
+                    session_id=session_id,
+                    status="success",
+                    metadata={
+                        "agent_id": agent_id,
+                        "message_length": len(request.message),
+                        "response_length": len(str(result))
+                    }
+                )
+
+            except HTTPException:
+                raise
+            except Exception as e:
+                logger.error(f"Agent {agent_id} invocation failed: {e}")
+                raise HTTPException(
+                    status_code=500,
+                    detail=f"Agent execution failed: {str(e)}"
+                )
+        finally:
+            # Release the per-request clone's resources (LLM pools, tool-executor
+            # threads, MCP subprocesses). No-op when the agent fell back to the
+            # shared template. Never masks the response.
+            await _safe_close_agent(agent)
 
     @router.get("/agents")
     async def list_agents(
@@ -669,6 +713,7 @@ async def invoke_agent_standalone(
             "available_agents": list_registered_agents()
         }
 
+    agent = None
     try:
         # Per-session isolated agent view (see resolve_session_agent). Done
         # inside the try so a clone/isolation failure returns a clean error
@@ -715,6 +760,10 @@ async def invoke_agent_standalone(
             "status": "error",
             "agent_id": agent_id
         }
+    finally:
+        # Release the per-request clone's resources; no-op for the shared
+        # template fallback. Never masks the response.
+        await _safe_close_agent(agent)
 
 
 # Example usage and helper functions