You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
DualMind is an innovative AI conversation simulator that facilitates engaging dialogues between two AI models using the Ollama API. It offers a command-line interface (CLI) for immersive and customizable AI interactions.
Features
🤖 Dual-model conversation: Engage two different AI models in a thought-provoking dialogue
🎭 Customizable system prompts: Tailor the behavior and personality of each AI model
🖥️ Command-line interface for quick interactions
🛠️ Conversation customization:
Adjust the number of exchanges
Modify the initial prompt
Select different AI models
💾 Save conversation logs
🔢 Token count display and limit: Monitor the token usage for each message and limit the conversation to a specified number of tokens.
Prerequisite: Ollama
This project is privacy oriented and for such reason it uses Ollama as a backend. You need an Ollama endpoint to use this software.
Please refer to Ollama to install Ollama on your machine.
Installation
Clone the repository:
git clone https://github.com/yourusername/DualMind.git
cd DualMind
Install the required dependencies:
pip install -r requirements.txt
Set up your environment variables by creating a .env file in the project root:
OLLAMA_ENDPOINT=https://localhost:11434
MODEL_1=llama2
MODEL_2=mistral
INITIAL_PROMPT="Let's discuss the future of AI. What are your thoughts on its potential impact on society?"
MAX_TOKENS=8000
Feel free to use the env.example file as a template.
Note: The INITIAL_PROMPT is the first message that will be sent and it will be send on behalf of the second AI to the first AI.
Usage
To run DualMind in CLI mode:
./start_cli.sh
Customization
System Prompts
You can customize the system prompts for each AI model by editing the system_prompt_1.txt and system_prompt_2.txt files in the project root.
Options
You can customize the options for the conversation by editing the options.json file in the project root.
Project Structure
main.py: Entry point of the application
ai_conversation.py: Core logic for AI conversations
start_cli.sh: Shell script to run the CLI version
Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
License
This project is open source and available under the MIT License.
Acknowledgements
This project uses the Ollama API for AI model interactions.
About
DualMind is an innovative AI conversation simulator that facilitates engaging dialogues between two AI models using the Ollama API.