robosystems-client 0.1.19__py3-none-any.whl → 0.2.1__py3-none-any.whl

robosystems_client/api/user/update_user_password.py

@@ -54,22 +54,27 @@ def _parse_response(
     response_200 = SuccessResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 400:
     response_400 = ErrorResponse.from_dict(response.json())
 
     return response_400
+
   if response.status_code == 404:
     response_404 = ErrorResponse.from_dict(response.json())
 
     return response_404
-  if response.status_code == 500:
-    response_500 = ErrorResponse.from_dict(response.json())
 
-    return response_500
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = ErrorResponse.from_dict(response.json())
+
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_analytics/get_detailed_user_analytics.py

@@ -53,10 +53,12 @@ def _parse_response(
     response_200 = UserAnalyticsResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_analytics/get_user_usage_overview.py

@@ -47,10 +47,12 @@ def _parse_response(
     response_200 = UserUsageSummaryResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_limits/get_all_shared_repository_limits.py

@@ -58,10 +58,12 @@ def _parse_response(
     )
 
     return response_200
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_limits/get_shared_repository_limits.py

@@ -54,10 +54,12 @@ def _parse_response(
     )
 
     return response_200
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:
@@ -96,7 +98,7 @@ def sync_detailed(
       - Remaining quota
       - Reset times
 
-      Note: All queries are FREE - this only shows rate limit status.
+      Note: All queries are included - this only shows rate limit status.
 
   Args:
       repository (str): Repository name (e.g., 'sec')
@@ -143,7 +145,7 @@ def sync(
       - Remaining quota
       - Reset times
 
-      Note: All queries are FREE - this only shows rate limit status.
+      Note: All queries are included - this only shows rate limit status.
 
   Args:
       repository (str): Repository name (e.g., 'sec')
@@ -185,7 +187,7 @@ async def asyncio_detailed(
       - Remaining quota
       - Reset times
 
-      Note: All queries are FREE - this only shows rate limit status.
+      Note: All queries are included - this only shows rate limit status.
 
   Args:
       repository (str): Repository name (e.g., 'sec')
@@ -230,7 +232,7 @@ async def asyncio(
       - Remaining quota
       - Reset times
 
-      Note: All queries are FREE - this only shows rate limit status.
+      Note: All queries are included - this only shows rate limit status.
 
   Args:
       repository (str): Repository name (e.g., 'sec')

robosystems_client/api/user_limits/get_user_limits.py

@@ -47,13 +47,16 @@ def _parse_response(
     response_200 = UserLimitsResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 404:
     response_404 = cast(Any, None)
     return response_404
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_limits/get_user_usage.py

@@ -47,10 +47,12 @@ def _parse_response(
     response_200 = UserUsageResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py

@@ -48,19 +48,24 @@ def _parse_response(
     response_200 = CancellationResponse.from_dict(response.json())
 
     return response_200
-  if response.status_code == 404:
-    response_404 = cast(Any, None)
-    return response_404
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
+  if response.status_code == 404:
+    response_404 = cast(Any, None)
+    return response_404
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/get_repository_credits.py

@@ -48,16 +48,20 @@ def _parse_response(
     response_200 = RepositoryCreditsResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/get_shared_repository_credits.py

@@ -47,16 +47,20 @@ def _parse_response(
     response_200 = CreditsSummaryResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py

@@ -50,16 +50,20 @@ def _parse_response(
     response_200 = UserSubscriptionsResponse.from_dict(response.json())
 
     return response_200
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py

@@ -53,19 +53,24 @@ def _parse_response(
     response_201 = SubscriptionResponse.from_dict(response.json())
 
     return response_201
+
   if response.status_code == 400:
     response_400 = cast(Any, None)
     return response_400
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py

@@ -52,22 +52,28 @@ def _parse_response(
   if response.status_code == 200:
     response_200 = response.json()
     return response_200
+
   if response.status_code == 400:
     response_400 = cast(Any, None)
     return response_400
-  if response.status_code == 404:
-    response_404 = cast(Any, None)
-    return response_404
+
   if response.status_code == 401:
     response_401 = cast(Any, None)
     return response_401
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
+
+  if response.status_code == 404:
+    response_404 = cast(Any, None)
+    return response_404
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
     return response_422
+
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:

robosystems_client/extensions/README.md

@@ -11,8 +11,7 @@ The RoboSystems Python Client Extensions provide enhanced functionality for the
 
 - **Server-Sent Events (SSE)** streaming with automatic reconnection
 - **Smart Query Execution** with automatic strategy selection
-- **Data Copy Operations** with S3 import and real-time progress tracking
-- **Operation Monitoring** for long-running operations  
+- **Operation Monitoring** for long-running operations
 - **Connection Pooling** and intelligent resource management
 - **Result Processing** and format conversion utilities
 - **Caching** with TTL and LRU eviction
@@ -85,62 +84,6 @@ async def main():
 asyncio.run(main())
 ```
 
-### Data Copy Operations
-
-```python
-from robosystems_client.extensions import CopyClient, CopyOptions
-from robosystems_client.models.s3_copy_request import S3CopyRequest
-from robosystems_client.models.s3_copy_request_file_format import S3CopyRequestFileFormat
-
-# Initialize copy client
-copy_client = CopyClient({
-    "base_url": "https://api.robosystems.ai",
-    "api_key": "your-api-key",
-})
-
-# Create S3 copy request
-request = S3CopyRequest(
-    table_name="companies",
-    s3_path="s3://my-bucket/data/companies.csv",
-    s3_access_key_id="AWS_ACCESS_KEY",
-    s3_secret_access_key="AWS_SECRET_KEY",
-    s3_region="us-east-1",
-    file_format=S3CopyRequestFileFormat.CSV,
-    ignore_errors=False,  # Stop on first error
-)
-
-# Set up progress callbacks
-def on_progress(message, percent):
-    if percent:
-        print(f"Progress: {message} ({percent}%)")
-    else:
-        print(f"Progress: {message}")
-
-def on_warning(warning):
-    print(f"Warning: {warning}")
-
-options = CopyOptions(
-    on_progress=on_progress,
-    on_warning=on_warning,
-)
-
-# Execute copy with progress monitoring
-result = copy_client.copy_from_s3("your_graph_id", request, options)
-
-# Check results
-if result.status == "completed":
-    print(f"✅ Successfully imported {result.rows_imported:,} rows")
-    stats = copy_client.calculate_statistics(result)
-    if stats:
-        print(f"Throughput: {stats.throughput:.2f} rows/second")
-elif result.status == "partial":
-    print(f"⚠️ Imported {result.rows_imported:,} rows, skipped {result.rows_skipped:,}")
-else:
-    print(f"❌ Copy failed: {result.error}")
-
-copy_client.close()
-```
-
 ## 🔐 Authentication
 
 ### API Key Authentication (Recommended)
@@ -198,79 +141,6 @@ dev_ext = create_extensions(
 
 ## 🛠 Advanced Features
 
-### Copy Operations with Advanced Features
-
-```python
-from robosystems_client.extensions import CopyClient, CopySourceType
-
-# Batch copy multiple tables
-copy_client = CopyClient({
-    "base_url": "https://api.robosystems.ai",
-    "api_key": "your-api-key",
-})
-
-copies = [
-    {
-        "request": S3CopyRequest(
-            table_name="companies",
-            s3_path="s3://bucket/companies.csv",
-            s3_access_key_id="KEY",
-            s3_secret_access_key="SECRET",
-            file_format=S3CopyRequestFileFormat.CSV,
-        ),
-    },
-    {
-        "request": S3CopyRequest(
-            table_name="transactions",
-            s3_path="s3://bucket/transactions.parquet",
-            s3_access_key_id="KEY",
-            s3_secret_access_key="SECRET",
-            file_format=S3CopyRequestFileFormat.PARQUET,
-            ignore_errors=True,  # Continue on errors
-        ),
-    },
-]
-
-# Execute batch copy
-results = copy_client.batch_copy_from_s3("graph_id", copies)
-
-for i, result in enumerate(results):
-    table_name = copies[i]["request"].table_name
-    print(f"{table_name}: {result.status}")
-    if result.rows_imported:
-        print(f"  Imported: {result.rows_imported:,} rows")
-
-# Copy with retry logic for resilient operations
-result = copy_client.copy_with_retry(
-    graph_id="graph_id",
-    request=S3CopyRequest(
-        table_name="large_dataset",
-        s3_path="s3://bucket/large-dataset.csv",
-        s3_access_key_id="KEY",
-        s3_secret_access_key="SECRET",
-        max_file_size_gb=50,
-        extended_timeout=True,
-    ),
-    source_type=CopySourceType.S3,
-    max_retries=3,
-    options=CopyOptions(
-        on_progress=lambda msg, _: print(msg)
-    ),
-)
-
-# Monitor multiple concurrent copy operations
-operation_ids = ["op-123", "op-456", "op-789"]
-results = copy_client.monitor_multiple_copies(operation_ids, options)
-
-for op_id, result in results.items():
-    print(f"Operation {op_id}: {result.status}")
-    if result.status == "completed":
-        stats = copy_client.calculate_statistics(result)
-        print(f"  Throughput: {stats.throughput:.2f} rows/sec")
-
-copy_client.close()
-```
-
 ### Query Builder
 
 Build complex Cypher queries programmatically:
@@ -403,87 +273,6 @@ client.close()
 
 ## 📊 Examples
 
-### Data Import with Real-Time Monitoring
-
-```python
-from robosystems_client.extensions import CopyClient, CopyOptions
-import time
-
-def import_financial_data():
-    """Import financial data with comprehensive monitoring"""
-    
-    copy_client = CopyClient({
-        "base_url": "https://api.robosystems.ai",
-        "api_key": "your-api-key",
-    })
-    
-    # Track progress history
-    progress_history = []
-    warnings_count = 0
-    
-    def on_progress(message, percent):
-        timestamp = time.strftime("%H:%M:%S")
-        progress_history.append({
-            "time": timestamp,
-            "message": message,
-            "percent": percent,
-        })
-        print(f"[{timestamp}] {message}" + (f" ({percent}%)" if percent else ""))
-    
-    def on_warning(warning):
-        nonlocal warnings_count
-        warnings_count += 1
-        print(f"⚠️  Warning #{warnings_count}: {warning}")
-    
-    def on_queue_update(position, wait_time):
-        print(f"📊 Queue position: {position} (ETA: {wait_time}s)")
-    
-    # Configure copy with all callbacks
-    options = CopyOptions(
-        on_progress=on_progress,
-        on_warning=on_warning,
-        on_queue_update=on_queue_update,
-        timeout=1800000,  # 30 minutes
-    )
-    
-    # Execute copy operation
-    start_time = time.time()
-    
-    result = copy_client.copy_s3(
-        graph_id="financial_graph",
-        table_name="quarterly_reports",
-        s3_path="s3://financial-data/reports-2024-q1.parquet",
-        access_key_id="AWS_KEY",
-        secret_access_key="AWS_SECRET",
-        file_format="parquet",
-        ignore_errors=True,  # Continue on validation errors
-    )
-    
-    # Print summary
-    elapsed = time.time() - start_time
-    
-    print("\n" + "="*50)
-    print("📈 IMPORT SUMMARY")
-    print("="*50)
-    print(f"Status: {result.status.upper()}")
-    print(f"Rows Imported: {result.rows_imported or 0:,}")
-    print(f"Rows Skipped: {result.rows_skipped or 0:,}")
-    print(f"Warnings: {warnings_count}")
-    print(f"Execution Time: {elapsed:.2f} seconds")
-    
-    if result.status == "completed":
-        stats = copy_client.calculate_statistics(result)
-        if stats:
-            print(f"Throughput: {stats.throughput:.2f} rows/second")
-            print(f"Data Processed: {stats.bytes_processed / (1024*1024):.2f} MB")
-    
-    copy_client.close()
-    return result
-
-# Run the import
-result = import_financial_data()
-```
-
 ### Financial Data Analysis
 
 ```python

robosystems_client/extensions/__init__.py

@@ -20,13 +20,12 @@ from .operation_client import (
   OperationProgress,
   OperationResult,
 )
-from .copy_client import (
-  CopyClient,
-  AsyncCopyClient,
-  CopySourceType,
-  CopyOptions,
-  CopyResult,
-  CopyStatistics,
+from .table_ingest_client import (
+  TableIngestClient,
+  UploadOptions,
+  IngestOptions,
+  UploadResult,
+  TableInfo,
 )
 from .extensions import (
   RoboSystemsExtensions,
@@ -117,13 +116,12 @@ __all__ = [
   "OperationStatus",
   "OperationProgress",
   "OperationResult",
-  # Copy Client
-  "CopyClient",
-  "AsyncCopyClient",
-  "CopySourceType",
-  "CopyOptions",
-  "CopyResult",
-  "CopyStatistics",
+  # Table Ingest Client
+  "TableIngestClient",
+  "UploadOptions",
+  "IngestOptions",
+  "UploadResult",
+  "TableInfo",
   # Utilities
   "QueryBuilder",
   "ResultProcessor",
@@ -179,20 +177,6 @@ def stream_query(graph_id: str, query: str, parameters=None, chunk_size=None):
   return extensions.query.stream_query(graph_id, query, parameters, chunk_size)
 
 
-def copy_from_s3(
-  graph_id: str,
-  table_name: str,
-  s3_path: str,
-  access_key_id: str,
-  secret_access_key: str,
-  **kwargs,
-):
-  """Copy data from S3 using the default extensions instance"""
-  return extensions.copy_from_s3(
-    graph_id, table_name, s3_path, access_key_id, secret_access_key, **kwargs
-  )
-
-
 # DataFrame convenience functions (if pandas is available)
 if HAS_PANDAS:
 

robosystems_client/extensions/extensions.py

@@ -8,7 +8,7 @@ from typing import Dict, Any, Optional, Callable
 
 from .query_client import QueryClient
 from .operation_client import OperationClient
-from .copy_client import CopyClient
+from .table_ingest_client import TableIngestClient
 from .sse_client import SSEClient
 
 
@@ -56,9 +56,9 @@ class RoboSystemsExtensions:
       self.config["token"] = token
 
     # Initialize clients
-    self.copy = CopyClient(self.config)
     self.query = QueryClient(self.config)
     self.operations = OperationClient(self.config)
+    self.tables = TableIngestClient(self.config)
 
   def monitor_operation(
     self, operation_id: str, on_progress: Optional[Callable] = None
@@ -85,9 +85,9 @@ class RoboSystemsExtensions:
 
   def close(self):
     """Clean up all active connections"""
-    self.copy.close()
     self.query.close()
     self.operations.close_all()
+    self.tables.close()
 
   # Convenience methods that delegate to the appropriate clients
   def execute_query(self, graph_id: str, query: str, parameters: Dict[str, Any] = None):
@@ -112,20 +112,6 @@ class RoboSystemsExtensions:
     """Cancel an operation using the operation client"""
     return self.operations.cancel_operation(operation_id)
 
-  def copy_from_s3(
-    self,
-    graph_id: str,
-    table_name: str,
-    s3_path: str,
-    access_key_id: str,
-    secret_access_key: str,
-    **kwargs,
-  ):
-    """Copy data from S3 using the copy client"""
-    return self.copy.copy_s3(
-      graph_id, table_name, s3_path, access_key_id, secret_access_key, **kwargs
-    )
-
 
 class AsyncRoboSystemsExtensions:
   """Async version of the extensions class"""