robosystems-client 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py

@@ -7,35 +7,19 @@ from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.http_validation_error import HTTPValidationError
 from ...models.tier_upgrade_request import TierUpgradeRequest
-from ...types import UNSET, Response, Unset
+from ...types import Response
 
 
 def _get_kwargs(
   subscription_id: str,
   *,
   body: TierUpgradeRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
-  if not isinstance(authorization, Unset):
-    headers["authorization"] = authorization
-
-  params: dict[str, Any] = {}
-
-  json_token: Union[None, Unset, str]
-  if isinstance(token, Unset):
-    json_token = UNSET
-  else:
-    json_token = token
-  params["token"] = json_token
-
-  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 
   _kwargs: dict[str, Any] = {
     "method": "put",
     "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}/upgrade",
-    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -96,8 +80,6 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -105,8 +87,6 @@ def sync_detailed(
 
   Args:
       subscription_id (str):
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -120,8 +100,6 @@ def sync_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
-    token=token,
-    authorization=authorization,
   )
 
   response = client.get_httpx_client().request(
@@ -136,8 +114,6 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -145,8 +121,6 @@ def sync(
 
   Args:
       subscription_id (str):
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -161,8 +135,6 @@ def sync(
     subscription_id=subscription_id,
     client=client,
     body=body,
-    token=token,
-    authorization=authorization,
   ).parsed
 
 
@@ -171,8 +143,6 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -180,8 +150,6 @@ async def asyncio_detailed(
 
   Args:
       subscription_id (str):
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -195,8 +163,6 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
-    token=token,
-    authorization=authorization,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -209,8 +175,6 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -218,8 +182,6 @@ async def asyncio(
 
   Args:
       subscription_id (str):
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -235,7 +197,5 @@ async def asyncio(
       subscription_id=subscription_id,
       client=client,
       body=body,
-      token=token,
-      authorization=authorization,
     )
   ).parsed

robosystems_client/extensions/__init__.py

@@ -94,6 +94,7 @@ except ImportError:
   HAS_PANDAS = False
   DataFrameQueryClient = None
   # Set placeholders for optional functions
+  query_result_to_dataframe = None
   parse_datetime_columns = None
   _stream_to_dataframe = None
   dataframe_to_cypher_params = None
@@ -189,14 +190,20 @@ def stream_query(graph_id: str, query: str, parameters=None, chunk_size=None):
 
 
 # DataFrame convenience functions (if pandas is available)
-if HAS_PANDAS:
+if (
+  HAS_PANDAS
+  and query_result_to_dataframe is not None
+  and _stream_to_dataframe is not None
+):
 
   def query_to_dataframe(graph_id: str, query: str, parameters=None, **kwargs):
     """Execute query and return results as pandas DataFrame"""
+    assert query_result_to_dataframe is not None
     result = execute_query(graph_id, query, parameters)
     return query_result_to_dataframe(result, **kwargs)
 
   def stream_to_dataframe(graph_id: str, query: str, parameters=None, chunk_size=10000):
     """Stream query results and return as pandas DataFrame"""
+    assert _stream_to_dataframe is not None
     stream = stream_query(graph_id, query, parameters, chunk_size)
     return _stream_to_dataframe(stream, chunk_size)

robosystems_client/extensions/auth_integration.py

@@ -60,12 +60,11 @@ class AuthenticatedExtensions(RoboSystemsExtensions):
 
     request = CypherQueryRequest(query=query, parameters=parameters or {})
 
-    # Pass the token parameter along with the client
+    # Execute the query
     response = sync_detailed(
       graph_id=graph_id,
       client=self._authenticated_client,
       body=request,
-      token=self._authenticated_client.token,
     )
 
     if response.parsed:

robosystems_client/extensions/query_client.py

@@ -476,7 +476,7 @@ class QueryClient:
     self,
     graph_id: str,
     queries: List[str],
-    parameters_list: Optional[List[Dict[str, Any]]] = None,
+    parameters_list: Optional[List[Optional[Dict[str, Any]]]] = None,
     parallel: bool = False,
   ) -> List[Union[QueryResult, Dict[str, Any]]]:
     """Execute multiple queries in batch
@@ -497,7 +497,8 @@ class QueryClient:
         ... ])
     """
     if parameters_list is None:
-      parameters_list = [None] * len(queries)
+      # Create a list of None values for each query
+      parameters_list = [None for _ in queries]
 
     if len(queries) != len(parameters_list):
       raise ValueError("queries and parameters_list must have same length")

robosystems_client/extensions/sse_client.py

@@ -53,7 +53,7 @@ class SSEEvent:
   data: Any
   id: Optional[str] = None
   retry: Optional[int] = None
-  timestamp: datetime = None
+  timestamp: Optional[datetime] = None
 
   def __post_init__(self) -> None:
     if self.timestamp is None:

robosystems_client/extensions/table_ingest_client.py

@@ -114,6 +114,9 @@ class TableIngestClient:
       file_or_buffer, "read"
     )
 
+    # Initialize file_path for type checking
+    file_path: Optional[Path] = None
+
     if is_buffer:
       # Handle buffer upload
       file_name = options.file_name or "data.parquet"
@@ -218,6 +221,8 @@ class TableIngestClient:
           file_content = file_or_buffer.read()
       else:
         # Read from file path
+        if file_path is None:
+          raise ValueError("file_path should not be None when not using buffer")
         with open(file_path, "rb") as f:
           file_content = f.read()
 

robosystems_client/extensions/utils.py

@@ -144,7 +144,7 @@ class ResultProcessor:
             writer.writerow(headers)
 
           for row in data:
-            if isinstance(row, dict):
+            if isinstance(row, dict) and headers:
               writer.writerow([row.get(h, "") for h in headers])
             elif isinstance(row, (list, tuple)):
               writer.writerow(row)
@@ -159,7 +159,7 @@ class ResultProcessor:
           writer.writerow(headers)
 
         for row in data:
-          if isinstance(row, dict):
+          if isinstance(row, dict) and headers:
             writer.writerow([row.get(h, "") for h in headers])
           elif isinstance(row, (list, tuple)):
             writer.writerow(row)

robosystems_client/models/__init__.py

@@ -124,9 +124,6 @@ from .get_graph_limits_response_getgraphlimits import (
 from .get_graph_monthly_bill_response_getgraphmonthlybill import (
   GetGraphMonthlyBillResponseGetgraphmonthlybill,
 )
-from .get_graph_schema_response_getgraphschema import (
-  GetGraphSchemaResponseGetgraphschema,
-)
 from .get_graph_usage_details_response_getgraphusagedetails import (
   GetGraphUsageDetailsResponseGetgraphusagedetails,
 )
@@ -202,6 +199,8 @@ from .schema_export_response_data_stats_type_0 import SchemaExportResponseDataSt
 from .schema_export_response_schema_definition_type_0 import (
   SchemaExportResponseSchemaDefinitionType0,
 )
+from .schema_info_response import SchemaInfoResponse
+from .schema_info_response_schema import SchemaInfoResponseSchema
 from .schema_validation_request import SchemaValidationRequest
 from .schema_validation_request_schema_definition_type_0 import (
   SchemaValidationRequestSchemaDefinitionType0,
@@ -361,7 +360,6 @@ __all__ = (
   "GetGraphBillingHistoryResponseGetgraphbillinghistory",
   "GetGraphLimitsResponseGetgraphlimits",
   "GetGraphMonthlyBillResponseGetgraphmonthlybill",
-  "GetGraphSchemaResponseGetgraphschema",
   "GetGraphUsageDetailsResponseGetgraphusagedetails",
   "GetOperationStatusResponseGetoperationstatus",
   "GetSharedRepositoryLimitsResponseGetsharedrepositorylimits",
@@ -417,6 +415,8 @@ __all__ = (
   "SchemaExportResponse",
   "SchemaExportResponseDataStatsType0",
   "SchemaExportResponseSchemaDefinitionType0",
+  "SchemaInfoResponse",
+  "SchemaInfoResponseSchema",
   "SchemaValidationRequest",
   "SchemaValidationRequestSchemaDefinitionType0",
   "SchemaValidationResponse",

robosystems_client/models/create_graph_request.py

@@ -20,9 +20,10 @@ class CreateGraphRequest:
   """Request model for creating a new graph.
 
   Example:
-      {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'}, 'instance_tier':
-          'kuzu-standard', 'metadata': {'description': 'Main production graph', 'graph_name': 'Production System',
-          'schema_extensions': ['roboledger']}, 'tags': ['production', 'finance']}
+      {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri': 'https://acmeconsulting.com'},
+          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Professional consulting services with full
+          accounting integration', 'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}
 
   Attributes:
       metadata (GraphMetadata): Metadata for graph creation.

robosystems_client/models/cypher_query_request.py

@@ -2,7 +2,6 @@ from collections.abc import Mapping
 from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
-from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
@@ -20,15 +19,16 @@ class CypherQueryRequest:
   """Request model for Cypher query execution.
 
   Attributes:
-      query (str): The Cypher query to execute
-      parameters (Union['CypherQueryRequestParametersType0', None, Unset]): Optional parameters for the Cypher query
+      query (str): The Cypher query to execute. Use parameters ($param_name) for all dynamic values to prevent
+          injection attacks.
+      parameters (Union['CypherQueryRequestParametersType0', None, Unset]): Query parameters for safe value
+          substitution. ALWAYS use parameters instead of string interpolation.
       timeout (Union[None, Unset, int]): Query timeout in seconds (1-300) Default: 60.
   """
 
   query: str
   parameters: Union["CypherQueryRequestParametersType0", None, Unset] = UNSET
   timeout: Union[None, Unset, int] = 60
-  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
     from ..models.cypher_query_request_parameters_type_0 import (
@@ -52,7 +52,7 @@ class CypherQueryRequest:
       timeout = self.timeout
 
     field_dict: dict[str, Any] = {}
-    field_dict.update(self.additional_properties)
+
     field_dict.update(
       {
         "query": query,
@@ -108,21 +108,4 @@ class CypherQueryRequest:
       timeout=timeout,
     )
 
-    cypher_query_request.additional_properties = d
     return cypher_query_request
-
-  @property
-  def additional_keys(self) -> list[str]:
-    return list(self.additional_properties.keys())
-
-  def __getitem__(self, key: str) -> Any:
-    return self.additional_properties[key]
-
-  def __setitem__(self, key: str, value: Any) -> None:
-    self.additional_properties[key] = value
-
-  def __delitem__(self, key: str) -> None:
-    del self.additional_properties[key]
-
-  def __contains__(self, key: str) -> bool:
-    return key in self.additional_properties

robosystems_client/models/schema_export_response.py

@@ -24,10 +24,12 @@ class SchemaExportResponse:
 
   Attributes:
       graph_id (str): Graph ID
-      schema_definition (Union['SchemaExportResponseSchemaDefinitionType0', str]): Exported schema definition
+      schema_definition (Union['SchemaExportResponseSchemaDefinitionType0', str]): Exported schema definition (format
+          depends on 'format' parameter)
       format_ (str): Export format used
       exported_at (str): Export timestamp
-      data_stats (Union['SchemaExportResponseDataStatsType0', None, Unset]): Data statistics if requested
+      data_stats (Union['SchemaExportResponseDataStatsType0', None, Unset]): Data statistics if requested (only when
+          include_data_stats=true)
   """
 
   graph_id: str

robosystems_client/models/schema_info_response.py

@@ -0,0 +1,77 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+  from ..models.schema_info_response_schema import SchemaInfoResponseSchema
+
+
+T = TypeVar("T", bound="SchemaInfoResponse")
+
+
+@_attrs_define
+class SchemaInfoResponse:
+  """Response model for runtime schema introspection.
+
+  This model represents the actual current state of the graph database,
+  showing what node labels, relationship types, and properties exist right now.
+
+      Attributes:
+          graph_id (str): Graph database identifier
+          schema (SchemaInfoResponseSchema): Runtime schema information showing actual database structure
+  """
+
+  graph_id: str
+  schema: "SchemaInfoResponseSchema"
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    graph_id = self.graph_id
+
+    schema = self.schema.to_dict()
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "graph_id": graph_id,
+        "schema": schema,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.schema_info_response_schema import SchemaInfoResponseSchema
+
+    d = dict(src_dict)
+    graph_id = d.pop("graph_id")
+
+    schema = SchemaInfoResponseSchema.from_dict(d.pop("schema"))
+
+    schema_info_response = cls(
+      graph_id=graph_id,
+      schema=schema,
+    )
+
+    schema_info_response.additional_properties = d
+    return schema_info_response
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/{get_graph_schema_response_getgraphschema.py → schema_info_response_schema.py}

@@ -4,12 +4,12 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="GetGraphSchemaResponseGetgraphschema")
+T = TypeVar("T", bound="SchemaInfoResponseSchema")
 
 
 @_attrs_define
-class GetGraphSchemaResponseGetgraphschema:
-  """ """
+class SchemaInfoResponseSchema:
+  """Runtime schema information showing actual database structure"""
 
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -22,10 +22,10 @@ class GetGraphSchemaResponseGetgraphschema:
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    get_graph_schema_response_getgraphschema = cls()
+    schema_info_response_schema = cls()
 
-    get_graph_schema_response_getgraphschema.additional_properties = d
-    return get_graph_schema_response_getgraphschema
+    schema_info_response_schema.additional_properties = d
+    return schema_info_response_schema
 
   @property
   def additional_keys(self) -> list[str]:

robosystems_client/models/schema_validation_response.py

@@ -25,12 +25,13 @@ class SchemaValidationResponse:
   Attributes:
       valid (bool): Whether the schema is valid
       message (str): Validation message
-      errors (Union[None, Unset, list[str]]): List of validation errors
-      warnings (Union[None, Unset, list[str]]): List of warnings
-      stats (Union['SchemaValidationResponseStatsType0', None, Unset]): Schema statistics (nodes, relationships,
-          properties)
-      compatibility (Union['SchemaValidationResponseCompatibilityType0', None, Unset]): Compatibility check results if
-          requested
+      errors (Union[None, Unset, list[str]]): List of validation errors (only present when valid=false)
+      warnings (Union[None, Unset, list[str]]): List of validation warnings (schema is still valid but has potential
+          issues)
+      stats (Union['SchemaValidationResponseStatsType0', None, Unset]): Schema statistics (only present when
+          valid=true)
+      compatibility (Union['SchemaValidationResponseCompatibilityType0', None, Unset]): Compatibility check results
+          (only when check_compatibility specified)
   """
 
   valid: bool

robosystems_client/models/table_query_request.py

@@ -1,8 +1,10 @@
 from collections.abc import Mapping
-from typing import Any, TypeVar
+from typing import Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 
+from ..types import UNSET, Unset
+
 T = TypeVar("T", bound="TableQueryRequest")
 
 
@@ -10,14 +12,27 @@ T = TypeVar("T", bound="TableQueryRequest")
 class TableQueryRequest:
   """
   Attributes:
-      sql (str): SQL query to execute on staging tables
+      sql (str): SQL query to execute on staging tables. Use ? placeholders or $param_name for dynamic values to
+          prevent SQL injection.
+      parameters (Union[None, Unset, list[Any]]): Query parameters for safe value substitution. ALWAYS use parameters
+          instead of string concatenation.
   """
 
   sql: str
+  parameters: Union[None, Unset, list[Any]] = UNSET
 
   def to_dict(self) -> dict[str, Any]:
     sql = self.sql
 
+    parameters: Union[None, Unset, list[Any]]
+    if isinstance(self.parameters, Unset):
+      parameters = UNSET
+    elif isinstance(self.parameters, list):
+      parameters = self.parameters
+
+    else:
+      parameters = self.parameters
+
     field_dict: dict[str, Any] = {}
 
     field_dict.update(
@@ -25,6 +40,8 @@ class TableQueryRequest:
         "sql": sql,
       }
     )
+    if parameters is not UNSET:
+      field_dict["parameters"] = parameters
 
     return field_dict
 
@@ -33,8 +50,26 @@ class TableQueryRequest:
     d = dict(src_dict)
     sql = d.pop("sql")
 
+    def _parse_parameters(data: object) -> Union[None, Unset, list[Any]]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      try:
+        if not isinstance(data, list):
+          raise TypeError()
+        parameters_type_0 = cast(list[Any], data)
+
+        return parameters_type_0
+      except:  # noqa: E722
+        pass
+      return cast(Union[None, Unset, list[Any]], data)
+
+    parameters = _parse_parameters(d.pop("parameters", UNSET))
+
     table_query_request = cls(
       sql=sql,
+      parameters=parameters,
     )
 
     return table_query_request

{robosystems_client-0.2.3.dist-info → robosystems_client-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.3
+Version: 0.2.4
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT
@@ -35,9 +35,9 @@ Provides-Extra: all
 Requires-Dist: pandas>=1.5.0; extra == 'all'
 Requires-Dist: pyarrow>=10.0.0; extra == 'all'
 Provides-Extra: dev
+Requires-Dist: basedpyright>=1.21.0; extra == 'dev'
 Requires-Dist: build>=1.0.0; extra == 'dev'
 Requires-Dist: openapi-python-client>=0.21.8; extra == 'dev'
-Requires-Dist: pyright>=1.1.402; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.26.0; extra == 'dev'
 Requires-Dist: pytest-httpx>=0.30.0; extra == 'dev'
 Requires-Dist: pytest>=8.3.5; extra == 'dev'
@@ -80,8 +80,6 @@ pip install robosystems-client[tables]
 pip install robosystems-client[all]
 ```
 
-See the [examples](./examples) directory for usage guides.
-
 ## API Reference
 
 - [API reference](https://api.robosystems.ai)