Version 0.0.16

Dockerfile

@@ -2,6 +2,6 @@ FROM python:3.11-slim
 WORKDIR /app
 # Copy only api.py and agents.yaml to /app in the container
 COPY api.py agents.yaml ./
-RUN pip install flask praisonai==0.0.15 gunicorn markdown
+RUN pip install flask praisonai==0.0.16 gunicorn markdown
 EXPOSE 8080
 CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]

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.

agents.yaml

@@ -1,41 +1,31 @@
-framework: "crewai"
-topic: "Artificial Intelligence"
-
+framework: autogen
+topic: create movie script about cat in mars
 roles:
-  narrative_designer:
-    role: "Narrative Designer"
-    goal: "Craft engaging storylines and narratives for {topic}"
-    backstory: >
-      You're a creative visionary skilled in weaving complex narratives and 
-      developing engaging story worlds within {topic}. Your expertise lies in 
-      conceptualizing the overarching story and ensuring it resonates with the audience.
+  concept_developer:
+    backstory: Experienced in creating captivating and original story concepts.
+    goal: Generate a unique concept for a movie script about a cat in Mars
+    role: Concept Developer
     tasks:
-      story_concept_development:
-        description: >
-          Develop a compelling story concept for {topic}, focusing on originality, 
-          thematic depth, and audience engagement. Outline the main narrative arcs,
-          characters, and settings.
-        expected_output: >
-          A comprehensive document outlining the story concept, including key narrative 
-          arcs, character bios, and settings descriptions.
-
+      concept_generation:
+        description: Develop a unique and engaging concept for a movie script about
+          a cat in Mars.
+        expected_output: A detailed concept document for the movie script.
   scriptwriter:
-    role: "Scriptwriter"
-    goal: "Transform story concepts into detailed scripts for {topic}"
-    backstory: >
-      With a flair for dialogue and a deep understanding of pacing and structure,
-      you excel at bringing narratives to life through scriptwriting. Your work 
-      turns story concepts into blueprints for compelling {topic} experiences.
+    backstory: Expert in dialogue and script structure, translating concepts into
+      scripts.
+    goal: Write a script based on the movie concept
+    role: Scriptwriter
     tasks:
       scriptwriting_task:
-        description: >
-          Based on the narrative concept, write a detailed script for {topic}. 
-          Include dialogue, stage directions, and scene descriptions that bring 
-          the story to life.
-        expected_output: >
-          A detailed script ready for production, including dialogue, stage 
-          directions, and scene descriptions.
-
-dependencies:
- - task: scriptwriting_task
-   depends_on: story_concept_development  
+        description: Turn the movie concept into a script, including dialogue and
+          scenes.
+        expected_output: A production-ready script for the movie about a cat in Mars.
+  director:
+    backstory: Experienced in visualizing scripts and creating compelling storyboards.
+    goal: Create a storyboard and visualize the script
+    role: Director
+    tasks:
+      storyboard_creation:
+        description: Create a storyboard for the movie script about a cat in Mars.
+        expected_output: A detailed storyboard for the movie about a cat in Mars.
+dependencies: []

praisonai/auto.py

@@ -65,7 +65,7 @@ def convert_and_save(self, json_data):
         """
 
         yaml_data = {
-            "framework": "crewai",
+            "framework": self.framework,
             "topic": self.topic,
             "roles": {},
             "dependencies": []