Update src/napari_chatgpt/omega/tools/async_base_tool.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
royerlab · Aug 31, 2024 · 6c8bced · 6c8bced
1 parent 6653244
commit 6c8bced
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/napari_chatgpt/omega/tools/async_base_tool.py b/src/napari_chatgpt/omega/tools/async_base_tool.py
@@ -17,10 +17,7 @@ class AsyncBaseTool(BaseTool):
     def normalise_to_string(self, kwargs):
 
         # extract the value for args key in kwargs:
-        if isinstance(kwargs, dict):
-            query = kwargs.get('args', '')
-        else:
-            query = kwargs
+        query = kwargs.get('args', '') if isinstance(kwargs, dict) else kwargs
 
         # If query is a singleton list, extract the value:
         if isinstance(query, list) and len(query) == 1: