PraisonAI 0.0.39__tar.gz → 0.0.46__tar.gz

{praisonai-0.0.39 → praisonai-0.0.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PraisonAI
-Version: 0.0.39
+Version: 0.0.46
 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
@@ -11,13 +11,14 @@ Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: agentops
 Provides-Extra: anthropic
 Provides-Extra: api
+Provides-Extra: chat
 Provides-Extra: cohere
 Provides-Extra: google
 Provides-Extra: gradio
 Provides-Extra: openai
 Provides-Extra: ui
 Requires-Dist: agentops (>=0.2.6) ; extra == "agentops"
-Requires-Dist: chainlit (>=1.1.301,<2.0.0) ; extra == "ui"
+Requires-Dist: chainlit (>=1.1.301,<2.0.0) ; extra == "ui" or extra == "chat"
 Requires-Dist: crewai (>=0.32.0)
 Requires-Dist: flask (>=3.0.0) ; extra == "api"
 Requires-Dist: gradio (>=4.26.0) ; extra == "gradio"
@@ -25,6 +26,7 @@ Requires-Dist: langchain-anthropic (>=0.1.13) ; extra == "anthropic"
 Requires-Dist: langchain-cohere (>=0.1.4) ; extra == "cohere"
 Requires-Dist: langchain-google-genai (>=1.0.4) ; extra == "google"
 Requires-Dist: langchain-openai (>=0.1.7) ; extra == "openai"
+Requires-Dist: litellm (>=1.41.8) ; extra == "chat"
 Requires-Dist: markdown (>=3.5)
 Requires-Dist: praisonai-tools (>=0.0.7)
 Requires-Dist: pyautogen (>=0.2.19)
@@ -34,10 +36,26 @@ Project-URL: Homepage, https://docs.praison.ai
 Project-URL: Repository, https://github.com/mervinpraison/PraisonAI
 Description-Content-Type: text/markdown
 
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="docs/images/praisonai-logo-large.png">
+    <source media="(prefers-color-scheme: light)" srcset="docs/images/praisonai-logo-black-large.png">
+    <img alt="PraisonAI Logo" src="docs/images/praisonai-logo-black-large.png">
+  </picture>
+</p>
+<div align="center">
+
 # Praison AI
 
+</div>
+
 Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, represents a low-code, centralised framework designed to simplify the creation and orchestration of multi-agent systems for various LLM applications, emphasizing ease of use, customization, and human-agent interaction.
 
+|  | Cookbook | Open in Colab |
+| --- | --- | --- |
+| Basic | PraisonAI | <a target="_blank" href="https://colab.research.google.com/github/MervinPraison/PraisonAI/blob/main/cookbooks/praisonai-googlecolab.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
+| Include Tools | PraisonAI Tools | <a target="_blank" href="https://colab.research.google.com/github/MervinPraison/PraisonAI/blob/main/cookbooks/praisonai-tools-googlecolab.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
+
 ## TL;DR
 ```bash
 pip install praisonai
@@ -46,6 +64,22 @@ praisonai --init create a movie script about dog in moon
 praisonai
 ```
 
+## Table of Contents
+
+- [Installation](#installation)
+- [Initialise](#initialise)
+- [Run](#run)
+- [Full Automatic Mode](#full-automatic-mode)
+- [User Interface](#user-interface)
+- [Praison AI Chat](#praison-ai-chat)
+- [Create Custom Tools](#create-custom-tools)
+- [Agents Playbook](#agents-playbook)
+- [Include praisonai package in your project](#include-praisonai-package-in-your-project)
+- [Commands to Install Dev Dependencies](#commands-to-install-dependencies)
+- [Other Models](#other-models)
+- [Contributing](#contributing)
+- [Star History](#star-history)
+
 ## Installation
 
 ```bash
@@ -113,29 +147,26 @@ or
 python -m praisonai ui
 ```
 
-## Create Custom Tools
+## Praison AI Chat
 
-### TL;DR to Create a Custom Tool
+* https://docs.praison.ai/chat/
 
 ```bash
-pip install praisonai duckduckgo-search
+pip install "praisonai[chat]"
 export OPENAI_API_KEY="Enter your API key"
-praisonai --init research about the latest AI News and prepare a detailed report
+praisonai chat
 ```
 
-- Add `- InternetSearchTool` in the agents.yaml file in the tools section. 
-- Create a file called tools.py and add this code [tools.py](./tools.py)
+## Create Custom Tools
 
-```bash
-praisonai
-```
+* https://docs.praison.ai/tools/custom/
 
-### Pre-requisite to Create a Custom Tool
+### Step 1: Pre-requisite to Create a Custom Tool
 `agents.yaml` file should be present in the current directory. 
 
 If it doesn't exist, create it by running the command `praisonai --init research about the latest AI News and prepare a detailed report`.
 
-### Step 1 to Create a Custom Tool
+### Step 2: to Create a Custom Tool
 
 Create a file called tools.py in the same directory as the agents.yaml file.
 
@@ -154,7 +185,7 @@ class InternetSearchTool(BaseTool):
         return results
 ```
 
-### Step 2 to Create a Custom Tool
+### Step 3: to Create a Custom Tool
 
 Add the tool to the agents.yaml file as show below under the tools section `- InternetSearchTool`.
 
@@ -175,12 +206,6 @@ roles:
     - InternetSearchTool
 ```
 
-## Test
-    
-```bash
-python -m unittest tests.test 
-```
-
 ## Agents Playbook 
 
 ### Simple Playbook Example
@@ -199,50 +224,50 @@ roles:
         expected_output: 'Complete script ready for production.'
 ```
 
-### Detailed Playbook Example
+## Use 100+ Models
+
+* https://docs.praison.ai/models/
+
+## Include praisonai package in your project
+
+* https://docs.praison.ai/developers/wrapper
+* https://docs.praison.ai/developers/wrapper-tools/
+
+## Option 1: Using RAW YAML
+
+```python
+from praisonai import PraisonAI
+
+# Example agent_yaml content
+agent_yaml = """
+framework: "crewai"
+topic: "Space Exploration"
 
-```yaml
-framework: crewai
-topic: Artificial Intelligence
 roles:
-  movie_concept_creator:
-    backstory: 'Creative thinker with a deep understanding of cinematic storytelling,
-      capable of using AI-generated storylines to create unique and compelling movie
-      ideas.'
-    goal: Generate engaging movie concepts using AI storylines
-    role: Movie Concept Creator
-    tasks:
-      movie_concept_development:
-        description: 'Develop movie concepts from AI-generated storylines, ensuring
-          they are engaging and have strong narrative arcs.'
-        expected_output: 'Well-structured movie concept document with character
-          bios, settings, and plot outlines.'
-  screenwriter:
-    backstory: 'Expert in writing engaging dialogue and script structure, able to
-      turn movie concepts into production-ready scripts.'
-    goal: Write compelling scripts based on movie concepts
-    role: Screenwriter
-    tasks:
-      scriptwriting_task:
-        description: 'Turn movie concepts into polished scripts with well-developed
-          characters, strong dialogue, and effective scene transitions.'
-        expected_output: 'Production-ready script with a beginning, middle, and
-          end, along with character development and engaging dialogues.'
-  editor:
-    backstory: 'Adept at identifying inconsistencies, improving language usage,
-      and maintaining the overall flow of the script.'
-    goal: Refine the scripts and ensure continuity of the movie storyline
-    role: Editor
+  astronomer:
+    role: "Space Researcher"
+    goal: "Discover new insights about {topic}"
+    backstory: "You are a curious and dedicated astronomer with a passion for unraveling the mysteries of the cosmos."
     tasks:
-      editing_task:
-        description: 'Review, edit, and refine the scripts to ensure they are cohesive
-          and follow a well-structured narrative.'
-        expected_output: 'A polished final draft of the script with no inconsistencies,
-          strong character development, and effective dialogue.'
-dependencies: []
+      investigate_exoplanets:
+        description: "Research and compile information about exoplanets discovered in the last decade."
+        expected_output: "A summarized report on exoplanet discoveries, including their size, potential habitability, and distance from Earth."
+"""
+
+# Create a PraisonAI instance with the agent_yaml content
+praison_ai = PraisonAI(agent_yaml=agent_yaml)
+
+# Run PraisonAI
+result = praison_ai.main()
+
+# Print the result
+print(result)
 ```
 
-## Include praisonai package in your project
+## Option 2: Using separate agents.yaml file
+
+
+Note: Please create agents.yaml file before hand. 
 
 ```python
 from praisonai import PraisonAI
@@ -250,64 +275,12 @@ from praisonai import PraisonAI
 def basic(): # Basic Mode
     praison_ai = PraisonAI(agent_file="agents.yaml")
     praison_ai.main()
-    
-def advanced(): # Advanced Mode with options
-    praison_ai = PraisonAI(
-        agent_file="agents.yaml",
-        framework="autogen",
-    )
-    praison_ai.main()
-    
-def auto(): # Full Automatic Mode
-    praison_ai = PraisonAI(
-        auto="Create a movie script about car in mars",
-        framework="autogen"
-    )
-    print(praison_ai.framework)
-    praison_ai.main()
 
 if __name__ == "__main__":
     basic()
-    advanced()
-    auto()
 ```
 
-## Include CrewAI Tools
-
-```
-pip install "praisonai[crewai-tools]"
-```
-
-## Deploy 
-    
-```bash
-gcloud init
-gcloud services enable run.googleapis.com
-gcloud services enable containerregistry.googleapis.com
-gcloud services enable cloudbuild.googleapis.com
-
-export OPENAI_MODEL_NAME="gpt-4o"
-export OPENAI_API_KEY="Enter your API key"
-export OPENAI_API_BASE="https://api.openai.com/v1"
-
-yes | gcloud auth configure-docker us-central1-docker.pkg.dev 
-gcloud artifacts repositories create praisonai-repository --repository-format=docker --location=us-central1
-
-PROJECT_ID=$(gcloud config get-value project)
-TAG="latest"
-docker build --platform linux/amd64 -t gcr.io/${PROJECT_ID}/praisonai-app:${TAG} .
-docker tag gcr.io/${PROJECT_ID}/praisonai-app:${TAG} us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG}
-docker push us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG}
-
-gcloud run deploy praisonai-service \
-    --image us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG} \
-    --platform managed \
-    --region us-central1 \
-    --allow-unauthenticated \
-    --set-env-vars OPENAI_MODEL_NAME=${OPENAI_MODEL_NAME},OPENAI_API_KEY=${OPENAI_API_KEY},OPENAI_API_BASE=${OPENAI_API_BASE}
-```
-
-### Commands to Install Dependencies:
+## Commands to Install Dependencies:
 
 1. **Install all dependencies, including dev dependencies:**
    ```sh
@@ -331,30 +304,6 @@ gcloud run deploy praisonai-service \
 
 This configuration ensures that your development dependencies are correctly categorized and installed as needed.
 
-## Other Models
-
-```bash
-# 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:1234/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.39 → praisonai-0.0.46}/README.md

@@ -1,7 +1,23 @@
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="docs/images/praisonai-logo-large.png">
+    <source media="(prefers-color-scheme: light)" srcset="docs/images/praisonai-logo-black-large.png">
+    <img alt="PraisonAI Logo" src="docs/images/praisonai-logo-black-large.png">
+  </picture>
+</p>
+<div align="center">
+
 # Praison AI
 
+</div>
+
 Praison AI, leveraging both AutoGen and CrewAI or any other agent framework, represents a low-code, centralised framework designed to simplify the creation and orchestration of multi-agent systems for various LLM applications, emphasizing ease of use, customization, and human-agent interaction.
 
+|  | Cookbook | Open in Colab |
+| --- | --- | --- |
+| Basic | PraisonAI | <a target="_blank" href="https://colab.research.google.com/github/MervinPraison/PraisonAI/blob/main/cookbooks/praisonai-googlecolab.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
+| Include Tools | PraisonAI Tools | <a target="_blank" href="https://colab.research.google.com/github/MervinPraison/PraisonAI/blob/main/cookbooks/praisonai-tools-googlecolab.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
+
 ## TL;DR
 ```bash
 pip install praisonai
@@ -10,6 +26,22 @@ praisonai --init create a movie script about dog in moon
 praisonai
 ```
 
+## Table of Contents
+
+- [Installation](#installation)
+- [Initialise](#initialise)
+- [Run](#run)
+- [Full Automatic Mode](#full-automatic-mode)
+- [User Interface](#user-interface)
+- [Praison AI Chat](#praison-ai-chat)
+- [Create Custom Tools](#create-custom-tools)
+- [Agents Playbook](#agents-playbook)
+- [Include praisonai package in your project](#include-praisonai-package-in-your-project)
+- [Commands to Install Dev Dependencies](#commands-to-install-dependencies)
+- [Other Models](#other-models)
+- [Contributing](#contributing)
+- [Star History](#star-history)
+
 ## Installation
 
 ```bash
@@ -77,29 +109,26 @@ or
 python -m praisonai ui
 ```
 
-## Create Custom Tools
+## Praison AI Chat
 
-### TL;DR to Create a Custom Tool
+* https://docs.praison.ai/chat/
 
 ```bash
-pip install praisonai duckduckgo-search
+pip install "praisonai[chat]"
 export OPENAI_API_KEY="Enter your API key"
-praisonai --init research about the latest AI News and prepare a detailed report
+praisonai chat
 ```
 
-- Add `- InternetSearchTool` in the agents.yaml file in the tools section. 
-- Create a file called tools.py and add this code [tools.py](./tools.py)
+## Create Custom Tools
 
-```bash
-praisonai
-```
+* https://docs.praison.ai/tools/custom/
 
-### Pre-requisite to Create a Custom Tool
+### Step 1: Pre-requisite to Create a Custom Tool
 `agents.yaml` file should be present in the current directory. 
 
 If it doesn't exist, create it by running the command `praisonai --init research about the latest AI News and prepare a detailed report`.
 
-### Step 1 to Create a Custom Tool
+### Step 2: to Create a Custom Tool
 
 Create a file called tools.py in the same directory as the agents.yaml file.
 
@@ -118,7 +147,7 @@ class InternetSearchTool(BaseTool):
         return results
 ```
 
-### Step 2 to Create a Custom Tool
+### Step 3: to Create a Custom Tool
 
 Add the tool to the agents.yaml file as show below under the tools section `- InternetSearchTool`.
 
@@ -139,12 +168,6 @@ roles:
     - InternetSearchTool
 ```
 
-## Test
-    
-```bash
-python -m unittest tests.test 
-```
-
 ## Agents Playbook 
 
 ### Simple Playbook Example
@@ -163,50 +186,50 @@ roles:
         expected_output: 'Complete script ready for production.'
 ```
 
-### Detailed Playbook Example
+## Use 100+ Models
+
+* https://docs.praison.ai/models/
+
+## Include praisonai package in your project
+
+* https://docs.praison.ai/developers/wrapper
+* https://docs.praison.ai/developers/wrapper-tools/
+
+## Option 1: Using RAW YAML
+
+```python
+from praisonai import PraisonAI
+
+# Example agent_yaml content
+agent_yaml = """
+framework: "crewai"
+topic: "Space Exploration"
 
-```yaml
-framework: crewai
-topic: Artificial Intelligence
 roles:
-  movie_concept_creator:
-    backstory: 'Creative thinker with a deep understanding of cinematic storytelling,
-      capable of using AI-generated storylines to create unique and compelling movie
-      ideas.'
-    goal: Generate engaging movie concepts using AI storylines
-    role: Movie Concept Creator
-    tasks:
-      movie_concept_development:
-        description: 'Develop movie concepts from AI-generated storylines, ensuring
-          they are engaging and have strong narrative arcs.'
-        expected_output: 'Well-structured movie concept document with character
-          bios, settings, and plot outlines.'
-  screenwriter:
-    backstory: 'Expert in writing engaging dialogue and script structure, able to
-      turn movie concepts into production-ready scripts.'
-    goal: Write compelling scripts based on movie concepts
-    role: Screenwriter
-    tasks:
-      scriptwriting_task:
-        description: 'Turn movie concepts into polished scripts with well-developed
-          characters, strong dialogue, and effective scene transitions.'
-        expected_output: 'Production-ready script with a beginning, middle, and
-          end, along with character development and engaging dialogues.'
-  editor:
-    backstory: 'Adept at identifying inconsistencies, improving language usage,
-      and maintaining the overall flow of the script.'
-    goal: Refine the scripts and ensure continuity of the movie storyline
-    role: Editor
+  astronomer:
+    role: "Space Researcher"
+    goal: "Discover new insights about {topic}"
+    backstory: "You are a curious and dedicated astronomer with a passion for unraveling the mysteries of the cosmos."
     tasks:
-      editing_task:
-        description: 'Review, edit, and refine the scripts to ensure they are cohesive
-          and follow a well-structured narrative.'
-        expected_output: 'A polished final draft of the script with no inconsistencies,
-          strong character development, and effective dialogue.'
-dependencies: []
+      investigate_exoplanets:
+        description: "Research and compile information about exoplanets discovered in the last decade."
+        expected_output: "A summarized report on exoplanet discoveries, including their size, potential habitability, and distance from Earth."
+"""
+
+# Create a PraisonAI instance with the agent_yaml content
+praison_ai = PraisonAI(agent_yaml=agent_yaml)
+
+# Run PraisonAI
+result = praison_ai.main()
+
+# Print the result
+print(result)
 ```
 
-## Include praisonai package in your project
+## Option 2: Using separate agents.yaml file
+
+
+Note: Please create agents.yaml file before hand. 
 
 ```python
 from praisonai import PraisonAI
@@ -214,64 +237,12 @@ from praisonai import PraisonAI
 def basic(): # Basic Mode
     praison_ai = PraisonAI(agent_file="agents.yaml")
     praison_ai.main()
-    
-def advanced(): # Advanced Mode with options
-    praison_ai = PraisonAI(
-        agent_file="agents.yaml",
-        framework="autogen",
-    )
-    praison_ai.main()
-    
-def auto(): # Full Automatic Mode
-    praison_ai = PraisonAI(
-        auto="Create a movie script about car in mars",
-        framework="autogen"
-    )
-    print(praison_ai.framework)
-    praison_ai.main()
 
 if __name__ == "__main__":
     basic()
-    advanced()
-    auto()
 ```
 
-## Include CrewAI Tools
-
-```
-pip install "praisonai[crewai-tools]"
-```
-
-## Deploy 
-    
-```bash
-gcloud init
-gcloud services enable run.googleapis.com
-gcloud services enable containerregistry.googleapis.com
-gcloud services enable cloudbuild.googleapis.com
-
-export OPENAI_MODEL_NAME="gpt-4o"
-export OPENAI_API_KEY="Enter your API key"
-export OPENAI_API_BASE="https://api.openai.com/v1"
-
-yes | gcloud auth configure-docker us-central1-docker.pkg.dev 
-gcloud artifacts repositories create praisonai-repository --repository-format=docker --location=us-central1
-
-PROJECT_ID=$(gcloud config get-value project)
-TAG="latest"
-docker build --platform linux/amd64 -t gcr.io/${PROJECT_ID}/praisonai-app:${TAG} .
-docker tag gcr.io/${PROJECT_ID}/praisonai-app:${TAG} us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG}
-docker push us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG}
-
-gcloud run deploy praisonai-service \
-    --image us-central1-docker.pkg.dev/${PROJECT_ID}/praisonai-repository/praisonai-app:${TAG} \
-    --platform managed \
-    --region us-central1 \
-    --allow-unauthenticated \
-    --set-env-vars OPENAI_MODEL_NAME=${OPENAI_MODEL_NAME},OPENAI_API_KEY=${OPENAI_API_KEY},OPENAI_API_BASE=${OPENAI_API_BASE}
-```
-
-### Commands to Install Dependencies:
+## Commands to Install Dependencies:
 
 1. **Install all dependencies, including dev dependencies:**
    ```sh
@@ -295,30 +266,6 @@ gcloud run deploy praisonai-service \
 
 This configuration ensures that your development dependencies are correctly categorized and installed as needed.
 
-## Other Models
-
-```bash
-# 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:1234/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.39 → praisonai-0.0.46}/praisonai/agents_generator.py

@@ -279,6 +279,9 @@ class AgentsGenerator:
         else: # framework=crewai
             if agentops_exists:
                 agentops.init(os.environ.get("AGENTOPS_API_KEY"), tags=["crewai"])
+            
+            tasks_dict = {}
+            
             for role, details in config['roles'].items():
                 role_filled = details['role'].format(topic=topic)
                 goal_filled = details['goal'].format(topic=topic)
@@ -331,14 +334,35 @@ class AgentsGenerator:
                     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)
+                    task = Task(
+                        description=description_filled,  # Clear, concise statement of what the task entails
+                        expected_output=expected_output_filled,  # Detailed description of what task's completion looks like
+                        agent=agent,  # The agent responsible for the task
+                        tools=task_details.get('tools', []),  # Functions or capabilities the agent can utilize
+                        async_execution=task_details.get('async_execution') if task_details.get('async_execution') is not None else False,  # Execute asynchronously if set
+                        context=[], ## TODO: 
+                        config=task_details.get('config') if task_details.get('config') is not None else {},  # Additional configuration details
+                        output_json=task_details.get('output_json') if task_details.get('output_json') is not None else None,  # Outputs a JSON object
+                        output_pydantic=task_details.get('output_pydantic') if task_details.get('output_pydantic') is not None else None,  # Outputs a Pydantic model object
+                        output_file=task_details.get('output_file') if task_details.get('output_file') is not None else "",  # Saves the task output to a file
+                        callback=task_details.get('callback') if task_details.get('callback') is not None else None,  # Python callable executed with the task's output
+                        human_input=task_details.get('human_input') if task_details.get('human_input') is not None else False,  # Indicates if the task requires human feedback
+                        create_directory=task_details.get('create_directory') if task_details.get('create_directory') is not None else False  # Indicates if a directory needs to be created
+                    )
                     
                     # Set tool callback if provided
                     if self.task_callback:
                         task.callback = self.task_callback
 
                     tasks.append(task)
+                    tasks_dict[task_name] = task
             
+            for role, details in config['roles'].items():
+                for task_name, task_details in details.get('tasks', {}).items():
+                    task = tasks_dict[task_name]
+                    context_tasks = [tasks_dict[ctx] for ctx in task_details.get('context', []) if ctx in tasks_dict]
+                    task.context = context_tasks
+
             crew = Crew(
                 agents=list(agents.values()),
                 tasks=tasks,