specify num_results

currently the server has a bug where this must be specified.
fixed in https://github.com/ioniccommerce/server/pull/179

Author: Gregory Kohler

ionic_langchain/tool.py

@@ -12,9 +12,11 @@
 
 class Ionic:
     _sdk: IonicSDK
+    _results_per_query: int
 
-    def __init__(self):
+    def __init__(self, results_per_query: int = 5):
         self._sdk = IonicSDK()
+        self._results_per_query = results_per_query or 5
 
     def query(self, queries: str) -> dict[str, Any]:
         """
@@ -23,9 +25,13 @@ def query(self, queries: str) -> dict[str, Any]:
         """
         request = QueryAPIRequest(
             queries=[
-                Query(query=query)
+                Query(
+                    query=query,
+                    num_results=self._results_per_query,
+                )
                 for query in queries.split(", ")
             ],
+
         )
         response: QueryResponse = self._sdk.query(
             request=request,