dominusnode-langchain 1.2.0__tar.gz → 1.2.2__tar.gz

{dominusnode_langchain-1.2.0 → dominusnode_langchain-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dominusnode-langchain
-Version: 1.2.0
+Version: 1.2.2
 Summary: LangChain tools for Dominus Node rotating proxy service
 License-Expression: MIT
 License-File: LICENSE

{dominusnode_langchain-1.2.0 → dominusnode_langchain-1.2.2}/dominusnode_langchain/toolkit.py

@@ -1,8 +1,8 @@
 """Dominus Node LangChain Toolkit -- bundles all 53 Dominus Node tools for agent use.
 
-The :class:`DominusNodeToolkit` creates a :class:`DominusNodeClient` internally
-and injects it into each tool so that a single authenticated session is shared
-across all tools in a LangChain agent.
+The :class:`DominusNodeToolkit` injects API credentials into each tool so that
+a single authenticated configuration is shared across all tools in a LangChain
+agent.  All tools use direct REST calls (no SDK dependency).
 
 Example::
 
@@ -22,8 +22,6 @@ from typing import List, Optional
 
 from langchain_core.tools import BaseTool, BaseToolkit
 
-from dominusnode import AsyncDominusNodeClient, DominusNodeClient
-
 from .tools import (
     # Proxy (3)
     DominusNodeProxiedFetchTool,
@@ -93,8 +91,7 @@ from .tools import (
 class DominusNodeToolkit(BaseToolkit):
     """LangChain toolkit that provides all 53 Dominus Node proxy service tools.
 
-    Creates and manages a :class:`DominusNodeClient` (sync) and optionally
-    an :class:`AsyncDominusNodeClient` (async) for use by LangChain agents.
+    Injects API key and base URL into each tool for direct REST API calls.
 
     Args:
         api_key: Dominus Node API key (``dn_live_...``).  Falls back to the
@@ -110,9 +107,6 @@ class DominusNodeToolkit(BaseToolkit):
         ValueError: If no API key is provided or found in environment.
     """
 
-    # Stored so tools can reference them; not exposed to LLM
-    _sync_client: Optional[DominusNodeClient] = None
-    _async_client: Optional[AsyncDominusNodeClient] = None
     _tools: Optional[List[BaseTool]] = None
 
     model_config = {"arbitrary_types_allowed": True}
@@ -143,72 +137,56 @@ class DominusNodeToolkit(BaseToolkit):
             "DOMINUSNODE_PROXY_HOST", "proxy.dominusnode.com"
         )
 
-        # Store credentials for direct REST API tools
         self._api_key = resolved_key
         self._base_url = resolved_base
         self._agent_secret = agent_secret or os.environ.get("DOMINUSNODE_AGENT_SECRET")
 
-        # Create sync client
-        self._sync_client = DominusNodeClient(
-            base_url=resolved_base,
-            api_key=resolved_key,
-            proxy_host=resolved_proxy_host,
-            http_proxy_port=http_proxy_port,
-            socks5_proxy_port=socks5_proxy_port,
-            agent_secret=self._agent_secret,
-        )
-
-        # Create async client (deferred connect -- will connect in __aenter__)
-        self._async_client = AsyncDominusNodeClient(
-            base_url=resolved_base,
-            api_key=resolved_key,
-            proxy_host=resolved_proxy_host,
-            http_proxy_port=http_proxy_port,
-            socks5_proxy_port=socks5_proxy_port,
-            agent_secret=self._agent_secret,
-        )
+        # Store proxy host/port for env-based tools
+        os.environ.setdefault("DOMINUSNODE_PROXY_HOST", resolved_proxy_host)
+        os.environ.setdefault("DOMINUSNODE_PROXY_PORT", str(http_proxy_port))
 
         # Build tools once
         self._tools = self._build_tools()
 
     def _build_tools(self) -> List[BaseTool]:
-        """Construct all 53 tool instances with injected clients."""
-        sc = self._sync_client
-        ac = self._async_client
+        """Construct all 53 tool instances with injected credentials."""
         _ak = self._api_key
         _bu = self._base_url
         _as = self._agent_secret
-
-        # Helper for SDK-client-based tools
-        def _sdk(cls: type) -> BaseTool:
-            return cls(sync_client=sc, async_client=ac)
+        _ph = os.environ.get("DOMINUSNODE_PROXY_HOST", "localhost")
+        _pp = int(os.environ.get("DOMINUSNODE_PROXY_PORT", "8080"))
 
         # Helper for REST-API-based tools (authenticated)
         def _rest(cls: type) -> BaseTool:
             return cls(api_key=_ak, base_url=_bu, agent_secret=_as)
 
+        # Helper for proxy tools that need proxy host/port
+        def _proxy(cls: type) -> BaseTool:
+            return cls(api_key=_ak, base_url=_bu, agent_secret=_as,
+                       proxy_host=_ph, proxy_port=_pp)
+
         # Helper for unauthenticated tools
         def _unauth(cls: type) -> BaseTool:
             return cls(base_url=_bu, agent_secret=_as)
 
         return [
             # Proxy (3)
-            _sdk(DominusNodeProxiedFetchTool),
-            _sdk(DominusNodeProxyConfigTool),
-            _sdk(DominusNodeProxyStatusTool),
+            _proxy(DominusNodeProxiedFetchTool),
+            _rest(DominusNodeProxyConfigTool),
+            _rest(DominusNodeProxyStatusTool),
             # Sessions (1)
-            _sdk(DominusNodeListSessionsTool),
+            _rest(DominusNodeListSessionsTool),
             # Wallet (8)
-            _sdk(DominusNodeBalanceTool),
+            _rest(DominusNodeBalanceTool),
             _rest(DominusNodeGetTransactionsTool),
             _rest(DominusNodeGetForecastTool),
-            _sdk(DominusNodeTopupPaypalTool),
-            _sdk(DominusNodeTopupStripeTool),
-            _sdk(DominusNodeTopupCryptoTool),
+            _rest(DominusNodeTopupPaypalTool),
+            _rest(DominusNodeTopupStripeTool),
+            _rest(DominusNodeTopupCryptoTool),
             _rest(DominusNodeCheckPaymentTool),
-            _sdk(DominusNodeX402InfoTool),
+            _rest(DominusNodeX402InfoTool),
             # Usage (3)
-            _sdk(DominusNodeUsageTool),
+            _rest(DominusNodeUsageTool),
             _rest(DominusNodeDailyUsageTool),
             _rest(DominusNodeTopHostsTool),
             # Account (6) -- register, login, verify_email are unauthenticated
@@ -263,11 +241,5 @@ class DominusNodeToolkit(BaseToolkit):
         return list(self._tools)
 
     def close(self) -> None:
-        """Close the underlying SDK clients and release resources."""
-        if self._sync_client is not None:
-            self._sync_client.close()
-            self._sync_client = None
-        if self._async_client is not None:
-            self._async_client.close()
-            self._async_client = None
+        """Release resources."""
         self._tools = None