PraisonAI 0.0.6__tar.gz → 0.0.7__tar.gz

{praisonai-0.0.6 → praisonai-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: praisonAI
-Version: 0.0.6
+Version: 0.0.7
 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
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: tools
 Requires-Dist: crewai (>=0.22.5)
+Requires-Dist: gradio (>=4.20.0)
 Requires-Dist: pyautogen (>=0.2.19)
 Requires-Dist: rich (>=13.7)
 Project-URL: Homepage, https://github.com/mervinpraison/praisonAI
@@ -31,3 +32,9 @@ pip install praisonai
 ```bash
 praisonai
 ```
+
+## Test
+    
+```bash
+python -m unittest tests.test 
+```

{praisonai-0.0.6 → praisonai-0.0.7}/README.md

@@ -12,4 +12,10 @@ pip install praisonai
 
 ```bash
 praisonai
+```
+
+## Test
+    
+```bash
+python -m unittest tests.test 
 ```

praisonai-0.0.7/praisonai/__init__.py

@@ -0,0 +1,2 @@
+from .cli import PraisonAI
+from .version import __version__

praisonai-0.0.7/praisonai/__main__.py

@@ -0,0 +1,7 @@
+# __main__.py
+
+from .cli import PraisonAI
+
+if __name__ == "__main__":
+    praison_ai = PraisonAI()
+    praison_ai.main()

praisonai-0.0.7/praisonai/auto-old.py

@@ -0,0 +1,151 @@
+from openai import OpenAI
+from pydantic import BaseModel, Field
+from typing import Dict
+import instructor
+import os
+import json
+
+config_list = [
+    {
+        'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-4-turbo-preview"),
+        'base_url': os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1"),
+    }
+]
+
+task = "Movie Story writing"
+
+# Define new Pydantic models to represent the script writing team structure
+class TaskDetails(BaseModel):
+    description: str
+    expected_output: str
+
+class RoleDetails(BaseModel):
+    role: str
+    goal: str
+    backstory: str
+    tasks: Dict[str, TaskDetails]
+
+class TeamStructure(BaseModel):
+    roles: Dict[str, RoleDetails]
+
+# Enables `response_model` in create call with the new model
+client = instructor.patch(
+    OpenAI(
+        base_url=config_list[0]['base_url'],
+        api_key=os.getenv("OPENAI_API_KEY"),
+    ),
+    mode=instructor.Mode.JSON,
+)
+
+response = client.chat.completions.create(
+  model=config_list[0]['model'],
+  response_model=TeamStructure,
+  max_retries=10,
+  messages=[
+    {"role": "system", "content": "You are a helpful assistant designed to output complex team structures."},
+    {"role": "user", "content": 
+"""Generate a team structure for """ + task + """ task. 
+No Input data will be provided to the team.
+The team will work in sequence. First role will pass the output to the next role, and so on.
+The last role will generate the final output.
+Think step by step.
+With maximum 3 roles, each with 1 task. Include role goals, backstories, task descriptions, and expected outputs.
+Output Structure:
+{
+"roles": {
+"<role_id>": {
+"role": "<role_name>",
+"goal": "<role_goal>",
+"backstory": "<role_backstory>",
+"tasks": {
+"<task_id>": {
+"description": "<task_description>",
+"expected_output": "<task_expected_output>"
+}
+}
+},
+"<role_id>": {
+"role": "<role_name>",
+"goal": "<role_goal>",
+"backstory": "<role_backstory>",
+"tasks": {
+"<task_id>": {
+"description": "<task_description>",
+"expected_output": "<task_expected_output>"
+}
+}
+}
+}
+}
+Example Below:
+{
+"roles": {
+"narrative_designer": {
+"role": "Narrative Designer",
+"goal": "Create AI storylines",
+"backstory": "Skilled in narrative development for AI, with a focus on story resonance.",
+"tasks": {
+"story_concept_development": {
+"description": "Craft a unique AI story concept with depth and engagement.",
+"expected_output": "Document with narrative arcs, character bios, and settings."
+}
+}
+},
+"scriptwriter": {
+"role": "Scriptwriter",
+"goal": "Write scripts from AI concepts",
+"backstory": "Expert in dialogue and script structure, translating concepts into scripts.",
+"tasks": {
+"scriptwriting_task": {
+"description": "Turn narrative concepts into scripts, including dialogue and scenes.",
+"expected_output": "Production-ready script with dialogue and scene details."
+}
+}
+}
+}
+}
+"""}
+  ]
+)
+
+print(response.model_dump_json(indent=2))
+
+import yaml
+
+def convert_and_save(json_data, topic="Artificial Intelligence", filename="test.yaml"):
+    """Converts the provided JSON data into the desired YAML format and saves it to a file.
+
+    Args:
+        json_data (dict): The JSON data representing the team structure.
+        topic (str, optional): The topic to be inserted into the YAML. Defaults to "Artificial Intelligence".
+        filename (str, optional): The name of the YAML file to save. Defaults to "test.yaml".
+    """
+
+    yaml_data = {
+        "framework": "crewai",
+        "topic": topic,
+        "roles": {},
+        "dependencies": []
+    }
+
+    # Process roles in the order they appear in JSON
+    for role_id, role_details in json_data['roles'].items():
+        yaml_data['roles'][role_id] = {
+            "backstory": "" + role_details['backstory'],
+            "goal": role_details['goal'],
+            "role": role_details['role'],
+            "tasks": {}
+        }
+
+        for task_id, task_details in role_details['tasks'].items():
+            yaml_data['roles'][role_id]['tasks'][task_id] = {
+                "description": "" + task_details['description'],
+                "expected_output": "" + task_details['expected_output']
+            }
+
+    # Save to YAML file, maintaining the order
+    with open(filename, 'w') as f:
+        yaml.dump(yaml_data, f, allow_unicode=True, sort_keys=False)
+    
+json_data = json.loads(response.model_dump_json())
+convert_and_save(json_data) 

praisonai-0.0.7/praisonai/auto.py

@@ -0,0 +1,135 @@
+from openai import OpenAI
+from pydantic import BaseModel
+from typing import Dict
+import instructor
+import os
+import json
+import yaml
+from rich import print
+
+# Define Pydantic models outside of the generate method
+class TaskDetails(BaseModel):
+    description: str
+    expected_output: str
+
+class RoleDetails(BaseModel):
+    role: str
+    goal: str
+    backstory: str
+    tasks: Dict[str, TaskDetails]
+
+class TeamStructure(BaseModel):
+    roles: Dict[str, RoleDetails]
+
+class AutoGenerator:
+    def __init__(self, topic="Movie Story writing about AI", filename="test.yaml", framework="crewai"):
+        self.config_list = [
+            {
+                'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-4-turbo-preview"),
+                'base_url': os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1"),
+            }
+        ]
+        self.topic = topic
+        self.filename = filename
+        self.framework = framework
+        self.client = instructor.patch(
+            OpenAI(
+                base_url=self.config_list[0]['base_url'],
+                api_key=os.getenv("OPENAI_API_KEY"),
+            ),
+            mode=instructor.Mode.JSON,
+        )
+
+    def generate(self):
+        response = self.client.chat.completions.create(
+            model=self.config_list[0]['model'],
+            response_model=TeamStructure,
+            max_retries=10,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant designed to output complex team structures."},
+                {"role": "user", "content": self.get_user_content()}
+            ]
+        )
+        json_data = json.loads(response.model_dump_json())
+        self.convert_and_save(json_data)
+        full_path = os.path.abspath(self.filename)
+        return full_path
+
+    def convert_and_save(self, json_data):
+        """Converts the provided JSON data into the desired YAML format and saves it to a file.
+
+        Args:
+            json_data (dict): The JSON data representing the team structure.
+            topic (str, optional): The topic to be inserted into the YAML. Defaults to "Artificial Intelligence".
+            filename (str, optional): The name of the YAML file to save. Defaults to "test.yaml".
+        """
+
+        yaml_data = {
+            "framework": "crewai",
+            "topic": self.topic,
+            "roles": {},
+            "dependencies": []
+        }
+        print(self.topic)
+        print(self.framework)
+
+        for role_id, role_details in json_data['roles'].items():
+            yaml_data['roles'][role_id] = {
+                "backstory": "" + role_details['backstory'],
+                "goal": role_details['goal'],
+                "role": role_details['role'],
+                "tasks": {}
+            }
+
+            for task_id, task_details in role_details['tasks'].items():
+                yaml_data['roles'][role_id]['tasks'][task_id] = {
+                    "description": "" + task_details['description'],
+                    "expected_output": "" + task_details['expected_output']
+                }
+
+        # Save to YAML file, maintaining the order
+        with open(self.filename, 'w') as f:
+            yaml.dump(yaml_data, f, allow_unicode=True, sort_keys=False)
+
+    def get_user_content(self):
+        user_content = """Generate a team structure for  \"""" + self.topic + """\" task. 
+No Input data will be provided to the team.
+The team will work in sequence. First role will pass the output to the next role, and so on.
+The last role will generate the final output.
+Think step by step.
+With maximum 3 roles, each with 1 task. Include role goals, backstories, task descriptions, and expected outputs.
+Example Below: 
+Use below example to understand the structure of the output. 
+The final role you create should satisfy the provided task: """ + self.topic + """.
+{
+"roles": {
+"narrative_designer": {
+"role": "Narrative Designer",
+"goal": "Create AI storylines",
+"backstory": "Skilled in narrative development for AI, with a focus on story resonance.",
+"tasks": {
+"story_concept_development": {
+"description": "Craft a unique AI story concept with depth and engagement.",
+"expected_output": "Document with narrative arcs, character bios, and settings."
+}
+}
+},
+"scriptwriter": {
+"role": "Scriptwriter",
+"goal": "Write scripts from AI concepts",
+"backstory": "Expert in dialogue and script structure, translating concepts into scripts.",
+"tasks": {
+"scriptwriting_task": {
+"description": "Turn narrative concepts into scripts, including dialogue and scenes.",
+"expected_output": "Production-ready script with dialogue and scene details."
+}
+}
+}
+}
+}
+        """
+        return user_content
+
+    
+generator = AutoGenerator(framework="crewai", topic="Create a snake game in python")
+print(generator.generate())

