added AGIEval module

app.py

@@ -21,7 +21,7 @@ def retrieve_api_key(url):
     return api_key
     
     
-def get_benchmark_uids(num_miner):
+def get_benchmark_uids(num_miner,mode):
     
     url="test.neuralinternet.ai"
     api_key = retrieve_api_key(url)
@@ -36,7 +36,7 @@ def get_benchmark_uids(num_miner):
             "Endpoint-Version": "2023-05-19",
         }
     
-    conn.request("GET", f"/top_miner_uids?n={num_miner}", headers=headers)
+    conn.request("GET", f"/top_miner_uids?n={num_miner}&mode={mode}", headers=headers)
     miner_response = conn.getresponse()
     miner_data = (
             miner_response.read().decode("utf-8").replace("\n", "").replace("\t", "")
@@ -85,18 +85,28 @@ def interface_fn(system_prompt, optn, arg, user_prompt):
     
     if optn == 'TOP':
         
-        if int(arg) > 30:
-            arg = 100
+        if int(arg) > 50:
+            arg = 50
         payload['top_n'] = int(arg)
         payload['messages'] = messages
         response = retrieve_response(payload)
         return response
     
-    elif optn == 'BENCHMARK':
+    elif optn == 'BENCHMARK_TextEval':
         
-        if int(arg) > 30:
-            arg = 30
-        uids = get_benchmark_uids(int(arg))
+        if int(arg) > 50:
+            arg = 50
+        uids = get_benchmark_uids(int(arg), 'TextEval')
+        payload['uids'] = uids
+        payload['messages'] = messages
+        response = retrieve_response(payload)
+        return response
+
+    elif optn == 'BENCHMARK_AGIEval':
+        
+        if int(arg) > 50:
+            arg = 50
+        uids = get_benchmark_uids(int(arg), 'AGIEval')
         payload['uids'] = uids
         payload['messages'] = messages
         response = retrieve_response(payload)
@@ -119,7 +129,7 @@ interface = gr.Interface(
     fn=interface_fn,
     inputs=[
         gr.inputs.Textbox(label="System Prompt", optional=True),
-        gr.inputs.Dropdown(["TOP", "BENCHMARK", "UIDs"], label="Select Function"),
+        gr.inputs.Dropdown(["TOP", "BENCHMARK_TextEval", "BENCHMARK_AGIEval", "UIDs"], label="Select Function"),
         gr.inputs.Textbox(label="Arguement"),
         gr.inputs.Textbox(label="Enter your question")
         ],
@@ -127,7 +137,7 @@ interface = gr.Interface(
     title="Explore Bittensor Miners",
     description="Enter parameters as per you want and get response",
     examples=[["Your task is to provide consise response of user prompts", "TOP", 5, 'What is Bittensor?']
-            ,["Your task is to provide accurate, lengthy response with good lexical flow", "BENCHMARK", 5, "What is neural network and how its feeding mechanism works?"], 
+            ,["Your task is to provide accurate, lengthy response with good lexical flow", "BENCHMARK_TextEval", 5, "What is neural network and how its feeding mechanism works?"], 
             ["Act like you're in the technology field for 10+ year and give unbiased opinion", "UIDs", '975,517,906,743,869' , "What are the potential ethical concerns surrounding artificial intelligence and machine learning in healthcare?"]])
 
 interface.launch(enable_queue=True)