adding player class

app.py

@@ -13,7 +13,7 @@ from autogen.agentchat.contrib.retrieve_user_proxy_agent import (
     PROMPT_CODE,
 )
 
-TIMEOUT = 60
+TIMEOUT = 120
 
 
 def initialize_agents(config_list, docs_path=None):
@@ -155,7 +155,7 @@ with gr.Blocks() as demo:
         def update_config(config_list):
             global assistant, ragproxyagent
             config_list = autogen.config_list_from_models(
-                model_list=[os.environ.get("MODEL", "gpt-35-turbo")],
+                model_list=[os.environ.get("MODEL", "gpt-3.5-turbo")],
             )
             if not config_list:
                 config_list = [
@@ -178,12 +178,12 @@ with gr.Blocks() as demo:
             ragproxyagent._model = config_list[0]["model"]
             return config_list
 
-        def set_params(model, oai_key, aoai_key, aoai_base):
+        def set_params(model, oai_key, aoai_key, aoai_base, aoai_context):
             os.environ["MODEL"] = model
             os.environ["OPENAI_API_KEY"] = oai_key
             os.environ["AZURE_OPENAI_API_KEY"] = aoai_key
             os.environ["AZURE_OPENAI_API_BASE"] = aoai_base
-            return model, oai_key, aoai_key, aoai_base
+            return model, oai_key, aoai_key, aoai_base, aoai_context
 
         txt_model = gr.Dropdown(
             label="Model",
@@ -195,6 +195,17 @@ with gr.Blocks() as demo:
             value="gpt-35-turbo",
             container=True,
         )
+        txt_aoai_context = gr.Dropdown(
+            label="Context",
+            choices=[
+                "Quest_Storage",
+                "Track_Function",
+                "Test_Track_Function"
+            ],
+            allow_custom_value=True,
+            value="Quest_Storage",
+            container=True,
+        )
         txt_oai_key = gr.Textbox(
             label="OpenAI API Key",
             placeholder="Enter key and press enter",
@@ -254,9 +265,9 @@ with gr.Blocks() as demo:
         # show_copy_button=True,
     )
 
-    def respond(message, chat_history, model, oai_key, aoai_key, aoai_base):
+    def respond(message, chat_history, model, oai_key, aoai_key, aoai_base, aoai_context):
         global config_list
-        set_params(model, oai_key, aoai_key, aoai_base)
+        set_params(model, oai_key, aoai_key, aoai_base, aoai_context)
         config_list = update_config(config_list)
         messages = chatbot_reply(message)
         _msg = (
@@ -302,7 +313,7 @@ with gr.Blocks() as demo:
 
     txt_input.submit(
         respond,
-        [txt_input, chatbot, txt_model, txt_oai_key, txt_aoai_key, txt_aoai_base_url],
+        [txt_input, chatbot, txt_model, txt_oai_key, txt_aoai_key, txt_aoai_base_url, txt_aoai_context],
         [txt_input, chatbot],
     )
     txt_prompt.submit(update_prompt, [txt_prompt], [txt_prompt])