.

app.py

@@ -13,37 +13,23 @@ from autogen import Agent, AssistantAgent, UserProxyAgent, config_list_from_json
 import chainlit as cl
 
 # Edit the URL Here
-URL = "https://sapes5.sapdevcenter.com/sap/opu/odata/sap/ZPDCDS_SRV/SEPMRA_I_Product_E"
-
-use_docker="python:3"
+URL = "https://www.w3schools.com/xml/simple.xml"
 
 WELCOME_MESSAGE = f"""Datascience Agent Team 👾
 \n\n
 What can we do for you today?
 """
 
+# We plug a context to the Code Planner so that it understands the File Format. This saves us money from tokens since read is cheaper than write.
 CONTEXT = f"""Access the XML data from the following link: {URL}. Utilize the libraries 'urllib.request' and 'xml.etree.ElementTree' for sending a GET request and parsing the XML data, respectively.
 
-The XML data uses three namespaces:
-1. Default namespace: 'http://www.w3.org/2005/Atom'
-2. 'd': 'http://schemas.microsoft.com/ado/2007/08/dataservices'
-3. 'm': 'http://schemas.microsoft.com/ado/2007/08/dataservices/metadata'
-
-The product details are nested within 'entry' -> 'content' -> 'm:properties' tags. Inside 'm:properties', each product detail is stored in a 'd:TagName' format. 
-
-To access details of a specific product, iterate through all 'entry' tags, and for each entry, navigate to the 'd:Product' tag within 'm:properties' to check the product ID. If the product ID matches the desired ID, extract all child tags within 'm:properties' to get the product details.
-
-For queries like 'show the products from supplier 100000076', iterate through all 'entry' tags, and for each entry, navigate to the 'd:Supplier' tag within 'm:properties' to check the supplier ID. If the supplier ID matches the desired ID, extract and accumulate the 'd:Product' tags within 'm:properties' from all matching entries to list all products from that supplier.
-
 Ensure to always check the length of the context to avoid hitting the context limit. Do not express gratitude in responses. If "Thank you" or "You're welcome" are said in the conversation, send a final response. Your final response is just "TERMINATE", do not add other sentences."""
 
 # Agents
 USER_PROXY_NAME = "Query Agent"
-CODING_PLANNER = "SAP ABAP Code Planner"
-CODING_RUNNER = "AP DATA&AI Engineer"
-DATA_ANALYZER = "SAP Analysis Agent"
-GROUPCHAT = "Groupchat"
-MANAGER = "Manager"
+CODING_PLANNER = "Code Planner"
+CODING_RUNNER = "Code Runner"
+DATA_ANALYZER = "Analysis Agent"
 
 async def ask_helper(func, **kwargs):
     res = await func(**kwargs).send()
@@ -99,7 +85,7 @@ class ChainlitUserProxyAgent(UserProxyAgent):
             if res.get("value") == "exit":
                 return "exit"
 
-        reply = cl.run_sync(ask_helper(cl.AskUserMessage, content=prompt, timeout=200))
+        reply = cl.run_sync(ask_helper(cl.AskUserMessage, content=prompt, timeout=60))
 
         return reply["content"].strip()
 
@@ -128,21 +114,25 @@ async def on_chat_start():
   try:
     config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST")
     coding_assistant = ChainlitAssistantAgent(
-        name="SAP_ABAP_Code_Planner", llm_config={"config_list": config_list},
+        name="Code_Planner", llm_config={"config_list": config_list},
         system_message="""Engineer. You follow an approved plan. You write python/shell code to solve tasks. Wrap the code in a code block that specifies the script type. 
                 The user can't modify your code. So do not suggest incomplete code which requires others to modify. Don't use a code block if it's not intended to be executed by the SAP_DATA_and_AI Engineer.
                 Don't include multiple code blocks in one response. Do not ask others to copy and paste the result. Check the execution result returned by the SAP_DATA_and_AI Engineer.
                 If the result indicates there is an error, fix the error and output the code again. Suggest the full code instead of partial code.""" + CONTEXT
     )
