Update query_solver.py

query_solver.py

@@ -1,10 +1,10 @@
 import sys
 import json
 import autogen
-from autogen.agentchat.contrib.math_user_proxy_agent import MathUserProxyAgent
+from autogen import config_list_from_json
 
 # Function to run the query
-def run_query(math_problem, api_key):
+def run_query(programming_problem, api_key):
     config_list = [
         {
             'model': 'gpt-3.5-turbo',
@@ -39,12 +39,12 @@ def run_query(math_problem, api_key):
     manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=llm_config)
 
 
-    return user_proxy.initiate_chat(manager, message=math_problem)
+    return user_proxy.initiate_chat(manager, message=programming_problem)
    
 
 if __name__ == "__main__":
     input_data = json.loads(sys.stdin.read())
-    math_problem = input_data['math_problem']
+    programming_problem = input_data['programming_problem']
     api_key = input_data['api_key']
-    result = run_query(math_problem, api_key)
+    result = run_query(programming_problem, api_key)
     print(result)