praisonaiagents 0.0.99__tar.gz → 0.0.100__tar.gz

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.99
+Version: 0.0.100
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/praisonaiagents/agent/agent.py

@@ -619,12 +619,9 @@ Your Goal: {self.goal}
         while retry_count <= self.max_guardrail_retries:
             # Create TaskOutput object
             task_output = TaskOutput(
+                description="Agent response output",
                 raw=current_response,
-                output=current_response,
-                pydantic=None,
-                json_dict=None,
-                name=f"{self.name}_output",
-                description="Agent response output"
+                agent=self.name
             )
             
             # Process guardrail

praisonaiagents-0.0.100/praisonaiagents/guardrails/__init__.py

@@ -0,0 +1,11 @@
+"""
+Guardrails module for PraisonAI Agents.
+
+This module provides validation and safety mechanisms for task outputs,
+including both function-based and LLM-based guardrails.
+"""
+
+from .guardrail_result import GuardrailResult
+from .llm_guardrail import LLMGuardrail
+
+__all__ = ["GuardrailResult", "LLMGuardrail"]

praisonaiagents-0.0.100/praisonaiagents/guardrails/guardrail_result.py

@@ -0,0 +1,43 @@
+"""
+Guardrail result classes for PraisonAI Agents.
+
+This module provides the result types for guardrail validation,
+following the same pattern as CrewAI for consistency.
+"""
+
+from typing import Any, Tuple, Union
+from pydantic import BaseModel, Field
+from ..main import TaskOutput
+
+
+class GuardrailResult(BaseModel):
+    """Result of a guardrail validation."""
+    
+    success: bool = Field(description="Whether the guardrail check passed")
+    result: Union[str, TaskOutput, None] = Field(description="The result if modified, or None if unchanged")
+    error: str = Field(default="", description="Error message if validation failed")
+    
+    @classmethod
+    def from_tuple(cls, result: Tuple[bool, Any]) -> "GuardrailResult":
+        """Create a GuardrailResult from a tuple returned by a guardrail function.
+        
+        Args:
+            result: Tuple of (success, result_or_error)
+            
+        Returns:
+            GuardrailResult: The structured result
+        """
+        success, data = result
+        
+        if success:
+            return cls(
+                success=True,
+                result=data,
+                error=""
+            )
+        else:
+            return cls(
+                success=False,
+                result=None,
+                error=str(data) if data else "Guardrail validation failed"
+            )

praisonaiagents-0.0.100/praisonaiagents/guardrails/llm_guardrail.py

@@ -0,0 +1,88 @@
+"""
+LLM-based guardrail implementation for PraisonAI Agents.
+
+This module provides LLM-powered guardrails that can validate task outputs
+using natural language descriptions, similar to CrewAI's implementation.
+"""
+
+import logging
+from typing import Any, Tuple, Union, Optional
+from pydantic import BaseModel
+from ..main import TaskOutput
+
+
+class LLMGuardrail:
+    """An LLM-powered guardrail that validates task outputs using natural language."""
+    
+    def __init__(self, description: str, llm: Any = None):
+        """Initialize the LLM guardrail.
+        
+        Args:
+            description: Natural language description of what to validate
+            llm: The LLM instance to use for validation
+        """
+        self.description = description
+        self.llm = llm
+        self.logger = logging.getLogger(__name__)
+    
+    def __call__(self, task_output: TaskOutput) -> Tuple[bool, Union[str, TaskOutput]]:
+        """Validate the task output using the LLM.
+        
+        Args:
+            task_output: The task output to validate
+            
+        Returns:
+            Tuple of (success, result) where result is the output or error message
+        """
+        try:
+            if not self.llm:
+                self.logger.warning("No LLM provided for guardrail validation")
+                return True, task_output
+            
+            # Create validation prompt
+            validation_prompt = f"""
+You are a quality assurance validator. Your task is to evaluate the following output against specific criteria.
+
+Validation Criteria: {self.description}
+
+Output to Validate:
+{task_output.raw}
+
+Please evaluate if this output meets the criteria. Respond with:
+1. "PASS" if the output meets all criteria
+2. "FAIL: [specific reason]" if the output does not meet criteria
+
+Your response:"""
+
+            # Get LLM response
+            if hasattr(self.llm, 'chat'):
+                # For Agent's LLM interface
+                response = self.llm.chat(validation_prompt, temperature=0.1)
+            elif hasattr(self.llm, 'get_response'):
+                # For custom LLM instances
+                response = self.llm.get_response(validation_prompt, temperature=0.1)
+            elif callable(self.llm):
+                # For simple callable LLMs
+                response = self.llm(validation_prompt)
+            else:
+                self.logger.error(f"Unsupported LLM type: {type(self.llm)}")
+                return True, task_output
+            
+            # Parse response
+            response = str(response).strip()
+            
+            if response.upper().startswith("PASS"):
+                return True, task_output
+            elif response.upper().startswith("FAIL"):
+                # Extract the reason
+                reason = response[5:].strip(": ")
+                return False, f"Guardrail validation failed: {reason}"
+            else:
+                # Unclear response, log and pass through
+                self.logger.warning(f"Unclear guardrail response: {response}")
+                return True, task_output
+                
+        except Exception as e:
+            self.logger.error(f"Error in LLM guardrail validation: {str(e)}")
+            # On error, pass through the original output
+            return True, task_output

