robosystems-client 0.2.23__py3-none-any.whl → 0.2.25__py3-none-any.whl

robosystems_client/api/subscriptions/upgrade_subscription.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any, cast
+from urllib.parse import quote
 
 import httpx
 
@@ -20,7 +21,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "put",
-    "url": f"/v1/graphs/{graph_id}/subscriptions/upgrade",
+    "url": "/v1/graphs/{graph_id}/subscriptions/upgrade".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   _kwargs["json"] = body.to_dict()

robosystems_client/api/tables/list_tables.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any, cast
+from urllib.parse import quote
 
 import httpx
 
@@ -16,7 +17,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/graphs/{graph_id}/tables",
+    "url": "/v1/graphs/{graph_id}/tables".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   return _kwargs

robosystems_client/api/tables/query_tables.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any, cast
+from urllib.parse import quote
 
 import httpx
 
@@ -21,7 +22,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "post",
-    "url": f"/v1/graphs/{graph_id}/tables/query",
+    "url": "/v1/graphs/{graph_id}/tables/query".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   _kwargs["json"] = body.to_dict()

robosystems_client/api/usage/get_graph_metrics.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -16,7 +17,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/graphs/{graph_id}/analytics",
+    "url": "/v1/graphs/{graph_id}/analytics".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   return _kwargs

robosystems_client/api/usage/get_graph_usage_analytics.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -36,7 +37,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/graphs/{graph_id}/analytics/usage",
+    "url": "/v1/graphs/{graph_id}/analytics/usage".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
     "params": params,
   }
 

robosystems_client/api/user/revoke_user_api_key.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -16,7 +17,9 @@ def _get_kwargs(
 ) -> dict[str, Any]:
   _kwargs: dict[str, Any] = {
     "method": "delete",
-    "url": f"/v1/user/api-keys/{api_key_id}",
+    "url": "/v1/user/api-keys/{api_key_id}".format(
+      api_key_id=quote(str(api_key_id), safe=""),
+    ),
   }
 
   return _kwargs

robosystems_client/api/user/update_user_api_key.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -20,7 +21,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "put",
-    "url": f"/v1/user/api-keys/{api_key_id}",
+    "url": "/v1/user/api-keys/{api_key_id}".format(
+      api_key_id=quote(str(api_key_id), safe=""),
+    ),
   }
 
   _kwargs["json"] = body.to_dict()

robosystems_client/api/views/create_view.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -19,7 +20,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "post",
-    "url": f"/v1/graphs/{graph_id}/views",
+    "url": "/v1/graphs/{graph_id}/views".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   _kwargs["json"] = body.to_dict()

robosystems_client/api/views/save_view.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 
 import httpx
 
@@ -20,7 +21,9 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "post",
-    "url": f"/v1/graphs/{graph_id}/views/save",
+    "url": "/v1/graphs/{graph_id}/views/save".format(
+      graph_id=quote(str(graph_id), safe=""),
+    ),
   }
 
   _kwargs["json"] = body.to_dict()

robosystems_client/extensions/graph_client.py

@@ -1,13 +1,17 @@
 """Graph Management Client
 
 Provides high-level graph management operations with automatic operation monitoring.
+Supports both SSE (Server-Sent Events) for real-time updates and polling fallback.
 """
 
 from dataclasses import dataclass
 from typing import Dict, Any, Optional, Callable
 import time
+import json
 import logging
 
+import httpx
+
 logger = logging.getLogger(__name__)
 
 
@@ -62,10 +66,14 @@ class GraphClient:
     timeout: int = 60,
     poll_interval: int = 2,
     on_progress: Optional[Callable[[str], None]] = None,
+    use_sse: bool = True,
   ) -> str:
     """
     Create a graph and wait for completion.
 
+    Uses SSE (Server-Sent Events) for real-time progress updates with
+    automatic fallback to polling if SSE connection fails.
+
     Args:
         metadata: Graph metadata
         initial_entity: Optional initial entity data
@@ -73,8 +81,9 @@ class GraphClient:
             Only applies when initial_entity is provided. Set to False to create
             graph without populating entity data (useful for file-based ingestion).
         timeout: Maximum time to wait in seconds
-        poll_interval: Time between status checks in seconds
+        poll_interval: Time between status checks in seconds (for polling fallback)
         on_progress: Callback for progress updates
+        use_sse: Whether to try SSE first (default True). Falls back to polling on failure.
 
     Returns:
         graph_id when creation completes
@@ -84,7 +93,6 @@ class GraphClient:
     """
     from ..client import AuthenticatedClient
     from ..api.graphs.create_graph import sync_detailed as create_graph
-    from ..api.operations.get_operation_status import sync_detailed as get_status
     from ..models.create_graph_request import CreateGraphRequest
     from ..models.graph_metadata import GraphMetadata as APIGraphMetadata
 
