Update platform/reworkd_platform/web/api/agent/tools/sidsearch.py

Co-authored-by: Adam Watkins <[email protected]>
reworkd · Oct 28, 2023 · 11e5253 · 11e5253
1 parent 65fbf51
commit 11e5253
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/platform/reworkd_platform/web/api/agent/tools/sidsearch.py b/platform/reworkd_platform/web/api/agent/tools/sidsearch.py
@@ -137,13 +137,7 @@ async def call(
         *args: Any,
         **kwargs: Any,
     ) -> FastAPIStreamingResponse:
-        res = await self._run_sid(
-            goal, task, input_str, user, oauth_crud
-        )
-        if res is None:
-            # fall back to search if no results are found
-            return await Search(self.model, self.language).call(
-                goal, task, input_str, user, oauth_crud
-            )
-        else:
-            return res
+         # fall back to search if no results are found
+        return await self._run_sid(goal, task, input_str, user, oauth_crud) or await Search(self.model, self.language).call(
+        goal, task, input_str, user, oauth_crud
+    )