praisonaiagents-0.0.100/praisonaiagents/memory/__init__.py

@@ -0,0 +1,15 @@
+"""
+Memory module for PraisonAI Agents
+
+This module provides memory management capabilities including:
+- Short-term memory (STM) for ephemeral context
+- Long-term memory (LTM) for persistent knowledge  
+- Entity memory for structured data
+- User memory for preferences/history
+- Quality-based storage decisions
+- Graph memory support via Mem0
+"""
+
+from .memory import Memory
+
+__all__ = ["Memory"] 

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/praisonaiagents/memory/memory.py

@@ -910,11 +910,14 @@ class Memory:
         """
 
         try:
-            # Use OpenAI client from main.py
-            from ..main import client
+            # Use LiteLLM for consistency with the rest of the codebase
+            import litellm
             
-            response = client.chat.completions.create(
-                model=llm or "gpt-4o",
+            # Convert model name if it's in litellm format
+            model_name = llm or "gpt-4o-mini"
+            
+            response = litellm.completion(
+                model=model_name,
                 messages=[{
                     "role": "user", 
                     "content": custom_prompt or default_prompt

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/praisonaiagents/task/task.py

@@ -308,7 +308,11 @@ class Task:
                 if self.agent:
                     if getattr(self.agent, '_using_custom_llm', False) and hasattr(self.agent, 'llm_instance'):
                         # For custom LLM instances (like Ollama)
-                        llm_model = self.agent.llm_instance
+                        # Extract the model name from the LLM instance
+                        if hasattr(self.agent.llm_instance, 'model'):
+                            llm_model = self.agent.llm_instance.model
+                        else:
+                            llm_model = "gpt-4o-mini"  # Default fallback
                     elif hasattr(self.agent, 'llm') and self.agent.llm:
                         # For standard model strings
                         llm_model = self.agent.llm

praisonaiagents-0.0.100/praisonaiagents/tools/searxng_tools.py

@@ -0,0 +1,94 @@
+"""SearxNG search functionality.
+
+Usage:
+from praisonaiagents.tools import searxng_search
+results = searxng_search("AI news")
+
+or 
+from praisonaiagents.tools import searxng
+results = searxng("AI news")
+"""
+
+from typing import List, Dict, Optional
+import logging
+from importlib import util
+
+def searxng_search(
+    query: str, 
+    max_results: int = 5,
+    searxng_url: Optional[str] = None
+) -> List[Dict]:
+    """Perform an internet search using SearxNG instance.
+    
+    Args:
+        query: Search query string
+        max_results: Maximum number of results to return
+        searxng_url: SearxNG instance URL (defaults to localhost:32768)
+        
+    Returns:
+        List[Dict]: Search results with title, url, and snippet keys
+                   Returns error dict on failure
+    """
+    # Check if requests is available
+    if util.find_spec("requests") is None:
+        error_msg = "SearxNG search requires requests package. Install with: pip install requests"
+        logging.error(error_msg)
+        return [{"error": error_msg}]
+    
+    try:
+        import requests
+        
+        # Default URL for local SearxNG instance
+        url = searxng_url or "http://localhost:32768/search"
+        
+        params = {
+            'q': query, 
+            'format': 'json',
+            'engines': 'google,bing,duckduckgo',  # Multiple engines
+            'safesearch': '1'  # Safe search enabled
+        }
+        
+        response = requests.get(url, params=params, timeout=10)
+        response.raise_for_status()
+        
+        raw_results = response.json().get('results', [])
+        
+        # Standardize to PraisonAI format
+        results = []
+        for i, result in enumerate(raw_results[:max_results]):
+            results.append({
+                "title": result.get("title", ""),
+                "url": result.get("url", ""),
+                "snippet": result.get("content", "")
+            })
+            
+        return results
+        
+    except requests.exceptions.ConnectionError:
+        error_msg = f"Could not connect to SearxNG at {url}. Ensure SearxNG is running."
+        logging.error(error_msg)
+        return [{"error": error_msg}]
+    except requests.exceptions.Timeout:
+        error_msg = "SearxNG search request timed out"
+        logging.error(error_msg)
+        return [{"error": error_msg}]
+    except requests.exceptions.RequestException as e:
+        error_msg = f"SearxNG search request failed: {e}"
+        logging.error(error_msg)
+        return [{"error": error_msg}]
+    except (ValueError, KeyError) as e:
+        error_msg = f"Error parsing SearxNG response: {e}"
+        logging.error(error_msg)
+        return [{"error": error_msg}]
+
+def searxng(query: str, max_results: int = 5, searxng_url: Optional[str] = None) -> List[Dict]:
+    """Alias for searxng_search function."""
+    return searxng_search(query, max_results, searxng_url)
+
+if __name__ == "__main__":
+    # Example usage
+    results = searxng_search("Python programming")
+    for result in results:
+        print(f"\nTitle: {result.get('title')}")
+        print(f"URL: {result.get('url')}")
+        print(f"Snippet: {result.get('snippet')}")

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/praisonaiagents.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.99
+Version: 0.0.100
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/praisonaiagents.egg-info/SOURCES.txt

@@ -15,6 +15,9 @@ praisonaiagents/agent/image_agent.py
 praisonaiagents/agents/__init__.py
 praisonaiagents/agents/agents.py
 praisonaiagents/agents/autoagents.py
+praisonaiagents/guardrails/__init__.py
+praisonaiagents/guardrails/guardrail_result.py
+praisonaiagents/guardrails/llm_guardrail.py
 praisonaiagents/knowledge/__init__.py
 praisonaiagents/knowledge/chunking.py
 praisonaiagents/knowledge/knowledge.py
@@ -23,6 +26,7 @@ praisonaiagents/llm/llm.py
 praisonaiagents/mcp/__init__.py
 praisonaiagents/mcp/mcp.py
 praisonaiagents/mcp/mcp_sse.py
+praisonaiagents/memory/__init__.py
 praisonaiagents/memory/memory.py
 praisonaiagents/process/__init__.py
 praisonaiagents/process/process.py
@@ -40,6 +44,7 @@ praisonaiagents/tools/json_tools.py
 praisonaiagents/tools/newspaper_tools.py
 praisonaiagents/tools/pandas_tools.py
 praisonaiagents/tools/python_tools.py
+praisonaiagents/tools/searxng_tools.py
 praisonaiagents/tools/shell_tools.py
 praisonaiagents/tools/spider_tools.py
 praisonaiagents/tools/test.py

{praisonaiagents-0.0.99 → praisonaiagents-0.0.100}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "praisonaiagents"
-version = "0.0.99"
+version = "0.0.100"
 description = "Praison AI agents for completing complex tasks with Self Reflection Agents"
 requires-python = ">=3.10"
 authors = [
@@ -63,5 +63,6 @@ all = [
     "praisonaiagents[api]"
 ]
 
-[tool.setuptools]
-packages = ["praisonaiagents"]
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["praisonaiagents*"]