@@ -151,13 +159,165 @@ class GraphClient:
         on_progress(f"Graph created: {graph_id}")
       return graph_id
 
-    # Otherwise, poll operation until complete
+    # Otherwise, wait for operation to complete
     if not operation_id:
       raise RuntimeError("No graph_id or operation_id in response")
 
     if on_progress:
       on_progress(f"Graph creation queued (operation: {operation_id})")
 
+    # Try SSE first, fall back to polling
+    if use_sse:
+      try:
+        return self._wait_with_sse(operation_id, timeout, on_progress)
+      except Exception as e:
+        logger.debug(f"SSE connection failed, falling back to polling: {e}")
+        if on_progress:
+          on_progress("SSE unavailable, using polling...")
+
+    # Fallback to polling
+    return self._wait_with_polling(
+      operation_id, timeout, poll_interval, on_progress, client
+    )
+
+  def _wait_with_sse(
+    self,
+    operation_id: str,
+    timeout: int,
+    on_progress: Optional[Callable[[str], None]],
+  ) -> str:
+    """
+    Wait for operation completion using SSE stream.
+
+    Args:
+        operation_id: Operation ID to monitor
+        timeout: Maximum time to wait in seconds
+        on_progress: Callback for progress updates
+
+    Returns:
+        graph_id when operation completes
+
+    Raises:
+        RuntimeError: If operation fails
+        TimeoutError: If operation times out
+    """
+    stream_url = f"{self.base_url}/v1/operations/{operation_id}/stream"
+    headers = {"X-API-Key": self.token, "Accept": "text/event-stream"}
+
+    with httpx.Client(timeout=httpx.Timeout(timeout + 5.0)) as http_client:
+      with http_client.stream("GET", stream_url, headers=headers) as response:
+        if response.status_code != 200:
+          raise RuntimeError(f"SSE connection failed: {response.status_code}")
+
+        start_time = time.time()
+        event_type = None
+        event_data = ""
+
+        for line in response.iter_lines():
+          # Check timeout
+          if time.time() - start_time > timeout:
+            raise TimeoutError(f"Graph creation timed out after {timeout}s")
+
+          line = line.strip()
+
+          if not line:
+            # Empty line = end of event, process it
+            if event_type and event_data:
+              result = self._process_sse_event(event_type, event_data, on_progress)
+              if result is not None:
+                return result
+            event_type = None
+            event_data = ""
+            continue
+
+          if line.startswith("event:"):
+            event_type = line[6:].strip()
+          elif line.startswith("data:"):
+            event_data = line[5:].strip()
+          # Ignore other lines (comments, id, retry, etc.)
+
+    raise TimeoutError(f"SSE stream ended without completion after {timeout}s")
+
+  def _process_sse_event(
+    self,
+    event_type: str,
+    event_data: str,
+    on_progress: Optional[Callable[[str], None]],
+  ) -> Optional[str]:
+    """
+    Process a single SSE event.
+
+    Returns:
+        graph_id if operation completed, None to continue waiting
+
+    Raises:
+        RuntimeError: If operation failed
+    """
+    try:
+      data = json.loads(event_data)
+    except json.JSONDecodeError:
+      logger.debug(f"Failed to parse SSE event data: {event_data}")
+      return None
+
+    if event_type == "operation_progress":
+      if on_progress:
+        message = data.get("message", "Processing...")
+        percent = data.get("progress_percent")
+        if percent is not None:
+          on_progress(f"{message} ({percent:.0f}%)")
+        else:
+          on_progress(message)
+      return None
+
+    elif event_type == "operation_completed":
+      result = data.get("result", {})
+      graph_id = result.get("graph_id") if isinstance(result, dict) else None
+
+      if graph_id:
+        if on_progress:
+          on_progress(f"Graph created: {graph_id}")
+        return graph_id
+      else:
+        raise RuntimeError("Operation completed but no graph_id in result")
+
+    elif event_type == "operation_error":
+      error = data.get("error", "Unknown error")
+      raise RuntimeError(f"Graph creation failed: {error}")
+
+    elif event_type == "operation_cancelled":
+      reason = data.get("reason", "Operation was cancelled")
+      raise RuntimeError(f"Graph creation cancelled: {reason}")
+
+    # Ignore other event types (keepalive, etc.)
+    return None
+
+  def _wait_with_polling(
+    self,
+    operation_id: str,
+    timeout: int,
+    poll_interval: int,
+    on_progress: Optional[Callable[[str], None]],
+    client: Any,
+  ) -> str:
+    """
+    Wait for operation completion using polling.
+
+    Args:
+        operation_id: Operation ID to monitor
+        timeout: Maximum time to wait in seconds
+        poll_interval: Time between status checks
+        on_progress: Callback for progress updates
+        client: Authenticated HTTP client
+
+    Returns:
+        graph_id when operation completes
+
+    Raises:
+        RuntimeError: If operation fails
+        TimeoutError: If operation times out
+    """
+    from ..api.operations.get_operation_status import sync_detailed as get_status
+
     max_attempts = timeout // poll_interval
     for attempt in range(max_attempts):
       time.sleep(poll_interval)

