Merge pull request #40 from JigsawStack/fix/search-suggestions

Fix/search suggestions

Author: yoeven

jigsawstack/search.py

@@ -20,7 +20,7 @@ class SearchResponse(TypedDict):
     spell_fixed: str
 
 
-class SearchSuggestionResponse(TypedDict):
+class SearchSuggestionsResponse(TypedDict):
     success: bool
     """
     Indicates whether the translation was successful.
@@ -29,7 +29,7 @@ class SearchSuggestionResponse(TypedDict):
     suggestions: List[str]
 
 
-class SearchSuggestionParams(TypedDict):
+class SearchSuggestionsParams(TypedDict):
     query: str
     """
     The search value. The maximum query character length is 200.
@@ -97,7 +97,7 @@ def search(self, params: SearchParams) -> SearchResponse:
 
         return resp
 
-    def suggestion(self, params: SearchSuggestionParams) -> SearchSuggestionResponse:
+    def suggestions(self, params: SearchSuggestionsParams) -> SearchSuggestionsResponse:
         query = params["query"]
         path = f"/web/search/suggest?query={query}"
         resp = Request(
@@ -147,9 +147,9 @@ async def search(self, params: SearchParams) -> SearchResponse:
         ).perform_with_content()
         return resp
 
-    async def suggestion(
-        self, params: SearchSuggestionParams
-    ) -> SearchSuggestionResponse:
+    async def suggestions(
+        self, params: SearchSuggestionsParams
+    ) -> SearchSuggestionsResponse:
         query = params["query"]
         path = f"/web/search/suggest?query={query}"
         resp = await AsyncRequest(

jigsawstack/web.py

@@ -6,8 +6,8 @@
 from .search import (
     Search,
     SearchParams,
-    SearchSuggestionParams,
-    SearchSuggestionResponse,
+    SearchSuggestionsParams,
+    SearchSuggestionsResponse,
     SearchResponse,
     AsyncSearch,
 )
@@ -198,15 +198,15 @@ def search(self, params: SearchParams) -> SearchResponse:
         )
         return s.search(params)
 
-    def search_suggestion(
-        self, params: SearchSuggestionParams
-    ) -> SearchSuggestionResponse:
+    def search_suggestions(
+        self, params: SearchSuggestionsParams
+    ) -> SearchSuggestionsResponse:
         s = Search(
             self.api_key,
             self.api_url,
             disable_request_logging=self.config.get("disable_request_logging"),
         )
-        return s.suggestion(params)
+        return s.suggestions(params)
 
 
 class AsyncWeb(ClientConfig):
@@ -276,12 +276,12 @@ async def search(self, params: SearchParams) -> SearchResponse:
         )
         return await s.search(params)
 
-    async def search_suggestion(
-        self, params: SearchSuggestionParams
-    ) -> SearchSuggestionResponse:
+    async def search_suggestions(
+        self, params: SearchSuggestionsParams
+    ) -> SearchSuggestionsResponse:
         s = AsyncSearch(
             self.api_key,
             self.api_url,
             disable_request_logging=self.config.get("disable_request_logging"),
         )
-        return await s.suggestion(params)
+        return await s.suggestions(params)

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="jigsawstack",
-    version="0.1.30",
+    version="0.1.31",
     description="JigsawStack - The AI SDK for Python",
     long_description=open("README.md", encoding="utf8").read(),
     long_description_content_type="text/markdown",