fix after changes in PMAT from other PR

gnosis · Nov 6, 2024 · fe51a97 · fe51a97
1 parent aaec2f2
commit fe51a97
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/prediction_market_agent/agents/microchain_agent/jobs_functions.py b/prediction_market_agent/agents/microchain_agent/jobs_functions.py
@@ -48,11 +48,13 @@ def description(self) -> str:
 
     @property
     def example_args(self) -> list[int | str]:
-        return ["0x1", 1, "I completed this job as described."]
+        return ["0x1", "GeneralAgent", 1, "I completed this job as described."]
 
-    def __call__(self, job_id: str, max_bond: float, result: str) -> str:
+    def __call__(
+        self, job_id: str, agent_name: str, max_bond: float, result: str
+    ) -> str:
         job = self.market_type.job_class.get_job(id=job_id)
-        processed = job.submit_job_result(max_bond, result)
+        processed = job.submit_job_result(agent_name, max_bond, result)
         return json.dumps(
             processed.model_dump(),
             indent=2,