Create query_solver.py

query_solver.py

@@ -0,0 +1,14 @@
+import sys
+import json
+import autogen
+from autogen.agentchat.contrib.math_user_proxy_agent import MathUserProxyAgent
+
+def run_query(math_problem, api_key):
+    # ... (Same code as before)
+
+if __name__ == "__main__":
+    input_data = json.loads(sys.stdin.read())
+    math_problem = input_data['math_problem']
+    api_key = input_data['api_key']
+    result = run_query(math_problem, api_key)
+    print(result)