robosystems-client 0.2.17__tar.gz → 0.2.18__tar.gz

{robosystems_client-0.2.17 → robosystems_client-0.2.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.17
+Version: 0.2.18
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT

{robosystems_client-0.2.17 → robosystems_client-0.2.18}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.17"
+version = "0.2.18"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"

{robosystems_client-0.2.17 → robosystems_client-0.2.18}/robosystems_client/extensions/file_client.py

@@ -134,9 +134,22 @@ class FileClient:
         table_name=table_name,
       )
 
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
         "body": upload_request,
       }
 
@@ -195,7 +208,7 @@ class FileClient:
       update_kwargs = {
         "graph_id": graph_id,
         "file_id": file_id,
-        "client": self.config.get("client"),
+        "client": client,
         "body": status_update,
       }
 
@@ -261,9 +274,22 @@ class FileClient:
         List of FileInfo objects
     """
     try:
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
       }
 
       if table_name:
@@ -311,10 +337,23 @@ class FileClient:
         FileInfo with multi-layer status tracking, or None if not found
     """
     try:
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
         "file_id": file_id,
-        "client": self.config.get("client"),
+        "client": client,
       }
 
       response = get_file(**kwargs)
@@ -355,10 +394,23 @@ class FileClient:
         True if deletion succeeded, False otherwise
     """
     try:
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
         "file_id": file_id,
-        "client": self.config.get("client"),
+        "client": client,
         "cascade": cascade,
       }
 

{robosystems_client-0.2.17 → robosystems_client-0.2.18}/robosystems_client/extensions/materialization_client.py

@@ -104,9 +104,22 @@ class MaterializationClient:
         force=options.force,
       )
 
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
         "body": request,
       }
 
@@ -182,9 +195,22 @@ class MaterializationClient:
         MaterializationStatus with staleness and timing information
     """
     try:
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
       }
 
       response = get_materialization_status(**kwargs)

{robosystems_client-0.2.17 → robosystems_client-0.2.18}/robosystems_client/extensions/table_client.py

@@ -62,9 +62,22 @@ class TableClient:
         List of TableInfo objects with metadata
     """
     try:
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
       }
 
       response = list_tables(**kwargs)
@@ -120,9 +133,22 @@ class TableClient:
 
       request = TableQueryRequest(sql=final_query)
 
+      from ..client import AuthenticatedClient
+
+      if not self.token:
+        raise Exception("No API key provided. Set X-API-Key in headers.")
+
+      client = AuthenticatedClient(
+        base_url=self.base_url,
+        token=self.token,
+        prefix="",
+        auth_header_name="X-API-Key",
+        headers=self.headers,
+      )
+
       kwargs = {
         "graph_id": graph_id,
-        "client": self.config.get("client"),
+        "client": client,
         "body": request,
       }