PraisonAI 0.0.22__tar.gz → 0.0.24__tar.gz

{praisonai-0.0.22 → praisonai-0.0.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.22
+Version: 0.0.24
 Summary: PraisonAI application combines AutoGen and CrewAI or similar frameworks into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customization, and efficient human-agent collaboration.
 Author: Mervin Praison
 Requires-Python: >=3.10,<3.13
@@ -8,10 +8,12 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: chainlit
+Provides-Extra: gradio
+Provides-Extra: ui
 Requires-Dist: Flask (>=3.0.0)
 Requires-Dist: crewai (>=0.30.4)
 Requires-Dist: crewai-tools (>=0.2.6,<0.3.0)
-Requires-Dist: gradio (>=4.20.0)
 Requires-Dist: markdown (>=3.5)
 Requires-Dist: praisonai-tools (>=0.0.4)
 Requires-Dist: pyautogen (>=0.2.19)

praisonai-0.0.24/praisonai/chainlit_ui.py

@@ -0,0 +1,67 @@
+from praisonai.agents_generator import AgentsGenerator 
+from praisonai.auto import AutoGenerator 
+import chainlit as cl
+import os
+from chainlit.types import ThreadDict
+from typing import Optional
+from dotenv import load_dotenv
+
+framework = "crewai"
+config_list = [
+            {
+                'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-4o"),
+                'base_url': os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1"),
+            }
+        ]
+agent_file = "test.yaml"
+
+@cl.on_chat_start
+async def start_chat():
+    cl.user_session.set(
+        "message_history",
+        [{"role": "system", "content": "You are a helpful assistant."}],
+    )
+    
+@cl.on_chat_resume
+async def on_chat_resume(thread: ThreadDict):
+    message_history = cl.user_session.get("message_history", [])
+    root_messages = [m for m in thread["steps"] if m["parentId"] is None]
+    for message in root_messages:
+        if message["type"] == "user_message":
+            message_history.append({"role": "user", "content": message["output"]})
+        elif message["type"] == "ai_message":
+            message_history.append({"role": "assistant", "content": message["content"]})
+    cl.user_session.set("message_history", message_history)
+
+@cl.on_message
+async def main(message: cl.Message):
+    """Run PraisonAI with the provided message as the topic."""
+    message_history = cl.user_session.get("message_history")
+    message_history.append({"role": "user", "content": message.content})
+    topic = message.content
+    agent_file = "test.yaml"
+    generator = AutoGenerator(topic=topic, framework=framework)
+    agent_file = generator.generate()
+    agents_generator = AgentsGenerator(agent_file, framework, config_list)
+    result = agents_generator.generate_crew_and_kickoff()
+    msg = cl.Message(content=result)
+    await msg.send()
+    message_history.append({"role": "assistant", "content": message.content})
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Get username and password from environment variables
+username = os.getenv("CHAINLIT_USERNAME", "admin")  # Default to "admin" if not found
+password = os.getenv("CHAINLIT_PASSWORD", "admin")  # Default to "admin" if not found
+
+@cl.password_auth_callback
+def auth_callback(username: str, password: str):
+    # Fetch the user matching username from your database
+    # and compare the hashed password with the value stored in the database
+    if (username, password) == ("admin", "admin"):
+        return cl.User(
+            identifier="admin", metadata={"role": "admin", "provider": "credentials"}
+        )
+    else:
+        return None

{praisonai-0.0.22 → praisonai-0.0.24}/praisonai/cli.py

@@ -14,6 +14,18 @@ from .auto import AutoGenerator
 from .agents_generator import AgentsGenerator
 from .inbuilt_tools import *
 
+try:
+    from chainlit.cli import chainlit_run
+    CHAINLIT_AVAILABLE = True
+except ImportError:
+    CHAINLIT_AVAILABLE = False
+
+try:
+    import gradio as gr
+    GRADIO_AVAILABLE = True
+except ImportError:
+    GRADIO_AVAILABLE = False
+
 class PraisonAI:
     def __init__(self, agent_file="agents.yaml", framework="", auto=False, init=False):
         """
@@ -72,9 +84,8 @@ class PraisonAI:
         
         self.framework = args.framework or self.framework 
         
-        ui = args.ui
         if args.agent_file:
-            if args.agent_file.startswith("tests.test"): # Argument used for testing purposes
+            if args.agent_file.startswith("tests.test"): # Argument used for testing purposes. eg: python -m unittest tests.test 
                 print("test")
             else:
                 self.agent_file = args.agent_file
@@ -100,8 +111,16 @@ class PraisonAI:
             print("File {} created successfully".format(self.agent_file))
             return "File {} created successfully".format(self.agent_file)
         
-        if ui:
-            self.create_gradio_interface()
+        if args.ui:
+            if args.ui == "gradio":
+                self.create_gradio_interface()
+            elif args.ui == "chainlit":
+                self.create_chainlit_interface()
+            else:
+                # Modify below code to allow default ui
+                agents_generator = AgentsGenerator(self.agent_file, self.framework, self.config_list)
+                result = agents_generator.generate_crew_and_kickoff()
+                return result
         else:
             agents_generator = AgentsGenerator(self.agent_file, self.framework, self.config_list)
             result = agents_generator.generate_crew_and_kickoff()
@@ -126,12 +145,11 @@ class PraisonAI:
         """
         parser = argparse.ArgumentParser(prog="praisonai", description="praisonAI command-line interface")
         parser.add_argument("--framework", choices=["crewai", "autogen"], help="Specify the framework")
-        parser.add_argument("--ui", action="store_true", help="Enable UI mode")
+        parser.add_argument("--ui", choices=["chainlit", "gradio"], help="Specify the UI framework (gradio or chainlit).")
         parser.add_argument("--auto", nargs=argparse.REMAINDER, help="Enable auto mode and pass arguments for it")
         parser.add_argument("--init", nargs=argparse.REMAINDER, help="Enable auto mode and pass arguments for it")
         parser.add_argument("agent_file", nargs="?", help="Specify the agent file")
         parser.add_argument("--deploy", action="store_true", help="Deploy the application")  # New argument
-
         args, unknown_args = parser.parse_known_args()
 
         if unknown_args and unknown_args[0] == '-b' and unknown_args[1] == 'api:app':
@@ -157,42 +175,61 @@ class PraisonAI:
         Example:
             >>> praison_ai.create_gradio_interface()
         """
-        def generate_crew_and_kickoff_interface(auto_args, framework):
-            """
-            Generate a crew and kick off tasks based on the provided auto arguments and framework.
-
-            Args:
-                auto_args (list): Topic.
-                framework (str): The framework to use for generating agents.
-
-            Returns:
-                str: A string representing the result of generating the crew and kicking off tasks.
+        if GRADIO_AVAILABLE:
+            def generate_crew_and_kickoff_interface(auto_args, framework):
+                """
+                Generate a crew and kick off tasks based on the provided auto arguments and framework.
+
+                Args:
+                    auto_args (list): Topic.
+                    framework (str): The framework to use for generating agents.
+
+                Returns:
+                    str: A string representing the result of generating the crew and kicking off tasks.
+
+                Raises:
+                    None: This method does not raise any exceptions.
+
+                Example:
+                    >>> result = generate_crew_and_kickoff_interface("Create a movie about Cat in Mars", "crewai")
+                    >>> print(result)
+                """
+                self.framework = framework
+                self.agent_file = "test.yaml"
+                generator = AutoGenerator(topic=auto_args , framework=self.framework)
+                self.agent_file = generator.generate()
+                agents_generator = AgentsGenerator(self.agent_file, self.framework, self.config_list)
+                result = agents_generator.generate_crew_and_kickoff()
+                return result
+
+            gr.Interface(
+                fn=generate_crew_and_kickoff_interface,
+                inputs=[gr.Textbox(lines=2, label="Auto Args"), gr.Dropdown(choices=["crewai", "autogen"], label="Framework")],
+                outputs="textbox",
+                title="Praison AI Studio",
+                description="Create Agents and perform tasks",
+                theme="default"
+            ).launch()
+        else:
+            print("ERROR: Gradio is not installed. Please install it with 'pip install \"praisonai[gradio]\"' to use this feature.") 
+        
+    def create_chainlit_interface(self):
+        """
+        Create a Chainlit interface for generating agents and performing tasks.
 
-            Raises:
-                None: This method does not raise any exceptions.
+        This function sets up a Chainlit application that listens for messages.
+        When a message is received, it runs PraisonAI with the provided message as the topic.
+        The generated agents are then used to perform tasks.
 
-            Example:
-                >>> result = generate_crew_and_kickoff_interface("Create a movie about Cat in Mars", "crewai")
-                >>> print(result)
-            """
-            self.framework = framework
-            self.agent_file = "test.yaml"
-            generator = AutoGenerator(topic=auto_args , framework=self.framework)
-            self.agent_file = generator.generate()
-            agents_generator = AgentsGenerator(self.agent_file, self.framework, self.config_list)
-            result = agents_generator.generate_crew_and_kickoff()
-            return result
-
-        gr.Interface(
-            fn=generate_crew_and_kickoff_interface,
-            inputs=[gr.Textbox(lines=2, label="Auto Args"), gr.Dropdown(choices=["crewai", "autogen"], label="Framework")],
-            outputs="textbox",
-            title="Praison AI Studio",
-            description="Create Agents and perform tasks",
-            theme="default"
-        ).launch()
+        Returns:
+            None: This function does not return any value. It starts the Chainlit application.
+        """
+        if CHAINLIT_AVAILABLE:
+            os.environ["CHAINLIT_PORT"] = "8082"  
+            chainlit_run(["praisonai/chainlit_ui.py"])  
+        else:
+            print("ERROR: Chainlit is not installed. Please install it with 'pip install \"praisonai\[ui]\"' to use the UI.")        
 
 if __name__ == "__main__":
     praison_ai = PraisonAI()
-    praison_ai.main()
-    
+    praison_ai.main()

{praisonai-0.0.22 → praisonai-0.0.24}/praisonai/deploy.py

@@ -56,7 +56,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==0.0.22 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.24 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

{praisonai-0.0.22 → praisonai-0.0.24}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PraisonAI"
-version = "0.0.22"
+version = "0.0.24"
 description = "PraisonAI application combines AutoGen and CrewAI or similar frameworks into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customization, and efficient human-agent collaboration."
 authors = ["Mervin Praison"]
 license = ""
@@ -18,7 +18,6 @@ python = ">=3.10,<3.13"
 rich = ">=13.7"
 pyautogen = ">=0.2.19"
 crewai = ">=0.30.4"
-gradio = ">=4.20.0"
 Flask = ">=3.0.0"
 markdown = ">=3.5"
 crewai-tools = "^0.2.6"
@@ -33,4 +32,9 @@ requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
-praisonai = "praisonai.__main__:main"
+praisonai = "praisonai.__main__:main"
+
+[tool.poetry.extras]
+ui = ["chainlit"]
+chainlit = ["chainlit"]
+gradio = ["gradio"]