PraisonAI 0.0.15__tar.gz → 0.0.17__tar.gz

{praisonai-0.0.15 → praisonai-0.0.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: praisonAI
-Version: 0.0.15
+Version: 0.0.17
 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
@@ -36,6 +36,11 @@ praisonai --init create a movie script about dog in moon
 ```
 This will automatically create agents.yaml file in the current directory.
 
+To initialse with a specific agent framework:
+
+```bash
+praisonai --framework autogen --init create movie script about cat in mars
+```
 
 ## Run
 
@@ -186,6 +191,28 @@ gcloud run deploy praisonai-service \
     --allow-unauthenticated \
     --set-env-vars OPENAI_MODEL_NAME=${OPENAI_MODEL_NAME},OPENAI_API_KEY=${OPENAI_API_KEY},OPENAI_API_BASE=${OPENAI_API_BASE}
 
+## Other Models
+
+Ollama
+OPENAI_API_BASE='http://localhost:11434/v1'
+OPENAI_MODEL_NAME='mistral'
+OPENAI_API_KEY='NA'
+
+FastChat¶
+OPENAI_API_BASE="http://localhost:8001/v1"
+OPENAI_MODEL_NAME='oh-2.5m7b-q51'
+OPENAI_API_KEY=NA
+
+LM Studio¶
+OPENAI_API_BASE="http://localhost:8000/v1"
+OPENAI_MODEL_NAME=NA
+OPENAI_API_KEY=NA
+
+Mistral API¶
+OPENAI_API_BASE=https://api.mistral.ai/v1
+OPENAI_MODEL_NAME="mistral-small"
+OPENAI_API_KEY=your-mistral-api-key
+
 ## Contributing
 
 - Fork on GitHub: Use the "Fork" button on the repository page.

{praisonai-0.0.15 → praisonai-0.0.17}/README.md

@@ -15,6 +15,11 @@ praisonai --init create a movie script about dog in moon
 ```
 This will automatically create agents.yaml file in the current directory.
 
+To initialse with a specific agent framework:
+
+```bash
+praisonai --framework autogen --init create movie script about cat in mars
+```
 
 ## Run
 
@@ -165,6 +170,28 @@ gcloud run deploy praisonai-service \
     --allow-unauthenticated \
     --set-env-vars OPENAI_MODEL_NAME=${OPENAI_MODEL_NAME},OPENAI_API_KEY=${OPENAI_API_KEY},OPENAI_API_BASE=${OPENAI_API_BASE}
 
+## Other Models
+
+Ollama
+OPENAI_API_BASE='http://localhost:11434/v1'
+OPENAI_MODEL_NAME='mistral'
+OPENAI_API_KEY='NA'
+
+FastChat¶
+OPENAI_API_BASE="http://localhost:8001/v1"
+OPENAI_MODEL_NAME='oh-2.5m7b-q51'
+OPENAI_API_KEY=NA
+
+LM Studio¶
+OPENAI_API_BASE="http://localhost:8000/v1"
+OPENAI_MODEL_NAME=NA
+OPENAI_API_KEY=NA
+
+Mistral API¶
+OPENAI_API_BASE=https://api.mistral.ai/v1
+OPENAI_MODEL_NAME="mistral-small"
+OPENAI_API_KEY=your-mistral-api-key
+
 ## Contributing
 
 - Fork on GitHub: Use the "Fork" button on the repository page.

{praisonai-0.0.15 → praisonai-0.0.17}/praisonai/auto.py

@@ -65,7 +65,7 @@ class AutoGenerator:
         """
 
         yaml_data = {
-            "framework": "crewai",
+            "framework": self.framework,
             "topic": self.topic,
             "roles": {},
             "dependencies": []

{praisonai-0.0.15 → praisonai-0.0.17}/praisonai/cli.py

@@ -121,8 +121,9 @@ class PraisonAI:
         invocation_cmd = "praisonai"
         version_string = f"praisonAI version {__version__}"
         
-        # if args.framework:
-        #     self.framework = args.framework
+        if args.framework:
+            self.framework = args.framework
+        
         ui = args.ui
 
         if args.agent_file:

{praisonai-0.0.15 → praisonai-0.0.17}/praisonai/deploy.py

@@ -8,6 +8,32 @@ class CloudDeployer:
         load_dotenv()
         self.set_environment_variables()
 
+    def create_dockerfile(self):
+        with open("Dockerfile", "w") as file:
+            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.17 gunicorn markdown\n")
+            file.write("EXPOSE 8080\n")
+            file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
+            
+    def create_api_file(self):
+        with open("api.py", "w") as file:
+            file.write("from flask import Flask\n")
+            file.write("from praisonai import PraisonAI\n")
+            file.write("import markdown\n\n")
+            file.write("app = Flask(__name__)\n\n")
+            file.write("def basic():\n")
+            file.write("    praison_ai = PraisonAI(agent_file=\"agents.yaml\")\n")
+            file.write("    return praison_ai.main()\n\n")
+            file.write("@app.route('/')\n")
+            file.write("def home():\n")
+            file.write("    output = basic()\n")
+            file.write("    html_output = markdown.markdown(output)\n")
+            file.write("    return f'<html><body>{html_output}</body></html>'\n\n")
+            file.write("if __name__ == \"__main__\":\n")
+            file.write("    app.run(debug=True)\n")
+    
     def set_environment_variables(self):
         """Sets environment variables with fallback to .env values or defaults."""
         os.environ["OPENAI_MODEL_NAME"] = os.getenv("OPENAI_MODEL_NAME", "gpt-4-turbo-preview")
@@ -15,6 +41,8 @@ class CloudDeployer:
         os.environ["OPENAI_API_BASE"] = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
 
     def run_commands(self):
+        self.create_api_file()
+        self.create_dockerfile()
         """Runs a sequence of shell commands for deployment, continues on error."""
         commands = [
             "yes | gcloud auth configure-docker us-central1-docker.pkg.dev",

{praisonai-0.0.15 → praisonai-0.0.17}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "praisonAI"
-version = "0.0.15"
+version = "0.0.17"
 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 = ""

praisonai-0.0.15/praisonai/auto-old.py

@@ -1,151 +0,0 @@
-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.15/praisonai/cli-old.py

@@ -1,143 +0,0 @@
-# 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
-
-config_list = [
-    {
-        'model': os.environ.get("OPENAI_MODEL_NAME", "gpt-3.5-turbo"),
-        'base_url': os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1"),
-    }
-]
-
-def generate_crew_and_kickoff(agent_file, framework=None):
-    with open(agent_file, 'r') as f:
-        config = yaml.safe_load(f)
-
-    topic = config['topic']  
-    framework = framework or config.get('framework')
-
-    agents = {}
-    tasks = []
-    if framework == "autogen":
-        # Load the LLM configuration dynamically
-        print(config_list)
-        llm_config = {"config_list": 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)+". 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(args=None):
-    if args is None:
-        args = parse_args()
-
-    invocation_cmd = "praisonai"
-    version_string = f"praisonAI version {__version__}"
-    framework = args.framework  # Default framework
-    ui = args.ui  # Default UI flag
-    print(args)
-
-    if args.agent_file:
-        agent_file = args.agent_file
-    else:
-        agent_file = "agents.yaml"
-
-    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()