praisonai-0.0.6/praisonai/cli.py → praisonai-0.0.7/praisonai/cli-old.py

@@ -1,13 +1,16 @@
-# cli.py
+# praisonai/cli.py
 
 import sys
-from .version import __version__
+from version import __version__
 import yaml, os
 from rich import print
 from dotenv import load_dotenv
 from crewai import Agent, Task, Crew
 load_dotenv()
 import autogen
+import gradio as gr
+import argparse
+
 config_list = [
     {
         'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-3.5-turbo"),
@@ -47,7 +50,7 @@ def generate_crew_and_kickoff(agent_file, framework=None):
                 chat_task = {
                     "recipient": agents[role],
                     "message": description_filled,
-                    "summary_method": "last_msg",  # Customize as needed
+                    "summary_method": "last_msg", 
                     # Additional fields like carryover can be added based on dependencies
                 }
                 tasks.append(chat_task)
@@ -56,7 +59,7 @@ def generate_crew_and_kickoff(agent_file, framework=None):
         user = autogen.UserProxyAgent(
             name="User",
             human_input_mode="NEVER",
-            is_termination_msg=lambda x: (x.get("content") or "").rstrip().endswith("TERMINATE"),
+            is_termination_msg=lambda x: (x.get("content") or "").rstrip().rstrip(".").lower().endswith("terminate") or "TERMINATE" in (x.get("content") or ""),
             code_execution_config={
                 "work_dir": "coding",
                 "use_docker": False,
@@ -72,7 +75,7 @@ def generate_crew_and_kickoff(agent_file, framework=None):
             backstory_filled = details['backstory'].format(topic=topic)
             
             # Assume tools are loaded and handled here as per your requirements
-            agent = Agent(role=role_filled, goal=goal_filled, backstory=backstory_filled)
+            agent = Agent(role=role_filled, goal=goal_filled, backstory=backstory_filled, allow_delegation=False)
             agents[role] = agent
 
             for task_name, task_details in details.get('tasks', {}).items():
@@ -88,29 +91,53 @@ def generate_crew_and_kickoff(agent_file, framework=None):
             verbose=2
         )
 
-        result = crew.kickoff()
+        response = crew.kickoff()
+        result = f"### Output ###\n{response}"
     return result
 
 def main(args=None):
     if args is None:
-        args = sys.argv[1:]  # Skip the script name
+        args = parse_args()
 
     invocation_cmd = "praisonai"
     version_string = f"praisonAI version {__version__}"
-    framework = "crewai"  # Default framework
-
-    if "--framework" in args:
-        framework_index = args.index("--framework")
-        framework = args[framework_index + 1]
-        args = args[:framework_index] + args[framework_index + 2:]
+    framework = args.framework  # Default framework
+    ui = args.ui  # Default UI flag
+    print(args)
 
-    if args:
-        agent_file = args[-1]  # Assuming the last argument is the agent file
+    if args.agent_file:
+        agent_file = args.agent_file
     else:
         agent_file = "agents.yaml"
 
-    result = generate_crew_and_kickoff(agent_file, framework)
-    print(result)
-    
+    if ui:
+        create_gradio_interface()
+    else:
+        result = generate_crew_and_kickoff(agent_file, framework)
+        print(result)
+
+def parse_args():
+    parser = argparse.ArgumentParser(prog="praisonai", description="praisonAI command-line interface")
+    parser.add_argument("--framework", choices=["crewai", "autogen"], default="crewai", help="Specify the framework")
+    parser.add_argument("--ui", action="store_true", help="Enable UI mode")
+    parser.add_argument("--auto", action="store_true", help="Enable auto mode")
+    parser.add_argument("agent_file", nargs="?", help="Specify the agent file")
+
+    return parser.parse_args()
+
+def create_gradio_interface():
+    def generate_crew_and_kickoff_interface(agent_file, framework):
+        result = generate_crew_and_kickoff(agent_file, framework)
+        return result
+
+    gr.Interface(
+        fn=generate_crew_and_kickoff_interface,
+        inputs=["textbox", "radio"],
+        outputs="textbox",
+        title="Generate Crew and Kickoff",
+        description="Generate a crew and kickoff the tasks",
+        theme="default"
+    ).launch()
+
 if __name__ == "__main__":
     main()

praisonai-0.0.7/praisonai/cli.py

@@ -0,0 +1,164 @@
+# praisonai/cli.py
+
+import sys
+from .version import __version__
+import yaml, os
+from rich import print
+from dotenv import load_dotenv
+from crewai import Agent, Task, Crew
+load_dotenv()
+import autogen
+import gradio as gr
+import argparse
+from .auto import AutoGenerator
+
+class PraisonAI:
+    def __init__(self, agent_file="agents.yaml", framework="crewai", auto=False):
+        self.config_list = [
+            {
+                'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-4-turbo-preview"),
+                'base_url': os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1"),
+            }
+        ]
+        self.agent_file = agent_file
+        self.framework = framework
+        self.auto = auto
+
+    def generate_crew_and_kickoff(self):
+        with open(self.agent_file, 'r') as f:
+            config = yaml.safe_load(f)
+
+        topic = config['topic']  
+        framework = self.framework or config.get('framework')
+
+        agents = {}
+        tasks = []
+        if framework == "autogen":
+            # Load the LLM configuration dynamically
+            print(self.config_list)
+            llm_config = {"config_list": self.config_list}
+            
+            for role, details in config['roles'].items():
+                agent_name = details['role'].format(topic=topic).replace("{topic}", topic)
+                agent_goal = details['goal'].format(topic=topic)
+                # Creating an AssistantAgent for each role dynamically
+                agents[role] = autogen.AssistantAgent(
+                    name=agent_name,
+                    llm_config=llm_config,
+                    system_message=details['backstory'].format(topic=topic)+". Must Reply \"TERMINATE\" in the end when everything is done.",
+                )
+
+                # Preparing tasks for initiate_chats
+                for task_name, task_details in details.get('tasks', {}).items():
+                    description_filled = task_details['description'].format(topic=topic)
+                    expected_output_filled = task_details['expected_output'].format(topic=topic)
+                    
+                    chat_task = {
+                        "recipient": agents[role],
+                        "message": description_filled,
+                        "summary_method": "last_msg", 
+                        # Additional fields like carryover can be added based on dependencies
+                    }
+                    tasks.append(chat_task)
+
+            # Assuming the user proxy agent is set up as per your requirements
+            user = autogen.UserProxyAgent(
+                name="User",
+                human_input_mode="NEVER",
+                is_termination_msg=lambda x: (x.get("content") or "").rstrip().rstrip(".").lower().endswith("terminate") or "TERMINATE" in (x.get("content") or ""),
+                code_execution_config={
+                    "work_dir": "coding",
+                    "use_docker": False,
+                },
+                # additional setup for the user proxy agent
+            )
+            response = user.initiate_chats(tasks)
+            result = "### Output ###\n"+response[-1].summary if hasattr(response[-1], 'summary') else ""
+        else:
+            for role, details in config['roles'].items():
+                role_filled = details['role'].format(topic=topic)
+                goal_filled = details['goal'].format(topic=topic)
+                backstory_filled = details['backstory'].format(topic=topic)
+                
+                # Assume tools are loaded and handled here as per your requirements
+                agent = Agent(role=role_filled, goal=goal_filled, backstory=backstory_filled, allow_delegation=False)
+                agents[role] = agent
+
+                for task_name, task_details in details.get('tasks', {}).items():
+                    description_filled = task_details['description'].format(topic=topic)
+                    expected_output_filled = task_details['expected_output'].format(topic=topic)
+
+                    task = Task(description=description_filled, expected_output=expected_output_filled, agent=agent)
+                    tasks.append(task)
+
+            crew = Crew(
+                agents=list(agents.values()),
+                tasks=tasks,
+                verbose=2
+            )
+
+            response = crew.kickoff()
+            result = f"### Output ###\n{response}"
+        return result
+
+    def main(self):
+        args = self.parse_args()
+        if args is None:
+            result = self.generate_crew_and_kickoff()
+            return result
+        invocation_cmd = "praisonai"
+        version_string = f"praisonAI version {__version__}"
+        
+        # if args.framework:
+        #     self.framework = args.framework
+        ui = args.ui  # Default UI flag
+
+        if args.agent_file:
+            self.agent_file = args.agent_file
+        else:
+            full_path = os.path.abspath(self.agent_file)
+            self.filename = full_path
+        
+        if args.auto:
+            self.topic = ' '.join(args.auto)
+        elif self.auto:  # Use the auto attribute if args.auto is not provided
+            self.topic = self.auto
+            
+        if args.auto or self.auto:
+            generator = AutoGenerator(topic=self.topic , framework=self.framework)
+            self.agent_file = generator.generate()
+            result = self.generate_crew_and_kickoff()
+            return result
+        
+        if ui:
+            self.create_gradio_interface()
+        else:
+            result = self.generate_crew_and_kickoff()
+            return result
+            
+    def parse_args(self):
+        parser = argparse.ArgumentParser(prog="praisonai", description="praisonAI command-line interface")
+        parser.add_argument("--framework", choices=["crewai", "autogen"], default="crewai", help="Specify the framework")
+        parser.add_argument("--ui", action="store_true", help="Enable UI mode")
+        parser.add_argument("--auto", nargs=argparse.REMAINDER, help="Enable auto mode and pass arguments for it")
+        parser.add_argument("agent_file", nargs="?", help="Specify the agent file")
+
+        return parser.parse_args()
+
+    def create_gradio_interface(self):
+        def generate_crew_and_kickoff_interface(agent_file, framework):
+            result = self.generate_crew_and_kickoff()
+            return result
+
+        gr.Interface(
+            fn=generate_crew_and_kickoff_interface,
+            inputs=["textbox"],
+            outputs="textbox",
+            title="Praison AI Studio",
+            description="Create Agents and perform tasks",
+            theme="default"
+        ).launch()
+
+if __name__ == "__main__":
+    praison_ai = PraisonAI()
+    praison_ai.main()

{praisonai-0.0.6 → praisonai-0.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "praisonAI"
-version = "0.0.6"
+version = "0.0.7"
 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,6 +18,7 @@ python = ">=3.10,<3.13"
 rich = ">=13.7"
 pyautogen = ">=0.2.19"
 crewai = ">=0.22.5"
+gradio = ">=4.20.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^5.2"

praisonai-0.0.6/praisonai/__init__.py

@@ -1 +0,0 @@
-from .version import __version__

praisonai-0.0.6/praisonai/__main__.py

@@ -1,6 +0,0 @@
-# __main__.py
-
-from .cli import main
-
-if __name__ == "__main__":
-    main()