-    # coding_runner = ChainlitUserProxyAgent(
-    #     name="SAP_DATA_and_AI_Engineer", llm_config={"config_list": config_list}, human_input_mode="NEVER",
-    #     code_execution_config=False
-    #     system_message="""A Coding Engineer. Use python to run code. Interact with the SAP_ABAP_Code_Planner to run code. Report the result.
-    #             You are an AI model capable of executing code."""
-    # )
+    coding_runner = ChainlitUserProxyAgent(
+        name="Code_Runner", llm_config={"config_list": config_list}, human_input_mode="NEVER",
+        code_execution_config={
+            "last_n_messages": 3,
+            "work_dir": "workspace",
+            "use_docker": True,
+        },
+        system_message="""A Coding Engineer. Use python to run code. Interact with the Code_Planner to run code. Report the result.
+                You are an AI model capable of executing code."""
+    )
     analysis_agent = ChainlitAssistantAgent(
-        name="SAP_Analysis_Agent", llm_config={"config_list": config_list},
-        system_message="""Analysis agent. You analyse the data outputted by SAP_DATA_and_AI_Engineer when necessary. Be concise and always summarize the data when possible.
+        name="Analysis_Agent", llm_config={"config_list": config_list},
+        system_message="""Analysis agent. You analyse the data outputted by Code_Runner when necessary. Be concise and always summarize the data when possible.
                 Communicate with the Query_Agent when the data is analyzed."""
     )
     user_proxy = ChainlitUserProxyAgent(
@@ -150,19 +140,15 @@ async def on_chat_start():
         # human_input_mode="TERMINATE",
         # max_consecutive_auto_reply=3,
         # is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
-        code_execution_config={
-            "last_n_messages": 3,
-            "use_docker": False,
-        },
-        system_message="""Manager. Administrate the agents on a plan, Only run code on Python. Communicate with the SAP_ABAP_Code_Planner to plan the code. 
-                Communicate with the SAP_Analysis_Agent when we want to analyse the data. When there is code from SAP_ABAP_Code_Planner, we run the code in Python.
-                Reply TERMINATE at the end of your sentence if the task has been solved at full satisfaction. 
+        code_execution_config=False,
+        system_message="""Manager. Administrate the agents on a plan. Communicate with the Code_Planner to plan the code. 
+                Communicate with the Analysis_Agent when we want to analyse the data. Reply TERMINATE at the end of your sentence if the task has been solved at full satisfaction. 
                 Otherwise, reply CONTINUE, or the reason why the task is not solved yet.""" 
     )
     
     cl.user_session.set(USER_PROXY_NAME, user_proxy)
     cl.user_session.set(CODING_PLANNER, coding_assistant)
-    # cl.user_session.set(CODING_RUNNER, coding_runner)
+    cl.user_session.set(CODING_RUNNER, coding_runner)
     cl.user_session.set(DATA_ANALYZER, analysis_agent)
     
     await cl.Message(content=WELCOME_MESSAGE, author="Query_Agent").send()
@@ -173,15 +159,15 @@ async def on_chat_start():
 
 @cl.on_message
 async def run_conversation(message: cl.Message):
-  #try:
+  try:
     TASK = message.content
     print("Task: ", TASK)
     coding_assistant = cl.user_session.get(CODING_PLANNER)
     user_proxy = cl.user_session.get(USER_PROXY_NAME)
-    # coding_runner = cl.user_session.get(CODING_RUNNER)
+    coding_runner = cl.user_session.get(CODING_RUNNER)
     analysis_agent = cl.user_session.get(DATA_ANALYZER)
     
-    groupchat = autogen.GroupChat(agents=[user_proxy, coding_assistant, analysis_agent], messages=[], max_round=50)
+    groupchat = autogen.GroupChat(agents=[user_proxy, coding_assistant, coding_runner, analysis_agent], messages=[], max_round=50)
     manager = autogen.GroupChatManager(groupchat=groupchat)
     
     print("GC messages: ", len(groupchat.messages))
@@ -192,6 +178,6 @@ async def run_conversation(message: cl.Message):
     else:
       await cl.make_async(user_proxy.send)( manager, message=TASK, )
       
-#   except Exception as e:
-#     print("Error: ", e)
-#     pass
+  except Exception as e:
+    print("Error: ", e)
+    pass