PraisonAI 2.2.95__tar.gz → 2.2.97__tar.gz

{praisonai-2.2.95 → praisonai-2.2.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 2.2.95
+Version: 2.2.97
 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,7 +11,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=0.0.166
+Requires-Dist: praisonaiagents>=0.0.168
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: instructor>=1.3.3
 Requires-Dist: PyYAML>=6.0
@@ -119,3 +119,13 @@ python src/praisonai/scripts/bump_version.py 2.2.96
 # With praisonaiagents dependency
 python src/praisonai/scripts/bump_version.py 2.2.96 --agents 0.0.167
 ```
+
+## Release
+
+```bash
+# From project root (runs uv lock, build, git tag, gh release)
+python src/praisonai/scripts/release.py
+
+# Then publish
+cd src/praisonai && uv publish
+```

{praisonai-2.2.95 → praisonai-2.2.97}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 2.2.95
+Version: 2.2.97
 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,7 +11,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=0.0.166
+Requires-Dist: praisonaiagents>=0.0.168
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: instructor>=1.3.3
 Requires-Dist: PyYAML>=6.0
@@ -119,3 +119,13 @@ python src/praisonai/scripts/bump_version.py 2.2.96
 # With praisonaiagents dependency
 python src/praisonai/scripts/bump_version.py 2.2.96 --agents 0.0.167
 ```
+
+## Release
+
+```bash
+# From project root (runs uv lock, build, git tag, gh release)
+python src/praisonai/scripts/release.py
+
+# Then publish
+cd src/praisonai && uv publish
+```

{praisonai-2.2.95 → praisonai-2.2.97}/PraisonAI.egg-info/requires.txt

@@ -1,7 +1,7 @@
 rich>=13.7
 markdown>=3.5
 pyparsing>=3.0.0
-praisonaiagents>=0.0.166
+praisonaiagents>=0.0.168
 python-dotenv>=0.19.0
 instructor>=1.3.3
 PyYAML>=6.0

{praisonai-2.2.95 → praisonai-2.2.97}/README.md

@@ -22,3 +22,13 @@ python src/praisonai/scripts/bump_version.py 2.2.96
 # With praisonaiagents dependency
 python src/praisonai/scripts/bump_version.py 2.2.96 --agents 0.0.167
 ```
+
+## Release
+
+```bash
+# From project root (runs uv lock, build, git tag, gh release)
+python src/praisonai/scripts/release.py
+
+# Then publish
+cd src/praisonai && uv publish
+```

{praisonai-2.2.95 → praisonai-2.2.97}/praisonai/cli.py

@@ -514,9 +514,8 @@ class PraisonAI:
         in_test_env = (
             'pytest' in sys.argv[0] or 
             'unittest' in sys.argv[0] or
-            any('test' in arg for arg in sys.argv[1:3]) or  # Check first few args for test indicators
-            'pytest' in sys.modules or
-            'unittest' in sys.modules
+            any(arg.startswith('tests.') or arg.startswith('tests/') for arg in sys.argv[1:3]) or  # Check for test module paths
+            'PYTEST_CURRENT_TEST' in os.environ
         )
         
         # Check if we're being used as a library (not from praisonai CLI)
@@ -578,7 +577,8 @@ class PraisonAI:
         parser.add_argument("--goal", type=str, help="Goal for context engineering")
         parser.add_argument("--auto-analyze", action="store_true", help="Enable automatic analysis in context engineering")
         parser.add_argument("--research", action="store_true", help="Run deep research on a topic")
-        parser.add_argument("--query-rewrite", action="store_true", help="Rewrite query before research for better results")
+        parser.add_argument("--query-rewrite", action="store_true", help="Rewrite query for better results (works with any command)")
+        parser.add_argument("--rewrite-tools", type=str, help="Tools for query rewriter (e.g., 'internet_search' or path to tools.py)")
         parser.add_argument("--tools", "-t", type=str, help="Path to tools.py file for research agent")
         parser.add_argument("--save", "-s", action="store_true", help="Save research output to file (output/research/)")
         parser.add_argument("--verbose", "-v", action="store_true", help="Enable verbose output for research")
@@ -724,8 +724,9 @@ class PraisonAI:
                 verbose = getattr(args, 'verbose', False)
                 save = getattr(args, 'save', False)
                 query_rewrite = getattr(args, 'query_rewrite', False)
+                rewrite_tools = getattr(args, 'rewrite_tools', None)
                 tools_path = getattr(args, 'tools', None)
-                self.handle_research_command(research_query, research_model, verbose, save, query_rewrite, tools_path)
+                self.handle_research_command(research_query, research_model, verbose, save, query_rewrite, tools_path, rewrite_tools)
                 sys.exit(0)
 
         # Only check framework availability for agent-related operations
@@ -746,10 +747,105 @@ class PraisonAI:
 
         return args
 
+    def _rewrite_query(self, query: str, rewrite_tools: str = None, verbose: bool = False) -> str:
+        """
+        Rewrite query using QueryRewriterAgent.
+        
+        Args:
+            query: The query to rewrite
+            rewrite_tools: Tool names (comma-separated) or path to tools.py
+            verbose: Enable verbose output
+            
+        Returns:
+            Rewritten query or original if rewriting fails
+        """
+        try:
+            from praisonaiagents import QueryRewriterAgent, RewriteStrategy
+            from rich import print
+            
+            print("[bold cyan]Rewriting query for better results...[/bold cyan]")
+            
+            # Load rewrite tools if specified
+            rewrite_tools_list = []
+            if rewrite_tools:
+                if os.path.isfile(rewrite_tools):
+                    # Load from file
+                    try:
+                        import inspect
+                        import importlib.util
+                        spec = importlib.util.spec_from_file_location("rewrite_tools_module", rewrite_tools)
+                        if spec and spec.loader:
+                            module = importlib.util.module_from_spec(spec)
+                            spec.loader.exec_module(module)
+                            for name, obj in inspect.getmembers(module):
+                                if inspect.isfunction(obj) and not name.startswith('_'):
+                                    rewrite_tools_list.append(obj)
+                            if rewrite_tools_list:
+                                print(f"[cyan]Loaded {len(rewrite_tools_list)} tools for query rewriter[/cyan]")
+                    except Exception as e:
+                        print(f"[yellow]Warning: Failed to load rewrite tools: {e}[/yellow]")
+                else:
+                    # Treat as comma-separated tool names
+                    try:
+                        from praisonaiagents.tools import TOOL_MAPPINGS
+                        import praisonaiagents.tools as tools_module
+                        
+                        tool_names = [t.strip() for t in rewrite_tools.split(',')]
+                        for tool_name in tool_names:
+                            if tool_name in TOOL_MAPPINGS:
+                                try:
+                                    tool = getattr(tools_module, tool_name)
+                                    rewrite_tools_list.append(tool)
+                                except Exception as e:
+                                    print(f"[yellow]Warning: Failed to load rewrite tool '{tool_name}': {e}[/yellow]")
+                            else:
+                                print(f"[yellow]Warning: Unknown rewrite tool '{tool_name}'[/yellow]")
+                        if rewrite_tools_list:
+                            print(f"[cyan]Using rewrite tools: {', '.join(tool_names)}[/cyan]")
+                    except ImportError:
+                        print("[yellow]Warning: Could not import tools module[/yellow]")
+            
+            rewriter = QueryRewriterAgent(
+                model="gpt-4o-mini", 
+                verbose=verbose, 
+                tools=rewrite_tools_list if rewrite_tools_list else None
+            )
+            result = rewriter.rewrite(query, strategy=RewriteStrategy.AUTO)
+            rewritten = result.primary_query
+            
+            print(f"[cyan]Original:[/cyan] {query}")
+            print(f"[cyan]Rewritten:[/cyan] {rewritten}")
+            
+            return rewritten
+            
+        except ImportError:
+            from rich import print
+            print("[yellow]Warning: QueryRewriterAgent not available, using original query[/yellow]")
+            return query
+        except Exception as e:
+            from rich import print
+            print(f"[yellow]Warning: Query rewrite failed ({e}), using original query[/yellow]")
+            return query
+
+    def _rewrite_query_if_enabled(self, query: str) -> str:
+        """
+        Rewrite query using QueryRewriterAgent if --query-rewrite is enabled.
+        Returns the rewritten query or original if rewriting is disabled/fails.
+        """
+        if not hasattr(self, 'args') or not getattr(self.args, 'query_rewrite', False):
+            return query
+        
+        rewrite_tools = getattr(self.args, 'rewrite_tools', None)
+        verbose = getattr(self.args, 'verbose', False)
+        return self._rewrite_query(query, rewrite_tools, verbose)
+
     def handle_direct_prompt(self, prompt):
         """
         Handle direct prompt by creating a single agent and running it.
         """
+        # Apply query rewriting if enabled
+        prompt = self._rewrite_query_if_enabled(prompt)
+        
         if PRAISONAI_AVAILABLE:
             agent_config = {
                 "name": "DirectAgent",
@@ -965,7 +1061,7 @@ class PraisonAI:
             print(f"[red]ERROR: Context engineering failed: {e}[/red]")
             sys.exit(1)
 
-    def handle_research_command(self, query: str, model: str = None, verbose: bool = False, save: bool = False, query_rewrite: bool = False, tools_path: str = None) -> str:
+    def handle_research_command(self, query: str, model: str = None, verbose: bool = False, save: bool = False, query_rewrite: bool = False, tools_path: str = None, rewrite_tools: str = None) -> str:
         """
         Handle the research command by creating a DeepResearchAgent and running it.
         
@@ -976,6 +1072,7 @@ class PraisonAI:
             save: Save output to file (default: False)
             query_rewrite: Rewrite query before research (default: False)
             tools_path: Path to tools.py file with custom tools (default: None)
+            rewrite_tools: Tools for query rewriter (tool names or file path)
             
         Returns:
             str: Research report
@@ -991,20 +1088,8 @@ class PraisonAI:
                 logging.getLogger('httpcore').setLevel(logging.WARNING)
             
             # Rewrite query if requested
-            original_query = query
             if query_rewrite:
-                try:
-                    from praisonaiagents import QueryRewriterAgent, RewriteStrategy
-                    print("[bold cyan]Rewriting query for better research results...[/bold cyan]")
-                    rewriter = QueryRewriterAgent(model="gpt-4o-mini", verbose=False)
-                    result = rewriter.rewrite(query, strategy=RewriteStrategy.AUTO)
-                    query = result.primary_query
-                    print(f"[cyan]Original:[/cyan] {original_query}")
-                    print(f"[cyan]Rewritten:[/cyan] {query}")
-                except ImportError:
-                    print("[yellow]Warning: QueryRewriterAgent not available, using original query[/yellow]")
-                except Exception as e:
-                    print(f"[yellow]Warning: Query rewrite failed ({e}), using original query[/yellow]")
+                query = self._rewrite_query(query, rewrite_tools, verbose)
             
             print("[bold green]Starting Deep Research...[/bold green]")
             print(f"Query: {query}")

{praisonai-2.2.95 → praisonai-2.2.97}/praisonai/deploy.py

@@ -57,7 +57,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==2.2.95 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==2.2.97 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai-2.2.97/praisonai/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.97"

{praisonai-2.2.95 → praisonai-2.2.97}/pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
     "rich>=13.7",
     "markdown>=3.5",
     "pyparsing>=3.0.0",
-    "praisonaiagents>=0.0.166",
+    "praisonaiagents>=0.0.168",
     "python-dotenv>=0.19.0",
     "instructor>=1.3.3",
     "PyYAML>=6.0",

praisonai-2.2.95/praisonai/version.py

@@ -1 +0,0 @@
-__version__ = "2.2.95"