Add Task to headers

Author: richwardle

neurons/validator.py

@@ -140,9 +140,9 @@ async def collect_responses(self, task: BaseTextTask) -> DendriteResponseEvent |
             return
 
 
-        body = {"seed": task.seed, "model": task.llm_model_id, "messages": [{'role': 'user', 'content': task.query},]}
+        body = {"seed": task.seed, "task": task.__class__.__name__, "model": task.llm_model_id, "messages": [{'role': 'user', 'content': task.query},]}
         body_bytes = json.dumps(body).encode("utf-8")
-        stream_results = await query_miners(task.__class__.__name__, uids, body_bytes)
+        stream_results = await query_miners(uids, body_bytes)
 
         log_stream_results(stream_results)
 

prompting/base/epistula.py

@@ -85,14 +85,14 @@ async def add_headers(request: httpx.Request):
 
     return add_headers
 
-async def query_miners(task, uids, body):
+async def query_miners(uids, body):
     try:
         tasks = []
         for uid in uids:
             tasks.append(
                 asyncio.create_task(
                     handle_inference(
-                        settings.METAGRAPH, settings.WALLET, task, body, uid,
+                        settings.METAGRAPH, settings.WALLET, body, uid,
                     )
                 )
             )
@@ -149,7 +149,6 @@ async def handle_availability(
 async def handle_inference(
     metagraph: "bt.NonTorchMetagraph",
     wallet: "bt.wallet",
-    task: str,
     body: Dict[str, Any],
     uid: int,
 ) -> SynapseStreamResult: