PraisonAI 2.2.64__tar.gz → 2.2.65__tar.gz

{praisonai-2.2.64 → praisonai-2.2.65}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: PraisonAI
-Version: 2.2.64
+Version: 2.2.65
 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
 Requires-Python: >=3.10
@@ -12,6 +12,7 @@ Provides-Extra: agentops
 Provides-Extra: anthropic
 Provides-Extra: api
 Provides-Extra: autogen
+Provides-Extra: autogen-v4
 Provides-Extra: call
 Provides-Extra: chat
 Provides-Extra: code
@@ -29,6 +30,9 @@ Requires-Dist: aiosqlite (>=0.20.0) ; extra == "chat"
 Requires-Dist: aiosqlite (>=0.20.0) ; extra == "code"
 Requires-Dist: aiosqlite (>=0.20.0) ; extra == "realtime"
 Requires-Dist: aiosqlite (>=0.20.0) ; extra == "ui"
+Requires-Dist: autogen-agentchat (>=0.4.0) ; extra == "autogen-v4"
+Requires-Dist: autogen-core (>=0.4.0) ; extra == "autogen-v4"
+Requires-Dist: autogen-ext[openai] (>=0.4.0) ; extra == "autogen-v4"
 Requires-Dist: chainlit (==2.5.5) ; extra == "chat"
 Requires-Dist: chainlit (==2.5.5) ; extra == "code"
 Requires-Dist: chainlit (==2.5.5) ; extra == "realtime"
@@ -38,6 +42,7 @@ Requires-Dist: crawl4ai (>=0.6.0) ; extra == "code"
 Requires-Dist: crawl4ai (>=0.6.0) ; extra == "realtime"
 Requires-Dist: crewai (>=0.32.0) ; extra == "crewai"
 Requires-Dist: crewai ; extra == "autogen"
+Requires-Dist: crewai ; extra == "autogen-v4"
 Requires-Dist: duckduckgo_search (>=6.3.0) ; extra == "realtime"
 Requires-Dist: fastapi (>=0.115.0) ; extra == "api"
 Requires-Dist: fastapi (>=0.95.0) ; extra == "call"
@@ -63,6 +68,7 @@ Requires-Dist: playwright (>=1.47.0) ; extra == "chat"
 Requires-Dist: playwright (>=1.47.0) ; extra == "code"
 Requires-Dist: plotly (>=5.24.0) ; extra == "realtime"
 Requires-Dist: praisonai-tools (>=0.0.22) ; extra == "autogen"
+Requires-Dist: praisonai-tools (>=0.0.22) ; extra == "autogen-v4"
 Requires-Dist: praisonai-tools (>=0.0.22) ; extra == "crewai"
 Requires-Dist: praisonaiagents (>=0.0.136)
 Requires-Dist: pyautogen (==0.2.29) ; extra == "autogen"

{praisonai-2.2.64 → praisonai-2.2.65}/praisonai/agents_generator.py

@@ -14,10 +14,13 @@ import importlib
 import importlib.util
 import os
 import logging
+import re
+import keyword
 
 # Framework-specific imports with availability checks
 CREWAI_AVAILABLE = False
 AUTOGEN_AVAILABLE = False
+AUTOGEN_V4_AVAILABLE = False
 PRAISONAI_TOOLS_AVAILABLE = False
 AGENTOPS_AVAILABLE = False
 PRAISONAI_AVAILABLE = False
@@ -41,6 +44,17 @@ try:
 except ImportError:
     pass
 
+try:
+    from autogen_agentchat.agents import AssistantAgent as AutoGenV4AssistantAgent
+    from autogen_ext.models.openai import OpenAIChatCompletionClient
+    from autogen_agentchat.teams import RoundRobinGroupChat
+    from autogen_agentchat.conditions import TextMentionTermination, MaxMessageTermination
+    from autogen_agentchat.messages import TextMessage
+    from autogen_core import CancellationToken
+    AUTOGEN_V4_AVAILABLE = True
+except ImportError:
+    pass
+
 try:
     import agentops
     AGENTOPS_AVAILABLE = True
@@ -70,6 +84,40 @@ os.environ["OTEL_SDK_DISABLED"] = "true"
 def noop(*args, **kwargs):
     pass
 
