debate.py

import autogen

# Clear the terminal (Unix/Linux/MacOS)
import os
os.system('clear')

class debate:    
    def __init__(self,api_key="NotRequired",llm="Local8801", saved_team='./speaker_selection/debateteam.json'):
        import os
        import json
        
        self.llm=llm
        tdict={"model":llm,"api_key":api_key,"base_url": "http://0.0.0.0:8801"}
        # tdict={"model":llm,"api_key":api_key,"base_url": "http://localhost:11434/v1"}
        os.environ['OAI_CONFIG_LIST']="["+json.dumps(tdict)+"]"
        self.config_file_or_env='OAI_CONFIG_LIST'
        self.saved_team=saved_team
        # self.llm_config={'temperature': 0}Th

        self.llm_config = {
            "timeout": 600,
            "cache_seed": 44,  # change the seed for different trials
            "config_list": autogen.config_list_from_json(
                "OAI_CONFIG_LIST",
                # filter_dict={"model": ["mixtralcopy"]},
            ),
            "temperature": 0,
        }
        os.environ['AUTOGEN_USE_DOCKER']='False'

    def load_team(self):
        import requests
        import tempfile
        import os
        from urllib.parse import urlparse
        from autogen.agentchat.contrib.agent_builder import AgentBuilder
        
        parsed=urlparse(self.saved_team) #checking for url or file designation
        if parsed.scheme and parsed.netloc: #if it is a url
            response=requests.get(self.saved_team)
            with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
                tmp_file_name = tmp_file.name
                # Write the content to the temporary file
                tmp_file.write(response.content)
            file_name=tmp_file_name
        else: # if it is a file
            file_name=self.saved_team
        new_builder = AgentBuilder(config_file_or_env=self.config_file_or_env)
        self.agent_list, self.agent_config = new_builder.load(file_name)
        #self.agent_list, self.agent_config = new_builder.load(r"C:\Users\tevsl\anaconda3\envs\autobuild\debateteam.json")
        if not self.saved_team==file_name: #if temp file
            os.remove(tmp_file_name) #remove it
        
    def do_debate(self,proposition):
        import autogen
        
        config_list = autogen.config_list_from_json(
            self.config_file_or_env,
            filter_dict={"model": [self.llm]})
        
        group_chat = autogen.GroupChat(
            agents=self.agent_list, 
            messages=[],
            # speaker_selection_method="round_robin", # Added for Run 08.
            max_round=15) # Reduced from 15 to 6 for Run 8, then back to 15 from Run 9 onward 
        
        manager = autogen.GroupChatManager(
            groupchat=group_chat,
            llm_config={"config_list": config_list, **self.llm_config}
        )

        self.agent_list[0].initiate_chat(manager, message=proposition)

from dotenv import load_dotenv
import os

load_dotenv()       
dm=debate()
dm.load_team()
proposition=" "
while len(proposition) > 0:  #loop until user terminates
    proposition=input("What proposition would you like to debate (hit ENTER to terminate): ")
    if len(proposition) > 0: #if not terminate
        dm.do_debate(f"Please debate the proposition '{proposition}'.") # The debate will start with Affirmative_Constructive_Debater.")