robosystems-client 0.2.20__py3-none-any.whl → 0.2.22__py3-none-any.whl

robosystems_client/extensions/extensions.py

@@ -25,6 +25,7 @@ class RoboSystemsExtensionConfig:
   max_retries: int = 5
   retry_delay: int = 1000
   timeout: int = 30
+  s3_endpoint_url: Optional[str] = None  # Override S3 endpoint (e.g., for LocalStack)
 
 
 class RoboSystemsExtensions:
@@ -41,6 +42,7 @@ class RoboSystemsExtensions:
       "max_retries": config.max_retries,
       "retry_delay": config.retry_delay,
       "timeout": config.timeout,
+      "s3_endpoint_url": config.s3_endpoint_url,
     }
 
     # Extract token from headers if it was set by auth classes

robosystems_client/extensions/file_client.py

@@ -37,7 +37,6 @@ class FileUploadOptions:
   """Options for file upload operations"""
 
   on_progress: Optional[Callable[[str], None]] = None
-  fix_localstack_url: bool = True
   ingest_to_graph: bool = False
 
 
@@ -78,6 +77,9 @@ class FileClient:
     self.base_url = config["base_url"]
     self.headers = config.get("headers", {})
     self.token = config.get("token")
+    self.s3_endpoint_url = config.get(
+      "s3_endpoint_url"
+    )  # Optional S3 endpoint override
     self._http_client = httpx.Client(timeout=120.0)
 
   def upload(
@@ -171,9 +173,23 @@ class FileClient:
       upload_url = upload_data.upload_url
       file_id = upload_data.file_id
 
-      # Fix LocalStack URL if needed
-      if options.fix_localstack_url and "localstack:4566" in upload_url:
-        upload_url = upload_url.replace("localstack:4566", "localhost:4566")
+      # Override S3 endpoint if configured (e.g., for LocalStack)
+      if self.s3_endpoint_url:
+        from urllib.parse import urlparse, urlunparse
+
+        parsed_url = urlparse(upload_url)
+        override_parsed = urlparse(self.s3_endpoint_url)
+        # Replace scheme, host, and port with the override endpoint
+        upload_url = urlunparse(
+          (
+            override_parsed.scheme or parsed_url.scheme,
+            override_parsed.netloc,
+            parsed_url.path,
+            parsed_url.params,
+            parsed_url.query,
+            parsed_url.fragment,
+          )
+        )
 
       # Step 2: Upload file to S3
       if options.on_progress:

{robosystems_client-0.2.20.dist-info → robosystems_client-0.2.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.20
+Version: 0.2.22
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT
@@ -19,6 +19,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Database :: Front-Ends
 Classifier: Topic :: Internet :: WWW/HTTP
 Classifier: Topic :: Office/Business :: Financial

{robosystems_client-0.2.20.dist-info → robosystems_client-0.2.22.dist-info}/RECORD

@@ -147,8 +147,8 @@ robosystems_client/extensions/agent_client.py,sha256=Db2C4hrakVsf6ScnBcNk6rte3Kw
 robosystems_client/extensions/auth_integration.py,sha256=ABOJ8aVjfHehNGNzim1iR9-Cdh7Mr22ce-WgWWeqJt0,6535
 robosystems_client/extensions/dataframe_utils.py,sha256=gK1bgkVqBF0TvWVdGQvqWrt-ur_Rw11j8uNtMoulLWE,12312
 robosystems_client/extensions/element_mapping_client.py,sha256=yuh0QPQBPM33E7r6QWWDiKm3T4TfCdbn2kvO3Jlw4Cs,18516
-robosystems_client/extensions/extensions.py,sha256=7vsD3QeIKbwhC1UqNskFjsfKkg_ZO3PPDnc6TxV3PoA,6722
-robosystems_client/extensions/file_client.py,sha256=WNttw8BtdpLsb3N2fkvim-a8eDSshi8LM3nMtfEmLt0,11843
+robosystems_client/extensions/extensions.py,sha256=IqTuVJNt6vvO5LhnH-9SexCF5k9Abqig1sBdY4SdD1I,6858
+robosystems_client/extensions/file_client.py,sha256=bknqmbYrIqxc2PnWskeDvcWYP2bemBQaxY_4A6zr5LY,12326
 robosystems_client/extensions/graph_client.py,sha256=OBi0xj0SLIRKLeSu_DiGt2ZakCmhggvNrMP3jdRfEgQ,10326
 robosystems_client/extensions/materialization_client.py,sha256=Pvq4Gz5fS4B2eTxFrpCGTj7-diOZwOtWnNo7wKT9mVE,6811
 robosystems_client/extensions/operation_client.py,sha256=B1qju-wWQrnrnVJixKGgsA_KEInviwJwdlJxzm_i7P0,13359
@@ -414,7 +414,7 @@ robosystems_client/models/view_axis_config_member_labels_type_0.py,sha256=kkzpHx
 robosystems_client/models/view_config.py,sha256=HQnqYjLMXRhjZLOc5ypwILriMFKuvPzu0hPQi2vyNoM,3795
 robosystems_client/models/view_source.py,sha256=h66cASj-P_-qOptKv26uAIe9PtIewU2nTs42Ls-lFFk,4098
 robosystems_client/models/view_source_type.py,sha256=KpgczHUeOinV01jdLvytZ2URKwcsRcp1doPx2D3USyw,169
-robosystems_client-0.2.20.dist-info/METADATA,sha256=YhEF7b7jE9rE1DWePu4P5JVfW8qOFDNx_9xic0C8_ts,3899
-robosystems_client-0.2.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-robosystems_client-0.2.20.dist-info/licenses/LICENSE,sha256=LjFqQPU4eQh7jAQ04SmE9eC0j74HCdXvzbo0hjW4mWo,1063
-robosystems_client-0.2.20.dist-info/RECORD,,
+robosystems_client-0.2.22.dist-info/METADATA,sha256=yEG5FHqaUwnCVaVxGPrcXOAMY3DMYelvRAvJK3p5ikE,3950
+robosystems_client-0.2.22.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+robosystems_client-0.2.22.dist-info/licenses/LICENSE,sha256=LjFqQPU4eQh7jAQ04SmE9eC0j74HCdXvzbo0hjW4mWo,1063
+robosystems_client-0.2.22.dist-info/RECORD,,

{robosystems_client-0.2.20.dist-info → robosystems_client-0.2.22.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any