+def sanitize_agent_name_for_autogen_v4(name):
+    """
+    Sanitize agent name to be a valid Python identifier for AutoGen v0.4.
+    
+    Args:
+        name (str): The original agent name
+        
+    Returns:
+        str: A valid Python identifier
+    """
+    # Convert to string and replace invalid characters with underscores
+    sanitized = re.sub(r'[^a-zA-Z0-9_]', '_', str(name))
+    
+    # Collapse only very excessive underscores (5 or more) to reduce extreme cases
+    sanitized = re.sub(r'_{5,}', '_', sanitized)
+    
+    # Remove trailing underscores only if not part of a dunder pattern and only if singular
+    if sanitized.endswith('_') and not sanitized.endswith('__') and sanitized != '_':
+        sanitized = sanitized.rstrip('_')
+    
+    # Ensure it starts with a letter or underscore (not a digit)
+    if sanitized and sanitized[0].isdigit():
+        sanitized = 'agent_' + sanitized
+    
+    # Handle empty string or only invalid characters (including single underscore from all invalid chars)
+    if not sanitized or sanitized == '_':
+        sanitized = 'agent'
+    
+    # Check if it's a Python keyword and append underscore if so
+    if keyword.iskeyword(sanitized):
+        sanitized += '_'
+    
+    return sanitized
+
 def disable_crewai_telemetry():
     if CREWAI_AVAILABLE:
         for attr in dir(Telemetry):
@@ -123,8 +171,8 @@ class AgentsGenerator:
         # Validate framework availability
         if framework == "crewai" and not CREWAI_AVAILABLE:
             raise ImportError("CrewAI is not installed. Please install it with 'pip install praisonai[crewai]'")
-        elif framework == "autogen" and not AUTOGEN_AVAILABLE:
-            raise ImportError("AutoGen is not installed. Please install it with 'pip install praisonai[autogen]'")
+        elif framework == "autogen" and not (AUTOGEN_AVAILABLE or AUTOGEN_V4_AVAILABLE):
+            raise ImportError("AutoGen is not installed. Please install it with 'pip install praisonai[autogen]' for v0.2 or 'pip install praisonai[autogen-v4]' for v0.4")
         elif framework == "praisonai" and not PRAISONAI_AVAILABLE:
             raise ImportError("PraisonAI is not installed. Please install it with 'pip install praisonaiagents'")
 
@@ -316,11 +364,35 @@ class AgentsGenerator:
         framework = self.framework or config.get('framework')
 
         if framework == "autogen":
-            if not AUTOGEN_AVAILABLE:
-                raise ImportError("AutoGen is not installed. Please install it with 'pip install praisonai[autogen]'")
+            if not (AUTOGEN_AVAILABLE or AUTOGEN_V4_AVAILABLE):
+                raise ImportError("AutoGen is not installed. Please install it with 'pip install praisonai[autogen]' for v0.2 or 'pip install praisonai[autogen-v4]' for v0.4")
+            
+            # Choose autogen version based on availability and environment preference
+            # AUTOGEN_VERSION can be set to "v0.2" or "v0.4" to force a specific version
+            autogen_version = os.environ.get("AUTOGEN_VERSION", "auto").lower()
+            
+            use_v4 = False
+            if autogen_version == "v0.4" and AUTOGEN_V4_AVAILABLE:
+                use_v4 = True
+            elif autogen_version == "v0.2" and AUTOGEN_AVAILABLE:
+                use_v4 = False
+            elif autogen_version == "auto":
+                # Default preference: use v0.4 if available, fallback to v0.2
+                use_v4 = AUTOGEN_V4_AVAILABLE
+            else:
+                # Fallback to whatever is available
+                use_v4 = AUTOGEN_V4_AVAILABLE and not AUTOGEN_AVAILABLE
+            
             if AGENTOPS_AVAILABLE:
-                agentops.init(os.environ.get("AGENTOPS_API_KEY"), default_tags=["autogen"])
-            return self._run_autogen(config, topic, tools_dict)
+                version_tag = "autogen-v4" if use_v4 else "autogen-v2"
+                agentops.init(os.environ.get("AGENTOPS_API_KEY"), default_tags=[version_tag])
+            
+            if use_v4:
+                self.logger.info("Using AutoGen v0.4")
+                return self._run_autogen_v4(config, topic, tools_dict)
+            else:
+                self.logger.info("Using AutoGen v0.2")
+                return self._run_autogen(config, topic, tools_dict)
         elif framework == "praisonai":
             if not PRAISONAI_AVAILABLE:
                 raise ImportError("PraisonAI is not installed. Please install it with 'pip install praisonaiagents'")
