DONATE TO THIS PROJECT

🦊🐻🐹 TeamForgeAI

AI Agent Collaboration Platform

TeamForgeAI is a platform designed for creating, managing, and collaborating with AI agents. Built using Streamlit, this application allows you to define agents with specific skills and expertise, and then engage them in a group chat setting to solve tasks collaboratively.

Features

AI-Powered Agent Creation: Generate specialized AI agents based on your project description. Each agent possesses unique skills and tools tailored to their role.
Collaborative Workflow: Agents interact and collaborate to solve problems, generate ideas, and complete tasks.
Agent Memory: All agents have the option to retain long-term memory.
Auto Workflow Mode: Let the chat_manager agent take the lead and manage the agents in a group chat workflow!
MoA Agent Mode: This fantastic feature (Mixture of Agents) gives each agent a layered thinking process to generate their responses.
Project Context Continuity: You can switch back and forth between auto and manual mode while working on the same goal. You choose how you want to work without losing your work.
Project Goal, Objectives, and Deliverables: Define and track your project's overall goal, specific objectives, and tangible deliverables.
Skill Integration: Integrate custom Python skills to extend agent capabilities and automate tasks.
Web Search and Content Summarization: Agents can perform web searches and summarize content from provided URLs.
Image Generation: Generate images using Stable Diffusion, powered by the Automatic1111 API.
Interactive Discussion and Whiteboard: Track agent interactions, share ideas, and store code snippets on a virtual whiteboard.
Downloadable Agent and Workflow Files: Export agents and workflows as JSON files for use in AutoGen and CrewAI.
Team Management: Create and manage teams of agents, allowing for flexible collaboration and organization.
Visual Virtual Office: An engaging virtual office with animated emojis representing each agent, providing a fun and intuitive way to visualize agent interactions.

Installation

First things first. You will need Ollama installed and running: https://github.com/ollama/ollama
You need a capable 'instruct' model. The default settings use mistral:instruct. So you'll want to do 'ollama pull mistral:instruct' after you install Ollama.
You also need Stable Diffusion Web UI running and listening in API mode (./webui.sh --listen --api): https://github.com/AUTOMATIC1111/stable-diffusion-webui
For the web_search skill, you will need to get a free API key from Serper: https://serpapi.com/manage-api-key

Clone the repository:

git clone https://github.com/marc-shade/TeamForgeAI.git

Create virtual environment:

conda create --name teamforge python=3.11

conda activate teamforge

Install dependencies:

pip install -r TeamForgeAI/requirements.txt

Start the application:

streamlit run TeamForgeAI/main.py

Updates

You may also need to rerun the requirements.txt if there are new libraries.

The most recent update requires the 'ollama' Python library, which you may have to install manually. I'm having an issue getting it installed via the requirements.txt file.

https://github.com/ollama/ollama-python

pip install -U -r requirements.txt

or if you need to do it manually, you can just do:

pip install ollama

Usage

Create Agents:

Use the sidebar to define new agents.
Provide a name, description, select skills, and list tools for each agent.

Organize Teams:

Create teams to group agents based on projects or tasks.
Save agents into teams

Start a Discussion:

Input your request or problem statement in the main input area, 'Enter your request.'
The application will rephrase the request into an optimized prompt and automatically generate the agents to complete the task!
Add any 'Additional Input' if you need to add to interject in the team's discussion

Interact with Agents:

Click on an agent's button to trigger their response.
Agents will communicate and collaborate within the discussion area.
Customize each agent's settings.
Refine the agent's prompts dynamically based on the Discussion History context.
Click 'Toggle Auto Mode' to put the agents in a group chat for a few turns to work autonomously.

Skills:

The skills are implemented using a 'sure-shot' method, guaranteeing they will trigger without relying solely on an AI prompt. The skills are assigned to an agent, effectively switching them into a sort of 'tools mode' that mandates that they use the tool as a programmatic response. This avoids issues often seen in group chat situations where an agent hallucinates and pretends to run the skill script. When you push the agent button in TeamForgeAI, the agent will always use the skill.

'Web Search' uses Serper, and you must provide your API for this skill to function.
'Fetch Web Content' reads the content of a web page and puts it into the conversation.
'Generate SD Images' creates an image based on the list of requested images that reflect the context.
'Plot Diagram' allows an agent to generate a plot or graph directly in the UI.
'Update Project Status', 'Summarize Project Status', and 'Generate Agent Instructions' empower any agent with these abilities to manage projects and provide status updates to the team.

TeamForgeAI supports custom Python skills that can be integrated into agents. To create a new skill:

Create a new Python file in the skills directory.
Define a function with the same name as the skill.
Add a docstring to describe the skill's functionality.
Save the file with a .py extension.
The skill will be automatically loaded and available for assignment to agents.

Review and Download:

The discussion history is preserved for future reference.
Download agent configurations and workflows as needed.

Examples:

Write a children's book: Request TeamForgeAI to create a children's book about a llama. Agents such as a "Children's Book Writer", "Illustrator", and "Language Specialist" will be generated to collaborate on the project.
Develop a marketing plan: Ask TeamForgeAI to develop a marketing plan for a new product. Agents like a "Market Research Analyst", "Content Creator", and "Social Media Manager" will work together to create a comprehensive plan.
Build a website: Request TeamForgeAI to build a website for your business. Agents such as a "Web Developer", "Designer", and "Content Writer" will collaborate to bring your website to life.

Contributing

We welcome contributions to enhance TeamForgeAI's capabilities. To contribute:

Fork the repository.
Create a new branch for your feature.
Implement your changes and ensure they are well-documented.
Submit a pull request for review.

License

This project is licensed under the MIT License. Portions of this code are derived from AutoGroq by J. Gravelle.

MIT License

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.