robosystems_client/extensions/query_client.py

@@ -266,6 +266,23 @@ class QueryClient:
       else:
         raise Exception(f"Query execution failed: {error_msg}")
 
+    # Handle error responses (4xx/5xx) where parsed is None
+    if hasattr(response, "status_code") and response.status_code >= 400:
+      import json as _json
+
+      detail = f"HTTP {response.status_code}"
+      try:
+        body = (
+          response.content.decode("utf-8")
+          if isinstance(response.content, bytes)
+          else str(response.content)
+        )
+        error_data = _json.loads(body)
+        detail = error_data.get("detail", error_data.get("message", body))
+      except Exception:
+        pass
+      raise Exception(f"Query failed ({response.status_code}): {detail}")
+
     # Unexpected response format
     raise Exception("Unexpected response format from query endpoint")
 

robosystems_client/models/token_pricing.py

@@ -14,12 +14,12 @@ class TokenPricing:
   """AI token pricing for a specific model.
 
   Attributes:
-      input_per_1k_tokens (int): Credits per 1K input tokens
-      output_per_1k_tokens (int): Credits per 1K output tokens
+      input_per_1k_tokens (float): Credits per 1K input tokens
+      output_per_1k_tokens (float): Credits per 1K output tokens
   """
 
-  input_per_1k_tokens: int
-  output_per_1k_tokens: int
+  input_per_1k_tokens: float
+  output_per_1k_tokens: float
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:

{robosystems_client-0.2.23.dist-info → robosystems_client-0.2.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.23
+Version: 0.2.25
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT
@@ -33,8 +33,8 @@ Requires-Dist: pydantic>=2.11.7
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: typing-extensions>=4.0.0
 Provides-Extra: all
-Requires-Dist: pandas>=1.5.0; extra == 'all'
-Requires-Dist: pyarrow>=10.0.0; extra == 'all'
+Requires-Dist: pandas>=2.1.0; extra == 'all'
+Requires-Dist: pyarrow>=17.0.0; extra == 'all'
 Provides-Extra: dev
 Requires-Dist: basedpyright>=1.21.0; extra == 'dev'
 Requires-Dist: build>=1.0.0; extra == 'dev'
@@ -45,11 +45,11 @@ Requires-Dist: pytest>=8.3.5; extra == 'dev'
 Requires-Dist: ruff>=0.12; extra == 'dev'
 Requires-Dist: twine>=5.0.0; extra == 'dev'
 Provides-Extra: extensions
-Requires-Dist: pandas>=1.5.0; extra == 'extensions'
-Requires-Dist: pyarrow>=10.0.0; extra == 'extensions'
+Requires-Dist: pandas>=2.1.0; extra == 'extensions'
+Requires-Dist: pyarrow>=17.0.0; extra == 'extensions'
 Provides-Extra: tables
-Requires-Dist: pandas>=1.5.0; extra == 'tables'
-Requires-Dist: pyarrow>=10.0.0; extra == 'tables'
+Requires-Dist: pandas>=2.1.0; extra == 'tables'
+Requires-Dist: pyarrow>=17.0.0; extra == 'tables'
 Description-Content-Type: text/markdown
 
 # RoboSystems Python Client
@@ -81,18 +81,19 @@ pip install robosystems-client[tables]
 pip install robosystems-client[all]
 ```
 
-## API Reference
+## Resources
 
-- [API reference](https://api.robosystems.ai)
-- [API documentation](https://api.robosystems.ai/docs)
-- [OpenAPI specification](https://api.robosystems.ai/openapi.json)
+- [RoboSystems Platform](https://robosystems.ai)
+- [GitHub Repository](https://github.com/RoboFinSystems/robosystems)
+- [API Documentation](https://api.robosystems.ai/docs)
+- [OpenAPI Specification](https://api.robosystems.ai/openapi.json)
 
 ## Support
 
 - [Issues](https://github.com/RoboFinSystems/robosystems-python-client/issues)
-- [Discussions](https://github.com/RoboFinSystems/robosystems/discussions)
-- [Projects](https://github.com/RoboFinSystems/robosystems/projects)
 - [Wiki](https://github.com/RoboFinSystems/robosystems/wiki)
+- [Projects](https://github.com/orgs/RoboFinSystems/projects)
+- [Discussions](https://github.com/orgs/RoboFinSystems/discussions)
 
 ## License