@@ -407,6 +479,113 @@ class AgentsGenerator:
             
         return result
 
+    def _run_autogen_v4(self, config, topic, tools_dict):
+        """
+        Run agents using the AutoGen v0.4 framework with async, event-driven architecture.
+        
+        Args:
+            config (dict): Configuration dictionary
+            topic (str): The topic to process
+            tools_dict (dict): Dictionary of available tools
+            
+        Returns:
+            str: Result of the agent interactions
+        """
+        import asyncio
+        
+        async def run_autogen_v4_async():
+            # Create model client for v0.4
+            model_config = self.config_list[0] if self.config_list else {}
+            model_client = OpenAIChatCompletionClient(
+                model=model_config.get('model', 'gpt-4o'),
+                api_key=model_config.get('api_key', os.environ.get("OPENAI_API_KEY")),
+                base_url=model_config.get('base_url', "https://api.openai.com/v1")
+            )
+            
+            agents = []
+            combined_tasks = []
+            
+            # Create agents from config
+            for role, details in config['roles'].items():
+                # For AutoGen v0.4, ensure agent name is a valid Python identifier
+                agent_name = details['role'].format(topic=topic).replace("{topic}", topic)
+                agent_name = sanitize_agent_name_for_autogen_v4(agent_name)
+                backstory = details['backstory'].format(topic=topic)
+                
+                # Convert tools for v0.4 - simplified tool passing
+                agent_tools = []
+                for tool_name in details.get('tools', []):
+                    if tool_name in tools_dict:
+                        tool_instance = tools_dict[tool_name]
+                        # For v0.4, we can pass the tool's run method directly if it's callable
+                        if hasattr(tool_instance, 'run') and callable(tool_instance.run):
+                            agent_tools.append(tool_instance.run)
+                
+                # Create v0.4 AssistantAgent
+                assistant = AutoGenV4AssistantAgent(
+                    name=agent_name,
+                    system_message=backstory + ". Must reply with 'TERMINATE' when the task is complete.",
+                    model_client=model_client,
+                    tools=agent_tools,
+                    reflect_on_tool_use=True
+                )
+                
+                agents.append(assistant)
+                
+                # Collect all task descriptions for sequential execution
+                for task_name, task_details in details.get('tasks', {}).items():
+                    description_filled = task_details['description'].format(topic=topic)
+                    combined_tasks.append(description_filled)
+            
+            if not agents:
+                return "No agents created from configuration"
+            
+            # Create termination conditions
+            text_termination = TextMentionTermination("TERMINATE")
+            max_messages_termination = MaxMessageTermination(max_messages=20)
+            termination_condition = text_termination | max_messages_termination
+            
+            # Create RoundRobinGroupChat for parallel/sequential execution
+            group_chat = RoundRobinGroupChat(
+                agents,
+                termination_condition=termination_condition,
+                max_turns=len(agents) * 3  # Allow multiple rounds
+            )
+            
+            # Combine all tasks into a single task description
+            task_description = f"Topic: {topic}\n\nTasks to complete:\n" + "\n".join(
+                f"{i+1}. {task}" for i, task in enumerate(combined_tasks)
+            )
+            
+            # Run the group chat
+            try:
+                result = await group_chat.run(task=task_description)
+                
+                # Extract the final message content
+                if result.messages:
+                    final_message = result.messages[-1]
+                    if hasattr(final_message, 'content'):
+                        return f"### AutoGen v0.4 Output ###\n{final_message.content}"
+                    else:
+                        return f"### AutoGen v0.4 Output ###\n{str(final_message)}"
+                else:
+                    return "### AutoGen v0.4 Output ###\nNo messages generated"
+                    
+            except Exception as e:
+                self.logger.error(f"Error in AutoGen v0.4 execution: {str(e)}")
+                return f"### AutoGen v0.4 Error ###\n{str(e)}"
+            
+            finally:
+                # Close the model client
+                await model_client.close()
+        
+        # Run the async function
+        try:
+            return asyncio.run(run_autogen_v4_async())
+        except Exception as e:
+            self.logger.error(f"Error running AutoGen v0.4: {str(e)}")
+            return f"### AutoGen v0.4 Error ###\n{str(e)}"
+
     def _run_crewai(self, config, topic, tools_dict):
         """
         Run agents using the CrewAI framework.

{praisonai-2.2.64 → praisonai-2.2.65}/praisonai/auto.py

@@ -32,6 +32,13 @@ try:
 except ImportError:
     pass
 
+try:
+    from autogen_agentchat.agents import AssistantAgent
+    from autogen_ext.models.openai import OpenAIChatCompletionClient
+    AUTOGEN_V4_AVAILABLE = True
+except ImportError:
+    AUTOGEN_V4_AVAILABLE = False
+
 try:
     from praisonai_tools import (
         CodeDocsSearchTool, CSVSearchTool, DirectorySearchTool, DOCXSearchTool,
@@ -73,10 +80,11 @@ class AutoGenerator:
 CrewAI is not installed. Please install with:
     pip install "praisonai[crewai]"
 """)
