[ION-281] use ionic SDK in langchain tool

.gitignore

@@ -160,4 +160,4 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+.idea/

ionic_langchain/tool.py

@@ -1,60 +1,54 @@
-# Temp inlining of "SDK"
-import requests
+import dataclasses
+from typing import Any
 
-from langchain.tools import AIPluginTool, Tool
+from ionic import Ionic as IonicSDK
+from ionic.models.components import QueryAPIRequest, Query
+from ionic.models.operations import QuerySecurity, QueryResponse
+from langchain.tools import Tool
 
 from ionic_langchain.prompt import TOOL_PROMPT
 
 
 class Ionic:
-    PROD_SERVER = "https://api.ionicapi.com"
-
-    def __init__(self, base_url=None):
-        self.base_url = base_url or self.PROD_SERVER
-
-    def query(self, queries: str):
-        url = f"{self.base_url}/query"
-
-        query_list = [{"query": query} for query in queries.split(", ")]
-
-        payload = {"queries": query_list}
-        result = requests.post(url, json=payload)
+    _sdk: IonicSDK
+
+    def __init__(self):
+        self._sdk = IonicSDK()
+
+    def query(self, queries: str) -> dict[str, Any]:
+        """
+        FIXME: handle non-200 responses
+        TODO: better typing in response
+        """
+        request = QueryAPIRequest(
+            queries=[
+                Query(query=query)
+                for query in queries.split(", ")
+            ],
+        )
+        response: QueryResponse = self._sdk.query(
+            request=request,
+            security=QuerySecurity(),
+        )
 
-        return result.json()
+        return dataclasses.asdict(response)
 
 
 # TODO StructuredTool or BaseTool
-    # https://github.com/langchain-ai/langchain/issues/4197
-    # https://python.langchain.com/docs/modules/agents/tools/multi_input_tool
+# https://github.com/langchain-ai/langchain/issues/4197
+# https://python.langchain.com/docs/modules/agents/tools/multi_input_tool
 
 
-class IonicTool():
-    def __init__(self, base_url=None):
-        self.ionic = Ionic()
+class IonicTool:
+    _ionic: Ionic
 
-        if base_url:
-            self.ionic.base_url = base_url
+    def __init__(self):
+        self._ionic = Ionic()
 
-    def tool(self):
+    def tool(self) -> Tool:
         return Tool(
-            func=self.ionic.query,
+            func=self._ionic.query,
             name="Ionic Shopping",
             description=TOOL_PROMPT,
             verbose=True
         )
-
-
-
-
-class IonicPluginTool(): 
-    def __init__(self, base_url="https://api.ionicapi.com"):
-        self.base_url = base_url
-
-    def tool(self):
-        # tool = AIPluginTool.from_plugin_url("https://www.klarna.com/.well-known/ai-plugin.json")
-        tool = AIPluginTool.from_plugin_url(f'{self.base_url}/.well-known/ai-plugin.json')
-        print(tool)
-
-        return tool
-
-