From 303fa9ed617ff843d663719dd76d3a08c00cf724 Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Mon, 2 Sep 2024 16:12:46 +0300
Subject: Rename task_id to task_object.

---
 gn3/llms/client.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'gn3')

diff --git a/gn3/llms/client.py b/gn3/llms/client.py
index ad6c400..c225acc 100644
--- a/gn3/llms/client.py
+++ b/gn3/llms/client.py
@@ -90,9 +90,9 @@ class GeneNetworkQAClient(Session):
         res = self.custom_request('POST', f"{self.base_url}{ex_url}", *args, **kwargs)
         return res, json.loads(res.text)
 
-    def get_answer(self, taskid, *args, **kwargs):
+    def get_answer(self, task_obj, *args, **kwargs):
         """Fahamu get answer interface"""
-        query = f"{self.answer_url}?task_id={taskid['task_id']}"
+        query = f"{self.answer_url}?task_id={task_obj['task_id']}"
         res = self.custom_request('GET', query, *args, **kwargs)
         return res, 1
 
-- 
cgit v1.2.3