-        elif framework == "autogen" and not AUTOGEN_AVAILABLE:
+        elif framework == "autogen" and not (AUTOGEN_AVAILABLE or AUTOGEN_V4_AVAILABLE):
             raise ImportError("""
 AutoGen is not installed. Please install with:
-    pip install "praisonai[autogen]"
+    pip install "praisonai[autogen]" for v0.2
+    pip install "praisonai[autogen-v4]" for v0.4
 """)
         elif framework == "praisonai" and not PRAISONAI_AVAILABLE:
             raise ImportError("""
@@ -86,7 +94,14 @@ Praisonai is not installed. Please install with:
 
         # Only show tools message if using a framework and tools are needed
         if (framework in ["crewai", "autogen"]) and not PRAISONAI_TOOLS_AVAILABLE:
-            logging.warning(f"""
+            if framework == "autogen":
+                logging.warning("""
+Tools are not available for autogen. To use tools, install:
+    pip install "praisonai[autogen]" for v0.2
+    pip install "praisonai[autogen-v4]" for v0.4
+""")
+            else:
+                logging.warning(f"""
 Tools are not available for {framework}. To use tools, install:
     pip install "praisonai[{framework}]"
 """)

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

{praisonai-2.2.64 → praisonai-2.2.65}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "PraisonAI"
-version = "2.2.64"
+version = "2.2.65"
 description = "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."
 readme = "README.md"
 license = ""
@@ -92,10 +92,17 @@ call = [
 train = []
 crewai = ["crewai>=0.32.0", "praisonai-tools>=0.0.22"]
 autogen = ["pyautogen==0.2.29", "praisonai-tools>=0.0.22", "crewai"]
+autogen-v4 = [
+    "autogen-agentchat>=0.4.0", 
+    "autogen-ext[openai]>=0.4.0", 
+    "autogen-core>=0.4.0",
+    "praisonai-tools>=0.0.22", 
+    "crewai"
+]
 
 [tool.poetry]
 name = "PraisonAI"
-version = "2.2.64"
+version = "2.2.65"
 description = "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."
 authors = ["Mervin Praison"]
 license = ""
@@ -119,6 +126,9 @@ instructor = ">=1.3.3"
 PyYAML = ">=6.0"
 mcp = ">=1.6.0"
 pyautogen = {version = "==0.2.29", optional = true}
+autogen-agentchat = {version = ">=0.4.0", optional = true}
+autogen-ext = {version = ">=0.4.0", optional = true}
+autogen-core = {version = ">=0.4.0", optional = true}
 crewai = {version = ">=0.32.0", optional = true}
 praisonai-tools = {version = ">=0.0.22", optional = true}
 chainlit = {version = "==2.5.5", optional = true}
@@ -277,6 +287,7 @@ call = [
 ]
 crewai = ["crewai", "praisonai-tools"]
 autogen = ["pyautogen", "praisonai-tools", "crewai"]
+autogen-v4 = ["autogen-agentchat", "autogen-ext", "autogen-core", "praisonai-tools", "crewai"]
 
 [tool.poetry-dynamic-versioning]
 enable = true