praisonAI 2.5.2__tar.gz → 2.5.4__tar.gz

{praisonai-2.5.2 → praisonai-2.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 2.5.2
+Version: 2.5.4
 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.3.2
+Requires-Dist: praisonaiagents>=0.3.4
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: instructor>=1.3.3
 Requires-Dist: PyYAML>=6.0

{praisonai-2.5.2 → praisonai-2.5.4}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 2.5.2
+Version: 2.5.4
 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.3.2
+Requires-Dist: praisonaiagents>=0.3.4
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: instructor>=1.3.3
 Requires-Dist: PyYAML>=6.0

{praisonai-2.5.2 → praisonai-2.5.4}/PraisonAI.egg-info/requires.txt

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

{praisonai-2.5.2 → praisonai-2.5.4}/praisonai/adapters/rerankers.py

@@ -57,13 +57,10 @@ class LLMReranker:
         self._client = None
     
     def _get_client(self):
-        """Lazy load OpenAI client."""
-        if self._client is None:
-            if self.llm:
-                return self.llm
-            import openai
-            self._client = openai.OpenAI()
-        return self._client
+        """Get LLM client - returns self.llm if provided, otherwise None (use litellm)."""
+        if self.llm:
+            return self.llm
+        return None  # Will use litellm directly
     
     def rerank(
         self,
@@ -118,7 +115,7 @@ class LLMReranker:
         return results
     
     def _score_document(self, query: str, document: str) -> float:
-        """Score a single document's relevance to query."""
+        """Score a single document's relevance to query using litellm."""
         try:
             client = self._get_client()
             
@@ -131,18 +128,21 @@ Document: {document[:1000]}
 
 Relevance score (0-10):"""
             
-            if hasattr(client, 'chat'):
-                # PraisonAI Agent or similar
+            # Check if client has a callable 'chat' method (PraisonAI Agent)
+            if client is not None and hasattr(client, 'chat') and callable(getattr(client, 'chat', None)):
+                # PraisonAI Agent or similar with callable chat method
                 response = client.chat(prompt)
             else:
-                # OpenAI client
-                response = client.chat.completions.create(
+                # Use litellm for unified multi-provider support
+                import litellm
+                
+                # litellm handles model-specific quirks automatically
+                completion = litellm.completion(
                     model=self.model,
                     messages=[{"role": "user", "content": prompt}],
-                    max_tokens=10,
-                    temperature=0
+                    max_tokens=10
                 )
-                response = response.choices[0].message.content
+                response = completion.choices[0].message.content
             
             # Parse score
             if response:

{praisonai-2.5.2 → praisonai-2.5.4}/praisonai/adapters/vector_stores.py

@@ -107,7 +107,17 @@ class ChromaVectorStore:
     ) -> List[str]:
         """Add vectors to ChromaDB."""
         ids = ids or [str(uuid_module.uuid4()) for _ in texts]
-        metadatas = metadatas or [{} for _ in texts]
+        
+        # ChromaDB requires non-empty metadata dicts
+        # Ensure each metadata dict has at least one key
+        if metadatas is None:
+            metadatas = [{"_source": "praisonai"} for _ in texts]
+        else:
+            # Ensure no empty dicts - add placeholder if empty
+            metadatas = [
+                m if m else {"_source": "praisonai"} 
+                for m in metadatas
+            ]
         
         # Use different collection if namespace specified
         if namespace and namespace != self.namespace:

{praisonai-2.5.2 → praisonai-2.5.4}/praisonai/chat/__init__.py

@@ -79,24 +79,16 @@ def start_chat_server(
             _register_agent(a)
     
     # Start the chainlit server
-    import sys
     from pathlib import Path
     
     # Use the built-in app file
     app_file = Path(__file__).parent / "app.py"
     
-    sys.argv = [
-        "chainlit",
-        "run",
-        str(app_file),
-        "--host", config.host,
-        "--port", str(config.port),
-    ]
+    # Set environment variables for chainlit
+    os.environ["CHAINLIT_HOST"] = config.host
+    os.environ["CHAINLIT_PORT"] = str(config.port)
     
-    if config.debug:
-        sys.argv.append("-w")
-    
-    run_chainlit()
+    run_chainlit(str(app_file))
 
 
 # Global registry for agents

{praisonai-2.5.2 → praisonai-2.5.4}/praisonai/cli/features/knowledge.py

@@ -68,7 +68,7 @@ class KnowledgeHandler(CommandHandler):
         return "knowledge"
     
     def get_actions(self) -> List[str]:
-        return ["add", "query", "search", "list", "clear", "stats", "info", "help"]
+        return ["add", "query", "search", "list", "clear", "stats", "info", "export", "import", "help"]
     
     def get_help_text(self) -> str:
         return """
@@ -78,6 +78,8 @@ Knowledge Commands:
   praisonai knowledge list                 - List indexed documents
   praisonai knowledge clear                - Clear knowledge base
   praisonai knowledge stats                - Show knowledge base statistics
+  praisonai knowledge export <file.json>   - Export knowledge base to JSON file
+  praisonai knowledge import <file.json>   - Import knowledge base from JSON file
   praisonai knowledge help                 - Show this help message
 
 Options:
@@ -380,6 +382,150 @@ Examples:
         """Alias for action_stats."""
         return self.action_stats(args, **kwargs)
     
+    def action_export(self, args: List[str], **kwargs) -> bool:
+        """
+        Export knowledge base to JSON file.
+        
+        Exports all documents with their metadata, embeddings, and content
+        for backup or migration purposes.
+        
+        Args:
+            args: List containing output file path
+            
+        Returns:
+            True if successful
+        """
+        import json
+        from datetime import datetime
+        
+        if not args:
+            # Default filename with timestamp
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            output_file = os.path.join(self.workspace, f"knowledge_export_{timestamp}.json")
+        else:
+            output_file = args[0]
+            if not os.path.isabs(output_file):
+                output_file = os.path.join(self.workspace, output_file)
+        
+        knowledge = self._get_knowledge()
+        if not knowledge:
+            return False
+        
+        try:
+            export_data = {
+                "version": "1.0",
+                "exported_at": datetime.now().isoformat(),
+                "workspace": self.workspace,
+                "vector_store": self.vector_store,
+                "documents": []
+            }
+            
+            # Try to get all documents from knowledge/memory
+            if hasattr(knowledge, 'memory') and hasattr(knowledge.memory, 'get_all'):
+                all_docs = knowledge.memory.get_all()
+                if all_docs:
+                    for doc in all_docs:
+                        if isinstance(doc, dict):
+                            export_data["documents"].append(doc)
+                        else:
+                            export_data["documents"].append({
+                                "content": str(doc),
+                                "metadata": {}
+                            })
+            elif hasattr(knowledge, 'get_all'):
+                all_docs = knowledge.get_all()
+                if all_docs:
+                    for doc in all_docs:
+                        if isinstance(doc, dict):
+                            export_data["documents"].append(doc)
+                        else:
+                            export_data["documents"].append({
+                                "content": str(doc),
+                                "metadata": {}
+                            })
+            
+            # Ensure output directory exists
+            os.makedirs(os.path.dirname(output_file) or ".", exist_ok=True)
+            
+            with open(output_file, 'w', encoding='utf-8') as f:
+                json.dump(export_data, f, indent=2, default=str)
+            
+            doc_count = len(export_data["documents"])
+            self.print_status(f"✅ Exported {doc_count} documents to {output_file}", "success")
+            return True
+            
+        except Exception as e:
+            self.print_status(f"Export failed: {e}", "error")
+            return False
+    
+    def action_import(self, args: List[str], **kwargs) -> bool:
+        """
+        Import knowledge base from JSON file.
+        
+        Imports documents from a previously exported JSON file.
+        
+        Args:
+            args: List containing input file path
+            
+        Returns:
+            True if successful
+        """
+        import json
+        
+        if not args:
+            self.print_status("Usage: praisonai knowledge import <file.json>", "error")
+            return False
+        
+        input_file = args[0]
+        if not os.path.isabs(input_file):
+            input_file = os.path.join(self.workspace, input_file)
+        
+        if not os.path.exists(input_file):
+            self.print_status(f"File not found: {input_file}", "error")
+            return False
+        
+        knowledge = self._get_knowledge()
+        if not knowledge:
+            return False
+        
+        try:
+            with open(input_file, 'r', encoding='utf-8') as f:
+                import_data = json.load(f)
+            
+            # version field reserved for future schema migrations
+            _ = import_data.get("version", "unknown")
+            documents = import_data.get("documents", [])
+            
+            if not documents:
+                self.print_status("No documents found in import file", "warning")
+                return True
+            
+            imported_count = 0
+            for doc in documents:
+                try:
+                    content = doc.get("content") or doc.get("memory") or doc.get("text", "")
+                    metadata = doc.get("metadata", {})
+                    
+                    if content:
+                        if hasattr(knowledge, 'store'):
+                            knowledge.store(content, metadata=metadata)
+                        elif hasattr(knowledge, 'add'):
+                            knowledge.add(content, metadata=metadata)
+                        imported_count += 1
+                except Exception as e:
+                    if self.verbose:
+                        self.print_status(f"Skipped document: {e}", "warning")
+            
+            self.print_status(f"✅ Imported {imported_count} documents from {input_file}", "success")
+            return True
+            
+        except json.JSONDecodeError as e:
+            self.print_status(f"Invalid JSON file: {e}", "error")
+            return False
+        except Exception as e:
+            self.print_status(f"Import failed: {e}", "error")
+            return False
+    
     def execute(self, action: str, action_args: List[str], **kwargs) -> Any:
         """Execute knowledge command action."""
         return super().execute(action, action_args, **kwargs)

{praisonai-2.5.2 → praisonai-2.5.4}/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.5.2 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==2.5.4 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai-2.5.4/praisonai/version.py

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

{praisonai-2.5.2 → praisonai-2.5.4}/pyproject.toml

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

praisonai-2.5.2/praisonai/version.py

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