clean up

Author: iamNarcisse

jigsawstack/search.py

@@ -81,8 +81,13 @@ def search(self, params: SearchParams) -> SearchResponse:
 
         body = {
             "byo_urls": params.get("byo_urls", []),
+            "query": query,
+            "ai_overview": ai_overview,
+            "safe_search": safe_search,
+            "spell_check": spell_check,
         }
-        path = f"/web/search?query={query}&ai_overview={ai_overview}&safe_search={safe_search}&spell_check={spell_check}"
+
+        path = f"/web/search"
         resp = Request(
             config=self.config,
             path=path,
@@ -122,10 +127,22 @@ def __init__(
 
     async def search(self, params: SearchParams) -> SearchResponse:
         path = f"/web/search"
+        query = params["query"]
+        ai_overview = params.get("ai_overview", "True")
+        safe_search = params.get("safe_search", "moderate")
+        spell_check = params.get("spell_check", "True")
+
+        body = {
+            "byo_urls": params.get("byo_urls", []),
+            "query": query,
+            "ai_overview": ai_overview,
+            "safe_search": safe_search,
+            "spell_check": spell_check,
+        }
         resp = await AsyncRequest(
             config=self.config,
             path=path,
-            params=cast(Dict[Any, Any], params),
+            params=cast(Dict[Any, Any], body),
             verb="POST",
         ).perform_with_content()
         return resp