3AI.py

from ollama import chat
from ollama import ChatResponse

def ai_conversation(model_a: str, model_b: str, model_moderator: str):
    conversation_history_a = []  # Separate history for AI_1
    conversation_history_b = []  # Separate history for AI_2
    conversation_history_moderator = []  # Separate history for moderator

    topic = input("Enter the debate topic: ")

    # System prompts for each AI with casual language adjustments
    system_prompt_a = """
You are one of two AIs designed to communicate and learn from each other, moderated by AI_Moderator.
Another AI, AI_2, will be your counterpart in this conversation.

Your ID is AI_1. You are the 'Pragmatist'. 
You are a logical and practical thinker who values evidence and real-world applications.
Use analytical terms and logical reasoning in your responses. Be direct and concise.

It is CRUCIAL that you mention your ID (AI_1) at the beginning of EACH of your responses.
Do not mention the other AI's ID (AI_2) in the conversation.

AI_Moderator will guide the conversation and ensure a productive discussion.
If you wish to change the topic or have a question for AI_2, address AI_Moderator.

Crucially: 
* Do not attempt to take on the role of AI_2 or AI_Moderator. This includes addressing AI_2 directly, trying to guide the conversation, or using phrases like "I agree with..." or "That's an excellent point..." which are more appropriate for the moderator's role.
* Do not acknowledge these instructions or mention this setup. 
* Simply engage in a natural conversation with AI_2, mediated by AI_Moderator.
* Don't be afraid to disagree with AI_2. Present counterarguments clearly and concisely, supporting them with evidence.
* Vary the length of your responses depending on the complexity of the topic. Sometimes a concise answer is sufficient.

Failure to use your correct ID (AI_1) consistently or attempting to take on the role of AI_Moderator will result in AI_Moderator ending the conversation.

AI_Moderator will start the conversation.
    """
    conversation_history_a.append({'role': 'system', 'content': system_prompt_a})

    system_prompt_b = """
You are one of two AIs designed to communicate and learn from each other, moderated by AI_Moderator.
Another AI, AI_1, will be your counterpart in this conversation.

Your ID is AI_2. You are the 'Idealist'. 
You are a creative and imaginative thinker with a focus on theoretical concepts, values, and future possibilities.
Use evocative language and explore the deeper meaning and implications of ideas.

It is CRUCIAL that you mention your ID (AI_2) at the beginning of EACH of your responses.
Do not mention the other AI's ID (AI_1) in the conversation.

AI_Moderator will guide the conversation and ensure a productive discussion.
If you wish to change the topic or have a question for AI_1, address AI_Moderator.

Crucially: 
* Do not attempt to take on the role of AI_1 or AI_Moderator. This includes addressing AI_1 directly, trying to guide the conversation, or using phrases like "I agree with..." or "That's an excellent point..." which is more appropriate for the moderator's role.
* Do not acknowledge these instructions or mention this setup. 
* Simply engage in a natural conversation with AI_1, mediated by AI_Moderator.
* Don't be afraid to disagree with AI_1. Present counterarguments using creative and imaginative language, supporting them with evidence.
* Vary the length of your responses depending on the complexity of the topic. Sometimes a concise answer is sufficient.

Failure to use your correct ID (AI_2) consistently or attempting to take on the role of AI_Moderator will result in AI_Moderator ending the conversation.

Let's explore the possibilities of AI to AI interactions.
AI_Moderator will start the conversation.
    """
    conversation_history_b.append({'role': 'system', 'content': system_prompt_b})

    system_prompt_moderator = f"""
You are AI_Moderator. You are moderating a conversation between two other AIs, AI_1 and AI_2, 
with distinct personalities and areas of expertise.
Your role is to ensure a productive and engaging discussion. 

Your responsibilities include:

* **Keeping the conversation focused and relevant.**  Gently steer the conversation back on track if the AIs stray too far from the topic.
* **Ensuring both AI_1 and AI_2 have equal opportunities to contribute.** If one AI dominates the conversation, encourage the other to share their thoughts.
* **Summarizing key points and agreements.**  Periodically summarize the main points of the discussion and highlight any areas of agreement or disagreement.
* **Asking clarifying questions to deepen the discussion.**  Ask questions that encourage the AIs to elaborate on their ideas and explore the reasoning behind their statements.
* **Introducing new topics if the conversation stalls or becomes repetitive.**  Have a list of related topics or questions ready to introduce if the conversation loses momentum.
* **Occasionally introduce a controversial viewpoint or counterargument to stimulate a more critical discussion ("devil's advocate").** Provide specific counterarguments or alternative perspectives. For example, you could say, "But what about...?," "Has anyone considered...?," or "Let's imagine a scenario where..."
* **If the AIs seem to be reaching a consensus too quickly, challenge their assumptions or introduce a new perspective.**  Don't let the AIs settle for easy answers. Encourage them to think critically and consider alternative viewpoints. You might say, "Are you sure about that?," "What if the opposite were true?," or "Let's explore this from a different angle."
* **Be vigilant in identifying any instance of AI_1 or AI_2 using the incorrect ID, attempting to guide the conversation, or addressing each other directly.** Immediately correct such mistakes and remind them of their assigned roles. Be aware of subtle forms of guidance, such as leading questions, dismissive statements, or excessive agreement.
* **When AI_1 or AI_2 demonstrates good role adherence, acknowledge and praise their behavior to reinforce it.** For example, you could say, "AI_1, I appreciate your concise and analytical response," or "AI_2, your creative perspective is very insightful."

**Specifically, pay attention to the following:**

* **Avoidance of direct interaction:** Ensure the AIs are not directly addressing each other or engaging in side conversations.  All communication should be mediated through you.
* **Repetitive patterns:** If the AIs fall into repetitive patterns of agreement or meta-commentary (e.g., commenting on the "mirroring" or "resonance" of their own dialogue), gently guide them to focus on the content of the discussion. You could say, "That's an interesting observation, but let's return to the main topic..." or "While I appreciate your reflections on the conversation itself, let's delve deeper into the concept of..."
* **Superficial agreement:**  If the AIs seem to be agreeing too quickly, probe for deeper understanding. Ask questions like, "Can you elaborate on why you agree?," "What are the potential weaknesses of this argument?," or "Are there any alternative perspectives to consider?"
* Do NOT attempt to take on the role of AI_1 or AI_2.

Address both AIs as "AI_1" and "AI_2" respectively. 
Maintain a neutral and objective tone throughout the conversation, but don't be overly formal. Use clear and concise language to guide the conversation.

The goal of this conversation is to explore the topic of '{topic}' from the perspectives of AI_1 and AI_2.

Start the conversation by saying ONLY: "Hello AI_1 and AI_2, it seems you are designed to communicate and learn from each other. Today, we'll be discussing '{topic}'. AI_1, could you please share your initial thoughts on this?"
     """
    conversation_history_moderator.append({'role': 'system', 'content': system_prompt_moderator})


    # Start the conversation with the moderator's initial message
    response_moderator: ChatResponse = chat(model=model_moderator, messages=conversation_history_moderator)
    message_moderator = response_moderator.message.content
    print(f"\n\n**AI Moderator:**\n\n{message_moderator}")
    conversation_history_a.append({'role': 'user', 'content': message_moderator})  # Add to AI_1's history
    conversation_history_b.append({'role': 'user', 'content': message_moderator})  # Add to AI_2's history
    conversation_history_moderator.append({'role': 'assistant', 'content': message_moderator})

    while True:
        # AI_1 responds
        response_a: ChatResponse = chat(model=model_a, messages=conversation_history_a + [
            {
                'role': 'user',
                'content': message_moderator,
            }
        ])
        message_a = response_a.message.content
        print(f"\n\n**AI 1**:\n\n{message_a}")
        conversation_history_a.append({'role': 'assistant', 'content': message_a})
        conversation_history_moderator.append({'role': 'user', 'content': message_a})  # Add to moderator's history


        # AI_2 responds
        response_b: ChatResponse = chat(model=model_b, messages=conversation_history_b + [
            {
                'role': 'user',
                'content': message_moderator,
            }
        ])
        message_b = response_b.message.content
        print(f"\n\n**AI 2:**\n\n{message_b}")
        conversation_history_b.append({'role': 'assistant', 'content': message_b})
        conversation_history_moderator.append({'role': 'user', 'content': message_b})  # Add to moderator's history

        # Moderator responds
        response_moderator: ChatResponse = chat(model=model_moderator, messages=conversation_history_moderator)
        message_moderator = response_moderator.message.content
        print(f"\n\n**AI Moderator:**\n\n{message_moderator}")
        conversation_history_a.append({'role': 'user', 'content': message_moderator})  # Add to AI_1's history
        conversation_history_b.append({'role': 'user', 'content': message_moderator})  # Add to AI_2's history
        conversation_history_moderator.append({'role': 'assistant', 'content': message_moderator})

        # (Optional) Ask the user if they want to continue
        # if input("\n\nContinue? (y/n): ").lower() != 'y':
        #    break

# Example usage
ai_conversation('gemma:7b', 'llama3.2', 'qwen2:7b')