From c988828cb807a2874e7f53e4e6734bfe5f35b070 Mon Sep 17 00:00:00 2001 From: Joey French Date: Wed, 29 Oct 2025 00:07:28 -0500 Subject: [PATCH 1/4] Refactor type checking and clean up project files ## Summary This commit refines the type checking process and removes unused example files, enhancing project organization and maintainability. ## Key Changes - Updated the type checking command in the `justfile` to use `basedpyright` for improved type analysis. - Removed several unused example files and their corresponding README, streamlining the codebase. - Added new type checking configurations in `pyproject.toml` to suppress specific type report warnings. ## Testing Notes - Ensure that type checking runs successfully with the new configuration. - Verify that the project builds correctly after the removal of example files. ## Infrastructure Considerations The cleanup of unused files reduces clutter, making the project easier to navigate for developers. --- .github/workflows/test.yml | 4 + README.md | 2 - examples/README.md | 272 ----------- examples/basic_query.py | 99 ---- examples/e2e_workflow.py | 445 ------------------ justfile | 3 +- pyproject.toml | 8 +- .../api/agent/auto_select_agent.py | 42 +- .../api/agent/batch_process_queries.py | 42 +- .../api/agent/execute_specific_agent.py | 42 +- .../api/agent/get_agent_metadata.py | 54 +-- robosystems_client/api/agent/list_agents.py | 46 +- .../api/agent/recommend_agent.py | 50 +- .../api/auth/generate_sso_token.py | 18 - .../api/auth/get_current_auth_user.py | 88 +--- robosystems_client/api/auth/logout_user.py | 64 +-- .../api/auth/refresh_auth_session.py | 64 +-- .../api/auth/resend_verification_email.py | 88 +--- .../api/backup/create_backup.py | 50 +- .../api/backup/get_backup_download_url.py | 46 +- .../api/backup/get_backup_stats.py | 54 +-- robosystems_client/api/backup/list_backups.py | 46 +- .../api/backup/restore_backup.py | 50 +- .../api/connections/create_connection.py | 50 +- .../api/connections/create_link_token.py | 50 +- .../api/connections/delete_connection.py | 54 +-- .../api/connections/exchange_link_token.py | 50 +- .../api/connections/get_connection.py | 54 +-- .../api/connections/get_connection_options.py | 54 +-- .../api/connections/init_o_auth.py | 50 +- .../api/connections/list_connections.py | 46 +- .../api/connections/oauth_callback.py | 50 +- .../api/connections/sync_connection.py | 50 +- .../api/graph_analytics/get_graph_metrics.py | 54 +-- .../graph_analytics/get_graph_usage_stats.py | 46 +- .../graph_billing/get_current_graph_bill.py | 54 +-- .../get_graph_billing_history.py | 46 +- .../graph_billing/get_graph_monthly_bill.py | 54 +-- .../graph_billing/get_graph_usage_details.py | 46 +- .../api/graph_credits/check_credit_balance.py | 38 -- .../api/graph_credits/check_storage_limits.py | 46 +- .../api/graph_credits/get_credit_summary.py | 46 +- .../api/graph_credits/get_storage_usage.py | 38 -- .../graph_credits/list_credit_transactions.py | 46 +- .../api/graph_health/get_database_health.py | 54 +-- .../api/graph_info/get_database_info.py | 54 +-- .../api/graph_limits/get_graph_limits.py | 54 +-- robosystems_client/api/graphs/create_graph.py | 42 +- robosystems_client/api/graphs/get_graphs.py | 89 +--- robosystems_client/api/graphs/select_graph.py | 46 +- robosystems_client/api/mcp/call_mcp_tool.py | 43 +- robosystems_client/api/mcp/list_mcp_tools.py | 54 +-- .../api/operations/cancel_operation.py | 46 +- .../api/operations/get_operation_status.py | 46 +- .../api/query/execute_cypher_query.py | 43 +- .../api/schema/export_graph_schema.py | 46 +- .../api/schema/get_graph_schema.py | 54 +-- .../api/schema/validate_schema.py | 50 +- .../api/subgraphs/create_subgraph.py | 50 +- .../api/subgraphs/delete_subgraph.py | 50 +- .../api/subgraphs/get_subgraph_info.py | 54 +-- .../api/subgraphs/get_subgraph_quota.py | 54 +-- .../api/subgraphs/list_subgraphs.py | 54 +-- robosystems_client/api/tables/delete_file.py | 54 +-- .../api/tables/get_file_info.py | 54 +-- .../api/tables/get_upload_url.py | 50 +- .../api/tables/ingest_tables.py | 50 +- .../api/tables/list_table_files.py | 54 +-- robosystems_client/api/tables/list_tables.py | 54 +-- robosystems_client/api/tables/query_tables.py | 50 +- .../api/tables/update_file_status.py | 50 +- .../api/user/create_user_api_key.py | 42 +- .../api/user/get_all_credit_summaries.py | 125 +---- .../api/user/get_current_user.py | 89 +--- .../api/user/list_user_api_keys.py | 89 +--- .../api/user/revoke_user_api_key.py | 46 +- robosystems_client/api/user/update_user.py | 42 +- .../api/user/update_user_api_key.py | 42 +- .../api/user/update_user_password.py | 42 +- .../get_detailed_user_analytics.py | 38 -- .../user_analytics/get_user_usage_overview.py | 89 +--- .../get_all_shared_repository_limits.py | 119 +---- .../get_shared_repository_limits.py | 46 +- .../api/user_limits/get_user_limits.py | 89 +--- .../api/user_limits/get_user_usage.py | 89 +--- .../cancel_shared_repository_subscription.py | 46 +- .../get_repository_credits.py | 46 +- .../get_shared_repository_credits.py | 89 +--- .../get_user_shared_subscriptions.py | 38 -- .../subscribe_to_shared_repository.py | 42 +- .../upgrade_shared_repository_subscription.py | 42 +- robosystems_client/extensions/__init__.py | 9 +- .../extensions/auth_integration.py | 3 +- robosystems_client/extensions/query_client.py | 5 +- robosystems_client/extensions/sse_client.py | 2 +- .../extensions/table_ingest_client.py | 5 + robosystems_client/extensions/utils.py | 4 +- 97 files changed, 456 insertions(+), 4960 deletions(-) delete mode 100644 examples/README.md delete mode 100644 examples/basic_query.py delete mode 100644 examples/e2e_workflow.py diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index a4ccfaf..355ca14 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -37,3 +37,7 @@ jobs: - name: Run Lint run: | uv run ruff check . + + - name: Run type checking + run: | + uv run basedpyright diff --git a/README.md b/README.md index 50224d8..324069b 100644 --- a/README.md +++ b/README.md @@ -27,8 +27,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) diff --git a/examples/README.md b/examples/README.md deleted file mode 100644 index a5b01aa..0000000 --- a/examples/README.md +++ /dev/null @@ -1,272 +0,0 @@ -# RoboSystems Python Client Examples - -This directory contains examples demonstrating how to use the RoboSystems Python Client SDK for various common workflows. - -## Prerequisites - -```bash -# Basic installation (for querying only) -pip install robosystems-client - -# With table ingestion support (for uploading Parquet files) -pip install robosystems-client[tables] - -# Or install all optional features -pip install robosystems-client[all] -``` - -## Examples - -### 1. Basic Query (`basic_query.py`) - -**Demonstrates:** Simple querying of an existing graph - -The simplest way to get started with the RoboSystems Python SDK. This example shows how to: -- Initialize the SDK with an API key -- Execute Cypher queries against a graph -- Display query results - -**Usage:** -```bash -python examples/basic_query.py \ - --api-key YOUR_API_KEY \ - --graph-id YOUR_GRAPH_ID \ - --base-url http://localhost:8000 -``` - -**Key Features:** -- Minimal setup -- Multiple query examples (count nodes, list labels, sample data) -- Proper error handling -- Clean resource management - ---- - -### 2. End-to-End Workflow (`e2e_workflow.py`) - -**Demonstrates:** Complete workflow from user creation to querying - -This comprehensive example shows the full RoboSystems workflow: -- User creation and authentication -- API key generation -- Graph creation -- Parquet file upload to staging tables -- Table ingestion into the graph -- Querying the populated graph - -**Usage:** - -**With new user creation:** -```bash -python examples/e2e_workflow.py -``` -This will auto-generate credentials and walk through the entire setup. - -**With existing API key:** -```bash -python examples/e2e_workflow.py --api-key YOUR_API_KEY -``` - -**Custom configuration:** -```bash -python examples/e2e_workflow.py \ - --name "John Doe" \ - --email "john@example.com" \ - --password "YourSecurePassword123!" \ - --base-url http://localhost:8000 -``` - -**Key Features:** -- Automated user setup -- Sample data generation (creates Parquet files) -- **TableIngestClient** extension usage for simplified uploads -- Progress callbacks -- LocalStack URL auto-fix for local development -- Multiple query examples - ---- - -## Extension Features Demonstrated - -### TableIngestClient - -The examples showcase the new `TableIngestClient` extension which dramatically simplifies the table upload process: - -**Before (Raw API calls - 3 steps):** -```python -# Step 1: Get presigned upload URL -response = client.post(f"{base_url}/v1/graphs/{graph_id}/tables/{table_name}/files", ...) -upload_url = response.json()["upload_url"] -file_id = response.json()["file_id"] - -# Step 2: Upload to S3 -with open(file_path, "rb") as f: - s3_client.put(upload_url, data=f.read()) - -# Step 3: Update file metadata -client.patch(f"{base_url}/v1/graphs/{graph_id}/tables/files/{file_id}", ...) -``` - -**After (SDK with TableIngestClient):** -```python -from robosystems_client.extensions import UploadOptions - -# All 3 steps in one call! -upload_result = extensions.tables.upload_parquet_file( - graph_id, - table_name, - file_path, - UploadOptions(on_progress=lambda msg: print(msg)) -) -``` - -**Additional TableIngestClient Methods:** -- `list_staging_tables(graph_id)` - List all staging tables -- `ingest_all_tables(graph_id, options)` - Ingest tables into graph -- `upload_and_ingest(graph_id, table_name, file_path, ...)` - Upload + ingest in one call - ---- - -## Common Patterns - -### Initializing the SDK - -```python -from robosystems_client.extensions import ( - RoboSystemsExtensions, - RoboSystemsExtensionConfig, -) - -# With API key -config = RoboSystemsExtensionConfig( - base_url="http://localhost:8000", - headers={"X-API-Key": "your-api-key"} -) -extensions = RoboSystemsExtensions(config) - -# Always close when done -extensions.close() -``` - -### Uploading Data - -```python -from robosystems_client.extensions import UploadOptions, IngestOptions -from pathlib import Path - -# Upload a parquet file -upload_options = UploadOptions( - on_progress=lambda msg: print(f"Upload: {msg}"), - fix_localstack_url=True # Auto-fix for local development -) - -result = extensions.tables.upload_parquet_file( - graph_id="your-graph-id", - table_name="Entity", - file_path=Path("data/entities.parquet"), - options=upload_options -) - -if result.success: - print(f"āœ… Uploaded {result.row_count:,} rows") -else: - print(f"āŒ Upload failed: {result.error}") - -# Ingest the uploaded tables -ingest_options = IngestOptions( - ignore_errors=True, # Continue on errors - rebuild=False, # Don't rebuild existing data - on_progress=lambda msg: print(f"Ingest: {msg}") -) - -ingest_result = extensions.tables.ingest_all_tables( - graph_id="your-graph-id", - options=ingest_options -) -``` - -### Querying Data - -```python -# Simple query -result = extensions.query.query( - graph_id="your-graph-id", - query="MATCH (n:Entity) RETURN count(n) AS total" -) - -print(f"Total nodes: {result.data[0]['total']}") - -# Query with parameters -result = extensions.query.query( - graph_id="your-graph-id", - query="MATCH (n:Entity) WHERE n.ticker = $ticker RETURN n", - parameters={"ticker": "AAPL"} -) - -for node in result.data: - print(f"Found: {node}") -``` - -## Environment Variables - -You can also use environment variables for configuration: - -```bash -export ROBOSYSTEMS_API_URL="http://localhost:8000" -export ROBOSYSTEMS_API_KEY="your-api-key" -export ROBOSYSTEMS_GRAPH_ID="your-graph-id" -``` - -Then reference them in your code: -```python -import os - -config = RoboSystemsExtensionConfig( - base_url=os.getenv("ROBOSYSTEMS_API_URL", "http://localhost:8000"), - headers={"X-API-Key": os.getenv("ROBOSYSTEMS_API_KEY")} -) -``` - -## Comparison with Raw API Approach - -| Operation | Raw httpx Calls | SDK with Extensions | Lines of Code Reduction | -|-----------|----------------|---------------------|------------------------| -| Upload file to table | ~25 lines (3 API calls) | ~6 lines (1 method) | **76% reduction** | -| Query graph | ~10 lines | ~3 lines | **70% reduction** | -| List tables | ~8 lines | ~2 lines | **75% reduction** | - -**Benefits of Using the SDK:** -- Type-safe model classes -- Automatic error handling -- Progress callbacks -- Connection pooling -- Cleaner, more maintainable code -- Built-in support for async operations - -## Additional Resources - -- **API Documentation:** https://api.robosystems.ai/docs -- **Main Repository:** https://github.com/RoboSystems/robosystems-python-client -- **Extension Documentation:** See `robosystems_client/extensions/README.md` - -## Getting Help - -If you encounter issues: -1. Check the error message and stack trace -2. Verify your API key and graph ID are correct -3. Ensure the API server is running (for local development) -4. Check the API documentation for endpoint changes -5. Open an issue on GitHub with a minimal reproducible example - -## Contributing - -Have a useful example to share? Contributions are welcome! - -1. Fork the repository -2. Create a new example file following the existing patterns -3. Update this README with your example -4. Submit a pull request - ---- - -**Happy coding with RoboSystems! šŸ¤–** diff --git a/examples/basic_query.py b/examples/basic_query.py deleted file mode 100644 index 55e712e..0000000 --- a/examples/basic_query.py +++ /dev/null @@ -1,99 +0,0 @@ -""" -Simple example of querying a RoboSystems graph using the Python SDK. - -This example demonstrates the simplest way to execute queries against -an existing graph using the RoboSystems Python Client SDK. - -Usage: - python examples/basic_query.py --api-key --graph-id -""" - -import argparse -import json -import sys - -from robosystems_client.extensions import ( - RoboSystemsExtensions, - RoboSystemsExtensionConfig, -) - - -def run_query_example( - api_key: str, graph_id: str, base_url: str = "http://localhost:8000" -): - """Run simple query examples against a graph.""" - print("\n" + "=" * 60) - print("šŸ” RoboSystems Query Example (SDK)") - print("=" * 60) - - # Initialize extensions with API key - config = RoboSystemsExtensionConfig(base_url=base_url, headers={"X-API-Key": api_key}) - extensions = RoboSystemsExtensions(config) - - # Example queries - queries = [ - {"name": "Count all nodes", "query": "MATCH (n) RETURN count(n) AS total_nodes"}, - { - "name": "Count nodes by label", - "query": "MATCH (n) RETURN labels(n) AS label, count(n) AS count ORDER BY count DESC LIMIT 10", - }, - {"name": "Sample nodes", "query": "MATCH (n) RETURN n LIMIT 5"}, - ] - - for example in queries: - print(f"\nšŸ“Š {example['name']}") - print(f" Query: {example['query']}") - - try: - # Execute the query - result = extensions.query.query(graph_id, example["query"]) - - # Display results - if hasattr(result, "data") and result.data: - print(f" āœ… Returned {len(result.data)} records") - - # Pretty print results - if len(result.data) <= 10: - print(" Results:") - print(" " + json.dumps(result.data, indent=4).replace("\n", "\n ")) - else: - print(f" Showing first 10 of {len(result.data)} records:") - print(" " + json.dumps(result.data[:10], indent=4).replace("\n", "\n ")) - else: - print(" āš ļø No results returned") - - except Exception as e: - print(f" āŒ Query failed: {e}") - - # Cleanup - extensions.close() - - print("\n" + "=" * 60) - print("āœ… Query examples complete!") - print("=" * 60 + "\n") - - -def main(): - parser = argparse.ArgumentParser(description="Simple RoboSystems query example") - parser.add_argument("--api-key", required=True, help="Your RoboSystems API key") - parser.add_argument("--graph-id", required=True, help="The graph ID to query") - parser.add_argument( - "--base-url", - default="http://localhost:8000", - help="API base URL (default: http://localhost:8000)", - ) - - args = parser.parse_args() - - try: - run_query_example(args.api_key, args.graph_id, args.base_url) - except Exception as e: - print(f"\nāŒ Error: {e}") - import traceback - - traceback.print_exc() - sys.exit(1) - - -if __name__ == "__main__": - main() diff --git a/examples/e2e_workflow.py b/examples/e2e_workflow.py deleted file mode 100644 index d3a10eb..0000000 --- a/examples/e2e_workflow.py +++ /dev/null @@ -1,445 +0,0 @@ -""" -End-to-end workflow demonstration using RoboSystems Python Client SDK. - -This script demonstrates the complete RoboSystems workflow: -1. User creation (or use existing API key) -2. Graph creation -3. Parquet file upload to staging tables -4. Data ingestion into graph -5. Graph querying - -Usage: - python examples/e2e_workflow.py --api-key # Use existing API key - python examples/e2e_workflow.py # Create new user - python examples/e2e_workflow.py --use-buffer # Use in-memory buffer instead of file - -Dependencies: - pip install robosystems-client[tables] -""" - -import argparse -import json -import secrets -import string -import sys -import time -from pathlib import Path -from typing import Optional - -import pandas as pd -import pyarrow as pa -import pyarrow.parquet as pq - -from robosystems_client import Client -from robosystems_client.extensions import ( - RoboSystemsExtensions, - RoboSystemsExtensionConfig, - UploadOptions, - IngestOptions, -) -from robosystems_client.api.auth.register_user import sync_detailed as register -from robosystems_client.api.auth.login_user import sync_detailed as login -from robosystems_client.api.user.create_user_api_key import ( - sync_detailed as create_api_key, -) -from robosystems_client.api.graphs.create_graph import sync_detailed as create_graph -from robosystems_client.models.register_request import RegisterRequest -from robosystems_client.models.login_request import LoginRequest -from robosystems_client.models.create_api_key_request import CreateAPIKeyRequest -from robosystems_client.models.create_graph_request import CreateGraphRequest -from robosystems_client.models.graph_metadata import GraphMetadata - - -def generate_secure_password(length: int = 16) -> str: - """Generate a cryptographically secure password.""" - chars_per_type = length // 4 - password = ( - "".join(secrets.choice(string.ascii_lowercase) for _ in range(chars_per_type)) - + "".join(secrets.choice(string.ascii_uppercase) for _ in range(chars_per_type)) - + "".join(secrets.choice(string.digits) for _ in range(chars_per_type)) - + "".join(secrets.choice("!@#$%^&*") for _ in range(chars_per_type)) - ) - password_list = list(password) - secrets.SystemRandom().shuffle(password_list) - return "".join(password_list) - - -def create_sample_dataframe(num_rows: int = 100) -> pd.DataFrame: - """Create a sample DataFrame with entity data.""" - return pd.DataFrame( - { - "identifier": [f"entity_{i:03d}" for i in range(1, num_rows + 1)], - "uri": [None] * num_rows, - "scheme": [None] * num_rows, - "cik": [None] * num_rows, - "ticker": [f"TKR{i}" if i % 10 == 0 else None for i in range(1, num_rows + 1)], - "exchange": [None] * num_rows, - "name": [f"Entity_{i}" for i in range(1, num_rows + 1)], - "legal_name": [None] * num_rows, - "industry": [None] * num_rows, - "entity_type": [None] * num_rows, - "sic": [None] * num_rows, - "sic_description": [None] * num_rows, - "category": [f"Category_{i % 5}" for i in range(1, num_rows + 1)], - "state_of_incorporation": [None] * num_rows, - "fiscal_year_end": [None] * num_rows, - "ein": [None] * num_rows, - "tax_id": [None] * num_rows, - "lei": [None] * num_rows, - "phone": [None] * num_rows, - "website": [None] * num_rows, - "status": [None] * num_rows, - "is_parent": pd.Series([None] * num_rows, dtype="boolean"), - "parent_entity_id": [None] * num_rows, - "created_at": pd.date_range("2025-01-01", periods=num_rows, freq="h").strftime( - "%Y-%m-%d %H:%M:%S" - ), - "updated_at": [None] * num_rows, - } - ) - - -def create_sample_parquet(output_path: Path, num_rows: int = 100): - """Create a sample Parquet file with entity data.""" - print(f"\nšŸ“„ Creating sample Parquet file with {num_rows} rows...") - - df = create_sample_dataframe(num_rows) - table = pa.Table.from_pandas(df) - pq.write_table(table, output_path) - - print(f"āœ… Created {output_path.name} ({output_path.stat().st_size:,} bytes)") - return df - - -def setup_user_and_api_key( - base_url: str, - name: Optional[str] = None, - email: Optional[str] = None, - password: Optional[str] = None, -) -> str: - """Create a new user and generate an API key.""" - client = Client(base_url=base_url) - - # Generate credentials if not provided - if not name or not email or not password: - timestamp = int(time.time()) - name = name or f"Demo User {timestamp}" - email = email or f"demo_{timestamp}@example.com" - password = password or generate_secure_password() - - print("\nšŸ“§ Auto-generated credentials:") - print(f" Name: {name}") - print(f" Email: {email}") - print(f" Password: {password}") - - # Create user - print("\nšŸ” Creating new user...") - user_create = RegisterRequest(name=name, email=email, password=password) - response = register(client=client, body=user_create) - if not response.parsed: - raise Exception("Failed to create user") - print(f"āœ… User created: {name} ({email})") - - # Login - print("\nšŸ”‘ Logging in...") - user_login = LoginRequest(email=email, password=password) - response = login(client=client, body=user_login) - if not response.parsed: - raise Exception("Failed to login") - token = response.parsed.token - print("āœ… Login successful") - - # Create API key - print("\nšŸ”‘ Creating API key...") - api_key_create = CreateAPIKeyRequest(name=f"E2E Demo Key - {name}") - response = create_api_key(client=client, token=token, body=api_key_create) - if not response.parsed: - raise Exception("Failed to create API key") - api_key = response.parsed.key - print(f"āœ… API key created: {api_key[:20]}...") - - return api_key - - -def create_demo_graph(base_url: str, api_key: str) -> str: - """Create a new graph for the demo.""" - client = Client(base_url=base_url, headers={"X-API-Key": api_key}) - - graph_name = f"demo_graph_{int(time.time())}" - print(f"\nšŸ“Š Creating graph: {graph_name}...") - - metadata = GraphMetadata( - graph_name=graph_name, - description=f"E2E workflow demo graph - {graph_name}", - ) - graph_create = CreateGraphRequest(metadata=metadata) - - response = create_graph(client=client, body=graph_create) - - if not response.parsed: - error_msg = f"Failed to create graph. Status: {response.status_code}" - if hasattr(response, "content"): - error_msg += f", Response: {response.content}" - raise Exception(error_msg) - - # Handle async graph creation - # Response can be either a dict or an object - if isinstance(response.parsed, dict): - graph_id = response.parsed.get("graph_id") - operation_id = response.parsed.get("operation_id") - else: - graph_id = getattr(response.parsed, "graph_id", None) - operation_id = getattr(response.parsed, "operation_id", None) - - if graph_id: - print(f"āœ… Graph created: {graph_id}") - return graph_id - elif operation_id: - print(f"āš ļø Graph creation queued. Operation ID: {operation_id}") - print(" Polling for completion...") - - # Use simple polling for fast operations like graph creation - # (SSE connection establishment takes longer than graph creation itself ~0.3s) - from robosystems_client.api.operations.get_operation_status import ( - sync_detailed as get_status, - ) - - max_attempts = 20 - for attempt in range(max_attempts): - time.sleep(0.5) # Poll every 500ms - - status_response = get_status(operation_id=operation_id, client=client) - - if status_response.parsed: - # Response is stored in additional_properties dict - status = status_response.parsed["status"] - print(f" Status: {status}") - - if status == "completed": - result = status_response.parsed["result"] - if isinstance(result, dict): - graph_id = result.get("graph_id") - else: - graph_id = getattr(result, "graph_id", None) - - if graph_id: - print(f"āœ… Graph created: {graph_id}") - return graph_id - else: - raise Exception("Operation completed but no graph_id in result") - - elif status == "failed": - error = ( - status_response.parsed.get("error") - if isinstance(status_response.parsed, dict) - else status_response.parsed["error"] - if "error" in status_response.parsed - else "Unknown error" - ) - raise Exception(f"Graph creation failed: {error}") - - raise Exception(f"Graph creation timed out after {max_attempts * 0.5}s") - else: - raise Exception( - f"Unexpected response from graph creation. Response: {response.parsed}" - ) - - -def run_workflow( - base_url: str = "http://localhost:8000", - name: Optional[str] = None, - email: Optional[str] = None, - password: Optional[str] = None, - api_key: Optional[str] = None, - graph_id: Optional[str] = None, # Reuse existing graph - use_buffer: bool = False, # Use buffer instead of file path -): - """Run the complete E2E workflow.""" - print("\n" + "=" * 60) - print("šŸ¤– RoboSystems E2E Workflow Demo (SDK Version)") - print("=" * 60) - - # Step 1: Get or create API key - if api_key: - print("\nāœ… Using provided API key") - else: - api_key = setup_user_and_api_key(base_url, name, email, password) - - # Step 2: Create or reuse graph - if graph_id: - print(f"\nāœ… Using existing graph: {graph_id}") - else: - graph_id = create_demo_graph(base_url, api_key) - - # Step 3: Initialize extensions with API key - config = RoboSystemsExtensionConfig(base_url=base_url, headers={"X-API-Key": api_key}) - extensions = RoboSystemsExtensions(config) - - # Step 4: Create and upload sample data - table_name = "Entity" - - if use_buffer: - # Buffer-based approach (no disk I/O required) - print("\nšŸ“„ Creating sample Parquet data in-memory (50 rows)...") - - df = create_sample_dataframe(num_rows=50) - - # Convert to Parquet in-memory - from io import BytesIO - - buffer = BytesIO() - table = pa.Table.from_pandas(df) - pq.write_table(table, buffer) - buffer.seek(0) - - print(f"āœ… Created in-memory buffer ({len(buffer.getvalue()):,} bytes)") - - # Step 5: Upload from buffer - upload_options = UploadOptions( - on_progress=lambda msg: print(f" {msg}"), - fix_localstack_url=True, - file_name="sample_data.parquet", # Override file name for buffer - ) - - upload_result = extensions.tables.upload_parquet_file( - graph_id, table_name, buffer, upload_options - ) - else: - # File-based approach (traditional) - temp_dir = Path("/tmp/robosystems_demo") - temp_dir.mkdir(exist_ok=True) - parquet_file = temp_dir / "sample_data.parquet" - - _df = create_sample_parquet(parquet_file, num_rows=50) - - # Step 5: Upload parquet file using the new TableIngestClient - upload_options = UploadOptions( - on_progress=lambda msg: print(f" {msg}"), - fix_localstack_url=True, - ) - - upload_result = extensions.tables.upload_parquet_file( - graph_id, table_name, parquet_file, upload_options - ) - - if not upload_result.success: - print(f"āŒ Upload failed: {upload_result.error}") - sys.exit(1) - - # Step 6: List staging tables - print("\nšŸ“‹ Listing staging tables...") - tables = extensions.tables.list_staging_tables(graph_id) - - if tables: - print("\nStaging Tables:") - print( - f"{'Table Name':<20} {'Row Count':<12} {'File Count':<12} {'Size (bytes)':<15}" - ) - print("-" * 60) - for tbl in tables: - print( - f"{tbl.table_name:<20} {tbl.row_count:<12} {tbl.file_count:<12} {tbl.total_size_bytes:>14,}" - ) - else: - print("No tables found") - - # Step 7: Ingest tables - print("\nāš™ļø Ingesting all tables to graph...") - ingest_options = IngestOptions( - ignore_errors=True, rebuild=False, on_progress=lambda msg: print(f" {msg}") - ) - - ingest_result = extensions.tables.ingest_all_tables(graph_id, ingest_options) - - if not ingest_result.get("success"): - print(f"āŒ Ingestion failed: {ingest_result.get('error')}") - sys.exit(1) - - # Step 8: Query the graph - queries = [ - "MATCH (n:Entity) RETURN count(n) AS total_nodes", - "MATCH (n:Entity) RETURN n.identifier, n.name, n.ticker, n.category LIMIT 5", - "MATCH (n:Entity) WHERE n.ticker IS NOT NULL RETURN n.identifier, n.name, n.ticker ORDER BY n.identifier DESC LIMIT 10", - ] - - for query in queries: - print(f"\nšŸ” Executing query: {query}") - result = extensions.query.query(graph_id, query) - - if hasattr(result, "data") and result.data: - print(f"āœ… Query returned {len(result.data)} records") - if len(result.data) <= 10: - print(json.dumps(result.data, indent=2)) - else: - print(f"Showing first 10 of {len(result.data)} records:") - print(json.dumps(result.data[:10], indent=2)) - else: - print("āŒ Query returned no results or error") - - # Cleanup - extensions.close() - if not use_buffer: - parquet_file.unlink() - - print("\n" + "=" * 60) - print("āœ… E2E Workflow Complete!") - print("=" * 60) - print(f"\nšŸ“Š Graph ID: {graph_id}") - print(f"šŸ”‘ API Key: {api_key}") - if use_buffer: - print("šŸ’” Note: Used in-memory buffer (no /tmp file created)") - print("\nšŸ’” You can continue querying this graph using the API key\n") - - -def main(): - parser = argparse.ArgumentParser( - description="RoboSystems E2E workflow demonstration using SDK" - ) - parser.add_argument( - "--api-key", - help="Use existing API key (skip user creation)", - ) - parser.add_argument( - "--name", - help="Name for new user (auto-generated if not provided)", - ) - parser.add_argument( - "--email", - help="Email for new user (auto-generated if not provided)", - ) - parser.add_argument( - "--password", - help="Password for new user (auto-generated if not provided)", - ) - parser.add_argument( - "--base-url", - default="http://localhost:8000", - help="API base URL (default: http://localhost:8000)", - ) - parser.add_argument( - "--use-buffer", - action="store_true", - help="Use in-memory buffer instead of file path (demonstrates buffer upload)", - ) - - args = parser.parse_args() - - try: - run_workflow( - base_url=args.base_url, - name=args.name, - email=args.email, - password=args.password, - api_key=args.api_key, - use_buffer=args.use_buffer, - ) - except Exception as e: - print(f"\nāŒ Error: {e}") - import traceback - - traceback.print_exc() - sys.exit(1) - - -if __name__ == "__main__": - main() diff --git a/justfile b/justfile index 2ef728f..a266b95 100644 --- a/justfile +++ b/justfile @@ -28,6 +28,7 @@ test-all: @just test @just lint @just format + @just typecheck # Run linting lint: @@ -40,7 +41,7 @@ format: # Run type checking typecheck: - uv run pyright + uv run basedpyright # Generate SDK from localhost API generate-sdk url="http://localhost:8000/openapi.json": diff --git a/pyproject.toml b/pyproject.toml index 6de95ad..c582abb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -41,7 +41,7 @@ dependencies = [ [project.optional-dependencies] dev = [ "openapi-python-client>=0.21.8", - "pyright>=1.1.402", + "basedpyright>=1.21.0", "pytest-asyncio>=0.26.0", "pytest>=8.3.5", "ruff>=0.12", @@ -73,8 +73,8 @@ build-backend = "hatchling.build" packages = ["robosystems_client"] [tool.basedpyright] -include = [] -exclude = ["**/*"] +include = ["robosystems_client/extensions"] +exclude = ["robosystems_client/extensions/tests"] extraPaths = ["."] pythonVersion = "3.12" venvPath = "." @@ -85,3 +85,5 @@ reportArgumentType = "none" reportGeneralTypeIssues = "none" reportOptionalMemberAccess = "none" reportReturnType = "none" +reportInvalidTypeForm = "none" +reportMissingImports = "none" diff --git a/robosystems_client/api/agent/auto_select_agent.py b/robosystems_client/api/agent/auto_select_agent.py index 4aacaf4..9179630 100644 --- a/robosystems_client/api/agent/auto_select_agent.py +++ b/robosystems_client/api/agent/auto_select_agent.py @@ -9,35 +9,19 @@ from ...models.agent_response import AgentResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: AgentRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/agent", - "params": params, } _kwargs["json"] = body.to_dict() @@ -100,8 +84,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Auto-select agent for query @@ -117,8 +99,6 @@ def sync_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -132,8 +112,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -148,8 +126,6 @@ def sync( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Auto-select agent for query @@ -165,8 +141,6 @@ def sync( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -181,8 +155,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -191,8 +163,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Auto-select agent for query @@ -208,8 +178,6 @@ async def asyncio_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -223,8 +191,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -237,8 +203,6 @@ async def asyncio( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Auto-select agent for query @@ -254,8 +218,6 @@ async def asyncio( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -271,7 +233,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/agent/batch_process_queries.py b/robosystems_client/api/agent/batch_process_queries.py index b705ad4..91e82b7 100644 --- a/robosystems_client/api/agent/batch_process_queries.py +++ b/robosystems_client/api/agent/batch_process_queries.py @@ -8,35 +8,19 @@ from ...models.batch_agent_request import BatchAgentRequest from ...models.batch_agent_response import BatchAgentResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: BatchAgentRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/agent/batch", - "params": params, } _kwargs["json"] = body.to_dict() @@ -94,8 +78,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: BatchAgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, BatchAgentResponse, HTTPValidationError]]: """Batch process multiple queries @@ -116,8 +98,6 @@ def sync_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BatchAgentRequest): Request for batch processing multiple queries. Raises: @@ -131,8 +111,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -147,8 +125,6 @@ def sync( *, client: AuthenticatedClient, body: BatchAgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, BatchAgentResponse, HTTPValidationError]]: """Batch process multiple queries @@ -169,8 +145,6 @@ def sync( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BatchAgentRequest): Request for batch processing multiple queries. Raises: @@ -185,8 +159,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -195,8 +167,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: BatchAgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, BatchAgentResponse, HTTPValidationError]]: """Batch process multiple queries @@ -217,8 +187,6 @@ async def asyncio_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BatchAgentRequest): Request for batch processing multiple queries. Raises: @@ -232,8 +200,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -246,8 +212,6 @@ async def asyncio( *, client: AuthenticatedClient, body: BatchAgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, BatchAgentResponse, HTTPValidationError]]: """Batch process multiple queries @@ -268,8 +232,6 @@ async def asyncio( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BatchAgentRequest): Request for batch processing multiple queries. Raises: @@ -285,7 +247,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/agent/execute_specific_agent.py b/robosystems_client/api/agent/execute_specific_agent.py index cee6207..e91f6a6 100644 --- a/robosystems_client/api/agent/execute_specific_agent.py +++ b/robosystems_client/api/agent/execute_specific_agent.py @@ -9,7 +9,7 @@ from ...models.agent_response import AgentResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -17,28 +17,12 @@ def _get_kwargs( agent_type: str, *, body: AgentRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/agent/{agent_type}", - "params": params, } _kwargs["json"] = body.to_dict() @@ -106,8 +90,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Execute specific agent @@ -123,8 +105,6 @@ def sync_detailed( Args: graph_id (str): agent_type (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -139,8 +119,6 @@ def sync_detailed( graph_id=graph_id, agent_type=agent_type, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -156,8 +134,6 @@ def sync( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Execute specific agent @@ -173,8 +149,6 @@ def sync( Args: graph_id (str): agent_type (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -190,8 +164,6 @@ def sync( agent_type=agent_type, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -201,8 +173,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Execute specific agent @@ -218,8 +188,6 @@ async def asyncio_detailed( Args: graph_id (str): agent_type (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -234,8 +202,6 @@ async def asyncio_detailed( graph_id=graph_id, agent_type=agent_type, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -249,8 +215,6 @@ async def asyncio( *, client: AuthenticatedClient, body: AgentRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]: """Execute specific agent @@ -266,8 +230,6 @@ async def asyncio( Args: graph_id (str): agent_type (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (AgentRequest): Request model for agent interactions. Raises: @@ -284,7 +246,5 @@ async def asyncio( agent_type=agent_type, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/agent/get_agent_metadata.py b/robosystems_client/api/agent/get_agent_metadata.py index aa2ecac..abb33b7 100644 --- a/robosystems_client/api/agent/get_agent_metadata.py +++ b/robosystems_client/api/agent/get_agent_metadata.py @@ -7,38 +7,18 @@ from ...client import AuthenticatedClient, Client from ...models.agent_metadata_response import AgentMetadataResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, agent_type: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/agent/{agent_type}/metadata", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -81,8 +61,6 @@ def sync_detailed( agent_type: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentMetadataResponse, Any, HTTPValidationError]]: """Get agent metadata @@ -99,10 +77,8 @@ def sync_detailed( Use this to understand agent capabilities before execution. Args: - graph_id (str): Graph database identifier + graph_id (str): agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -115,8 +91,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, agent_type=agent_type, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -131,8 +105,6 @@ def sync( agent_type: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentMetadataResponse, Any, HTTPValidationError]]: """Get agent metadata @@ -149,10 +121,8 @@ def sync( Use this to understand agent capabilities before execution. Args: - graph_id (str): Graph database identifier + graph_id (str): agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -166,8 +136,6 @@ def sync( graph_id=graph_id, agent_type=agent_type, client=client, - token=token, - authorization=authorization, ).parsed @@ -176,8 +144,6 @@ async def asyncio_detailed( agent_type: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentMetadataResponse, Any, HTTPValidationError]]: """Get agent metadata @@ -194,10 +160,8 @@ async def asyncio_detailed( Use this to understand agent capabilities before execution. Args: - graph_id (str): Graph database identifier + graph_id (str): agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -210,8 +174,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, agent_type=agent_type, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -224,8 +186,6 @@ async def asyncio( agent_type: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentMetadataResponse, Any, HTTPValidationError]]: """Get agent metadata @@ -242,10 +202,8 @@ async def asyncio( Use this to understand agent capabilities before execution. Args: - graph_id (str): Graph database identifier + graph_id (str): agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -260,7 +218,5 @@ async def asyncio( graph_id=graph_id, agent_type=agent_type, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/agent/list_agents.py b/robosystems_client/api/agent/list_agents.py index 7e0f378..765ce27 100644 --- a/robosystems_client/api/agent/list_agents.py +++ b/robosystems_client/api/agent/list_agents.py @@ -14,13 +14,7 @@ def _get_kwargs( graph_id: str, *, capability: Union[None, Unset, str] = UNSET, - 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_capability: Union[None, Unset, str] @@ -30,13 +24,6 @@ def _get_kwargs( json_capability = capability params["capability"] = json_capability - 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] = { @@ -45,7 +32,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -88,8 +74,6 @@ def sync_detailed( *, client: AuthenticatedClient, capability: Union[None, Unset, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentListResponse, Any, HTTPValidationError]]: """List available agents @@ -104,11 +88,9 @@ def sync_detailed( Use the optional `capability` filter to find agents with specific capabilities. Args: - graph_id (str): Graph database identifier + graph_id (str): capability (Union[None, Unset, str]): Filter by capability (e.g., 'financial_analysis', 'rag_search') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -121,8 +103,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, capability=capability, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -137,8 +117,6 @@ def sync( *, client: AuthenticatedClient, capability: Union[None, Unset, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentListResponse, Any, HTTPValidationError]]: """List available agents @@ -153,11 +131,9 @@ def sync( Use the optional `capability` filter to find agents with specific capabilities. Args: - graph_id (str): Graph database identifier + graph_id (str): capability (Union[None, Unset, str]): Filter by capability (e.g., 'financial_analysis', 'rag_search') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -171,8 +147,6 @@ def sync( graph_id=graph_id, client=client, capability=capability, - token=token, - authorization=authorization, ).parsed @@ -181,8 +155,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, capability: Union[None, Unset, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentListResponse, Any, HTTPValidationError]]: """List available agents @@ -197,11 +169,9 @@ async def asyncio_detailed( Use the optional `capability` filter to find agents with specific capabilities. Args: - graph_id (str): Graph database identifier + graph_id (str): capability (Union[None, Unset, str]): Filter by capability (e.g., 'financial_analysis', 'rag_search') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -214,8 +184,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, capability=capability, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -228,8 +196,6 @@ async def asyncio( *, client: AuthenticatedClient, capability: Union[None, Unset, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentListResponse, Any, HTTPValidationError]]: """List available agents @@ -244,11 +210,9 @@ async def asyncio( Use the optional `capability` filter to find agents with specific capabilities. Args: - graph_id (str): Graph database identifier + graph_id (str): capability (Union[None, Unset, str]): Filter by capability (e.g., 'financial_analysis', 'rag_search') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -263,7 +227,5 @@ async def asyncio( graph_id=graph_id, client=client, capability=capability, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/agent/recommend_agent.py b/robosystems_client/api/agent/recommend_agent.py index 3e1b99b..79e26a9 100644 --- a/robosystems_client/api/agent/recommend_agent.py +++ b/robosystems_client/api/agent/recommend_agent.py @@ -8,35 +8,19 @@ from ...models.agent_recommendation_request import AgentRecommendationRequest from ...models.agent_recommendation_response import AgentRecommendationResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: AgentRecommendationRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/agent/recommend", - "params": params, } _kwargs["json"] = body.to_dict() @@ -86,8 +70,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: AgentRecommendationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentRecommendationResponse, Any, HTTPValidationError]]: """Get agent recommendations @@ -107,9 +89,7 @@ def sync_detailed( Returns top agents ranked by confidence with explanations. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (AgentRecommendationRequest): Request for agent recommendations. Raises: @@ -123,8 +103,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -139,8 +117,6 @@ def sync( *, client: AuthenticatedClient, body: AgentRecommendationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentRecommendationResponse, Any, HTTPValidationError]]: """Get agent recommendations @@ -160,9 +136,7 @@ def sync( Returns top agents ranked by confidence with explanations. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (AgentRecommendationRequest): Request for agent recommendations. Raises: @@ -177,8 +151,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -187,8 +159,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: AgentRecommendationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[AgentRecommendationResponse, Any, HTTPValidationError]]: """Get agent recommendations @@ -208,9 +178,7 @@ async def asyncio_detailed( Returns top agents ranked by confidence with explanations. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (AgentRecommendationRequest): Request for agent recommendations. Raises: @@ -224,8 +192,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -238,8 +204,6 @@ async def asyncio( *, client: AuthenticatedClient, body: AgentRecommendationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[AgentRecommendationResponse, Any, HTTPValidationError]]: """Get agent recommendations @@ -259,9 +223,7 @@ async def asyncio( Returns top agents ranked by confidence with explanations. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (AgentRecommendationRequest): Request for agent recommendations. Raises: @@ -277,7 +239,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/auth/generate_sso_token.py b/robosystems_client/api/auth/generate_sso_token.py index 4ea201c..0a1325f 100644 --- a/robosystems_client/api/auth/generate_sso_token.py +++ b/robosystems_client/api/auth/generate_sso_token.py @@ -13,13 +13,8 @@ def _get_kwargs( *, - authorization: Union[None, Unset, str] = UNSET, auth_token: Union[None, Unset, str] = UNSET, ) -> dict[str, Any]: - headers: dict[str, Any] = {} - if not isinstance(authorization, Unset): - headers["authorization"] = authorization - cookies = {} if auth_token is not UNSET: cookies["auth-token"] = auth_token @@ -30,7 +25,6 @@ def _get_kwargs( "cookies": cookies, } - _kwargs["headers"] = headers return _kwargs @@ -72,7 +66,6 @@ def _build_response( def sync_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, auth_token: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SSOTokenResponse]]: """Generate SSO Token @@ -80,7 +73,6 @@ def sync_detailed( Generate a temporary SSO token for cross-app authentication. Args: - authorization (Union[None, Unset, str]): auth_token (Union[None, Unset, str]): Raises: @@ -92,7 +84,6 @@ def sync_detailed( """ kwargs = _get_kwargs( - authorization=authorization, auth_token=auth_token, ) @@ -106,7 +97,6 @@ def sync_detailed( def sync( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, auth_token: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SSOTokenResponse]]: """Generate SSO Token @@ -114,7 +104,6 @@ def sync( Generate a temporary SSO token for cross-app authentication. Args: - authorization (Union[None, Unset, str]): auth_token (Union[None, Unset, str]): Raises: @@ -127,7 +116,6 @@ def sync( return sync_detailed( client=client, - authorization=authorization, auth_token=auth_token, ).parsed @@ -135,7 +123,6 @@ def sync( async def asyncio_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, auth_token: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SSOTokenResponse]]: """Generate SSO Token @@ -143,7 +130,6 @@ async def asyncio_detailed( Generate a temporary SSO token for cross-app authentication. Args: - authorization (Union[None, Unset, str]): auth_token (Union[None, Unset, str]): Raises: @@ -155,7 +141,6 @@ async def asyncio_detailed( """ kwargs = _get_kwargs( - authorization=authorization, auth_token=auth_token, ) @@ -167,7 +152,6 @@ async def asyncio_detailed( async def asyncio( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, auth_token: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SSOTokenResponse]]: """Generate SSO Token @@ -175,7 +159,6 @@ async def asyncio( Generate a temporary SSO token for cross-app authentication. Args: - authorization (Union[None, Unset, str]): auth_token (Union[None, Unset, str]): Raises: @@ -189,7 +172,6 @@ async def asyncio( return ( await asyncio_detailed( client=client, - authorization=authorization, auth_token=auth_token, ) ).parsed diff --git a/robosystems_client/api/auth/get_current_auth_user.py b/robosystems_client/api/auth/get_current_auth_user.py index 1108307..d210cd8 100644 --- a/robosystems_client/api/auth/get_current_auth_user.py +++ b/robosystems_client/api/auth/get_current_auth_user.py @@ -9,34 +9,21 @@ from ...models.get_current_auth_user_response_getcurrentauthuser import ( GetCurrentAuthUserResponseGetcurrentauthuser, ) -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - authorization: Union[None, Unset, str] = UNSET, -) -> dict[str, Any]: - headers: dict[str, Any] = {} - if not isinstance(authorization, Unset): - headers["authorization"] = authorization - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/auth/me", } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Optional[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: if response.status_code == 200: response_200 = GetCurrentAuthUserResponseGetcurrentauthuser.from_dict( response.json() @@ -49,11 +36,6 @@ def _parse_response( return response_401 - 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: @@ -62,11 +44,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -78,30 +56,20 @@ def _build_response( def sync_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: """Get Current User Get the currently authenticated user. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError]] + Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -113,60 +81,41 @@ def sync_detailed( def sync( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Optional[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: """Get Current User Get the currently authenticated user. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError] + Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser] """ return sync_detailed( client=client, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: """Get Current User Get the currently authenticated user. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError]] + Response[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -176,30 +125,21 @@ async def asyncio_detailed( async def asyncio( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError - ] -]: +) -> Optional[Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser]]: """Get Current User Get the currently authenticated user. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser, HTTPValidationError] + Union[ErrorResponse, GetCurrentAuthUserResponseGetcurrentauthuser] """ return ( await asyncio_detailed( client=client, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/auth/logout_user.py b/robosystems_client/api/auth/logout_user.py index dd8ffc8..ee6a5de 100644 --- a/robosystems_client/api/auth/logout_user.py +++ b/robosystems_client/api/auth/logout_user.py @@ -5,41 +5,27 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.logout_user_response_logoutuser import LogoutUserResponseLogoutuser -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - authorization: Union[None, Unset, str] = UNSET, -) -> dict[str, Any]: - headers: dict[str, Any] = {} - if not isinstance(authorization, Unset): - headers["authorization"] = authorization - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "post", "url": "/v1/auth/logout", } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Optional[LogoutUserResponseLogoutuser]: if response.status_code == 200: response_200 = LogoutUserResponseLogoutuser.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: @@ -48,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Response[LogoutUserResponseLogoutuser]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -60,26 +46,20 @@ def _build_response( def sync_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Response[LogoutUserResponseLogoutuser]: """User Logout Logout user and invalidate session. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, LogoutUserResponseLogoutuser]] + Response[LogoutUserResponseLogoutuser] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -91,52 +71,41 @@ def sync_detailed( def sync( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Optional[LogoutUserResponseLogoutuser]: """User Logout Logout user and invalidate session. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, LogoutUserResponseLogoutuser] + LogoutUserResponseLogoutuser """ return sync_detailed( client=client, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Response[LogoutUserResponseLogoutuser]: """User Logout Logout user and invalidate session. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, LogoutUserResponseLogoutuser]] + Response[LogoutUserResponseLogoutuser] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -146,26 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, LogoutUserResponseLogoutuser]]: +) -> Optional[LogoutUserResponseLogoutuser]: """User Logout Logout user and invalidate session. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, LogoutUserResponseLogoutuser] + LogoutUserResponseLogoutuser """ return ( await asyncio_detailed( client=client, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/auth/refresh_auth_session.py b/robosystems_client/api/auth/refresh_auth_session.py index d552b0e..09bc4cd 100644 --- a/robosystems_client/api/auth/refresh_auth_session.py +++ b/robosystems_client/api/auth/refresh_auth_session.py @@ -7,30 +7,21 @@ from ...client import AuthenticatedClient, Client from ...models.auth_response import AuthResponse from ...models.error_response import ErrorResponse -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - authorization: Union[None, Unset, str] = UNSET, -) -> dict[str, Any]: - headers: dict[str, Any] = {} - if not isinstance(authorization, Unset): - headers["authorization"] = authorization - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "post", "url": "/v1/auth/refresh", } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Optional[Union[AuthResponse, ErrorResponse]]: if response.status_code == 200: response_200 = AuthResponse.from_dict(response.json()) @@ -41,11 +32,6 @@ def _parse_response( return response_401 - 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: @@ -54,7 +40,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Response[Union[AuthResponse, ErrorResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -66,26 +52,20 @@ def _build_response( def sync_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Response[Union[AuthResponse, ErrorResponse]]: """Refresh Session Refresh authentication session with a new JWT token. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]] + Response[Union[AuthResponse, ErrorResponse]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -97,52 +77,41 @@ def sync_detailed( def sync( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Optional[Union[AuthResponse, ErrorResponse]]: """Refresh Session Refresh authentication session with a new JWT token. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[AuthResponse, ErrorResponse, HTTPValidationError] + Union[AuthResponse, ErrorResponse] """ return sync_detailed( client=client, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Response[Union[AuthResponse, ErrorResponse]]: """Refresh Session Refresh authentication session with a new JWT token. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]] + Response[Union[AuthResponse, ErrorResponse]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -152,26 +121,21 @@ async def asyncio_detailed( async def asyncio( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]: +) -> Optional[Union[AuthResponse, ErrorResponse]]: """Refresh Session Refresh authentication session with a new JWT token. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[AuthResponse, ErrorResponse, HTTPValidationError] + Union[AuthResponse, ErrorResponse] """ return ( await asyncio_detailed( client=client, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/auth/resend_verification_email.py b/robosystems_client/api/auth/resend_verification_email.py index c868610..5f3d199 100644 --- a/robosystems_client/api/auth/resend_verification_email.py +++ b/robosystems_client/api/auth/resend_verification_email.py @@ -6,38 +6,25 @@ from ... import errors from ...client import AuthenticatedClient, Client from ...models.error_response import ErrorResponse -from ...models.http_validation_error import HTTPValidationError from ...models.resend_verification_email_response_resendverificationemail import ( ResendVerificationEmailResponseResendverificationemail, ) -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - authorization: Union[None, Unset, str] = UNSET, -) -> dict[str, Any]: - headers: dict[str, Any] = {} - if not isinstance(authorization, Unset): - headers["authorization"] = authorization - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "post", "url": "/v1/auth/email/resend", } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response ) -> Optional[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: if response.status_code == 200: response_200 = ResendVerificationEmailResponseResendverificationemail.from_dict( @@ -51,11 +38,6 @@ def _parse_response( return response_400 - if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) - - return response_422 - if response.status_code == 429: response_429 = ErrorResponse.from_dict(response.json()) @@ -75,11 +57,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response ) -> Response[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: return Response( status_code=HTTPStatus(response.status_code), @@ -92,32 +70,22 @@ def _build_response( def sync_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: """Resend Email Verification Resend verification email to the authenticated user. Rate limited to 3 per hour. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, HTTPValidationError, ResendVerificationEmailResponseResendverificationemail]] + Response[Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -129,64 +97,45 @@ def sync_detailed( def sync( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: """Resend Email Verification Resend verification email to the authenticated user. Rate limited to 3 per hour. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, HTTPValidationError, ResendVerificationEmailResponseResendverificationemail] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] """ return sync_detailed( client=client, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: """Resend Email Verification Resend verification email to the authenticated user. Rate limited to 3 per hour. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, HTTPValidationError, ResendVerificationEmailResponseResendverificationemail]] + Response[Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail]] """ - kwargs = _get_kwargs( - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -196,32 +145,23 @@ async def asyncio_detailed( async def asyncio( *, client: Union[AuthenticatedClient, Client], - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ - Union[ - ErrorResponse, - HTTPValidationError, - ResendVerificationEmailResponseResendverificationemail, - ] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] ]: """Resend Email Verification Resend verification email to the authenticated user. Rate limited to 3 per hour. - Args: - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, HTTPValidationError, ResendVerificationEmailResponseResendverificationemail] + Union[ErrorResponse, ResendVerificationEmailResponseResendverificationemail] """ return ( await asyncio_detailed( client=client, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/backup/create_backup.py b/robosystems_client/api/backup/create_backup.py index 0beadcd..54844b7 100644 --- a/robosystems_client/api/backup/create_backup.py +++ b/robosystems_client/api/backup/create_backup.py @@ -8,35 +8,19 @@ from ...models.backup_create_request import BackupCreateRequest from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: BackupCreateRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/backups", - "params": params, } _kwargs["json"] = body.to_dict() @@ -101,8 +85,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: BackupCreateRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Backup @@ -150,9 +132,7 @@ def sync_detailed( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BackupCreateRequest): Request model for creating a backup. Raises: @@ -166,8 +146,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -182,8 +160,6 @@ def sync( *, client: AuthenticatedClient, body: BackupCreateRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Backup @@ -231,9 +207,7 @@ def sync( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BackupCreateRequest): Request model for creating a backup. Raises: @@ -248,8 +222,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -258,8 +230,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: BackupCreateRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Backup @@ -307,9 +277,7 @@ async def asyncio_detailed( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BackupCreateRequest): Request model for creating a backup. Raises: @@ -323,8 +291,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -337,8 +303,6 @@ async def asyncio( *, client: AuthenticatedClient, body: BackupCreateRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Backup @@ -386,9 +350,7 @@ async def asyncio( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BackupCreateRequest): Request model for creating a backup. Raises: @@ -404,7 +366,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/backup/get_backup_download_url.py b/robosystems_client/api/backup/get_backup_download_url.py index 1a9806b..f084b42 100644 --- a/robosystems_client/api/backup/get_backup_download_url.py +++ b/robosystems_client/api/backup/get_backup_download_url.py @@ -17,24 +17,11 @@ def _get_kwargs( backup_id: str, *, expires_in: Union[Unset, int] = 3600, - 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] = {} params["expires_in"] = expires_in - 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] = { @@ -43,7 +30,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -101,8 +87,6 @@ def sync_detailed( *, client: AuthenticatedClient, expires_in: Union[Unset, int] = 3600, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError] ]: @@ -111,11 +95,9 @@ def sync_detailed( Generate a temporary download URL for a backup (unencrypted, compressed .kuzu files only) Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -129,8 +111,6 @@ def sync_detailed( graph_id=graph_id, backup_id=backup_id, expires_in=expires_in, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -146,8 +126,6 @@ def sync( *, client: AuthenticatedClient, expires_in: Union[Unset, int] = 3600, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError] ]: @@ -156,11 +134,9 @@ def sync( Generate a temporary download URL for a backup (unencrypted, compressed .kuzu files only) Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -175,8 +151,6 @@ def sync( backup_id=backup_id, client=client, expires_in=expires_in, - token=token, - authorization=authorization, ).parsed @@ -186,8 +160,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, expires_in: Union[Unset, int] = 3600, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError] ]: @@ -196,11 +168,9 @@ async def asyncio_detailed( Generate a temporary download URL for a backup (unencrypted, compressed .kuzu files only) Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -214,8 +184,6 @@ async def asyncio_detailed( graph_id=graph_id, backup_id=backup_id, expires_in=expires_in, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -229,8 +197,6 @@ async def asyncio( *, client: AuthenticatedClient, expires_in: Union[Unset, int] = 3600, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError] ]: @@ -239,11 +205,9 @@ async def asyncio( Generate a temporary download URL for a backup (unencrypted, compressed .kuzu files only) Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -259,7 +223,5 @@ async def asyncio( backup_id=backup_id, client=client, expires_in=expires_in, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/backup/get_backup_stats.py b/robosystems_client/api/backup/get_backup_stats.py index 638f803..1d527bd 100644 --- a/robosystems_client/api/backup/get_backup_stats.py +++ b/robosystems_client/api/backup/get_backup_stats.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.backup_stats_response import BackupStatsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/backups/stats", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -75,17 +55,13 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]: """Get backup statistics Get comprehensive backup statistics for the specified graph database Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -97,8 +73,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -112,17 +86,13 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]: """Get backup statistics Get comprehensive backup statistics for the specified graph database Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -135,8 +105,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -144,17 +112,13 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]: """Get backup statistics Get comprehensive backup statistics for the specified graph database Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -166,8 +130,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -179,17 +141,13 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]: """Get backup statistics Get comprehensive backup statistics for the specified graph database Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -203,7 +161,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/backup/list_backups.py b/robosystems_client/api/backup/list_backups.py index c1f9529..ad53cfd 100644 --- a/robosystems_client/api/backup/list_backups.py +++ b/robosystems_client/api/backup/list_backups.py @@ -15,26 +15,13 @@ def _get_kwargs( *, limit: Union[Unset, int] = 50, offset: Union[Unset, int] = 0, - 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] = {} params["limit"] = limit params["offset"] = offset - 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] = { @@ -43,7 +30,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -83,19 +69,15 @@ def sync_detailed( client: AuthenticatedClient, limit: Union[Unset, int] = 50, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[BackupListResponse, HTTPValidationError]]: """List graph database backups List all backups for the specified graph database Args: - graph_id (str): Graph database identifier + graph_id (str): limit (Union[Unset, int]): Maximum number of backups to return Default: 50. offset (Union[Unset, int]): Number of backups to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -109,8 +91,6 @@ def sync_detailed( graph_id=graph_id, limit=limit, offset=offset, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -126,19 +106,15 @@ def sync( client: AuthenticatedClient, limit: Union[Unset, int] = 50, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[BackupListResponse, HTTPValidationError]]: """List graph database backups List all backups for the specified graph database Args: - graph_id (str): Graph database identifier + graph_id (str): limit (Union[Unset, int]): Maximum number of backups to return Default: 50. offset (Union[Unset, int]): Number of backups to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -153,8 +129,6 @@ def sync( client=client, limit=limit, offset=offset, - token=token, - authorization=authorization, ).parsed @@ -164,19 +138,15 @@ async def asyncio_detailed( client: AuthenticatedClient, limit: Union[Unset, int] = 50, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[BackupListResponse, HTTPValidationError]]: """List graph database backups List all backups for the specified graph database Args: - graph_id (str): Graph database identifier + graph_id (str): limit (Union[Unset, int]): Maximum number of backups to return Default: 50. offset (Union[Unset, int]): Number of backups to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -190,8 +160,6 @@ async def asyncio_detailed( graph_id=graph_id, limit=limit, offset=offset, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -205,19 +173,15 @@ async def asyncio( client: AuthenticatedClient, limit: Union[Unset, int] = 50, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[BackupListResponse, HTTPValidationError]]: """List graph database backups List all backups for the specified graph database Args: - graph_id (str): Graph database identifier + graph_id (str): limit (Union[Unset, int]): Maximum number of backups to return Default: 50. offset (Union[Unset, int]): Number of backups to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -233,7 +197,5 @@ async def asyncio( client=client, limit=limit, offset=offset, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/backup/restore_backup.py b/robosystems_client/api/backup/restore_backup.py index a41590d..bd38adc 100644 --- a/robosystems_client/api/backup/restore_backup.py +++ b/robosystems_client/api/backup/restore_backup.py @@ -8,7 +8,7 @@ from ...models.backup_restore_request import BackupRestoreRequest from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -16,28 +16,12 @@ def _get_kwargs( backup_id: str, *, body: BackupRestoreRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/backups/{backup_id}/restore", - "params": params, } _kwargs["json"] = body.to_dict() @@ -103,8 +87,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: BackupRestoreRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Restore Encrypted Backup @@ -152,10 +134,8 @@ def sync_detailed( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BackupRestoreRequest): Request model for restoring from a backup. Raises: @@ -170,8 +150,6 @@ def sync_detailed( graph_id=graph_id, backup_id=backup_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -187,8 +165,6 @@ def sync( *, client: AuthenticatedClient, body: BackupRestoreRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Restore Encrypted Backup @@ -236,10 +212,8 @@ def sync( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BackupRestoreRequest): Request model for restoring from a backup. Raises: @@ -255,8 +229,6 @@ def sync( backup_id=backup_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -266,8 +238,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: BackupRestoreRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Restore Encrypted Backup @@ -315,10 +285,8 @@ async def asyncio_detailed( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BackupRestoreRequest): Request model for restoring from a backup. Raises: @@ -333,8 +301,6 @@ async def asyncio_detailed( graph_id=graph_id, backup_id=backup_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -348,8 +314,6 @@ async def asyncio( *, client: AuthenticatedClient, body: BackupRestoreRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Restore Encrypted Backup @@ -397,10 +361,8 @@ async def asyncio( Returns operation details for SSE monitoring. Args: - graph_id (str): Graph database identifier + graph_id (str): backup_id (str): Backup identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (BackupRestoreRequest): Request model for restoring from a backup. Raises: @@ -417,7 +379,5 @@ async def asyncio( backup_id=backup_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/create_connection.py b/robosystems_client/api/connections/create_connection.py index 188efc8..64b4709 100644 --- a/robosystems_client/api/connections/create_connection.py +++ b/robosystems_client/api/connections/create_connection.py @@ -9,35 +9,19 @@ from ...models.create_connection_request import CreateConnectionRequest from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: CreateConnectionRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections", - "params": params, } _kwargs["json"] = body.to_dict() @@ -103,8 +87,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: CreateConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Create Connection @@ -131,9 +113,7 @@ def sync_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateConnectionRequest): Request to create a new connection. Raises: @@ -147,8 +127,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -163,8 +141,6 @@ def sync( *, client: AuthenticatedClient, body: CreateConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Create Connection @@ -191,9 +167,7 @@ def sync( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateConnectionRequest): Request to create a new connection. Raises: @@ -208,8 +182,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -218,8 +190,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: CreateConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Create Connection @@ -246,9 +216,7 @@ async def asyncio_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateConnectionRequest): Request to create a new connection. Raises: @@ -262,8 +230,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -276,8 +242,6 @@ async def asyncio( *, client: AuthenticatedClient, body: CreateConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Create Connection @@ -304,9 +268,7 @@ async def asyncio( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateConnectionRequest): Request to create a new connection. Raises: @@ -322,7 +284,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/create_link_token.py b/robosystems_client/api/connections/create_link_token.py index 7fa2457..c3936ae 100644 --- a/robosystems_client/api/connections/create_link_token.py +++ b/robosystems_client/api/connections/create_link_token.py @@ -8,35 +8,19 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.link_token_request import LinkTokenRequest -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: LinkTokenRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections/link/token", - "params": params, } _kwargs["json"] = body.to_dict() @@ -96,8 +80,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: LinkTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Link Token @@ -117,9 +99,7 @@ def sync_detailed( No credits are consumed for creating link tokens. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (LinkTokenRequest): Request to create a link token for embedded authentication. Raises: @@ -133,8 +113,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -149,8 +127,6 @@ def sync( *, client: AuthenticatedClient, body: LinkTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Link Token @@ -170,9 +146,7 @@ def sync( No credits are consumed for creating link tokens. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (LinkTokenRequest): Request to create a link token for embedded authentication. Raises: @@ -187,8 +161,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -197,8 +169,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: LinkTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Link Token @@ -218,9 +188,7 @@ async def asyncio_detailed( No credits are consumed for creating link tokens. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (LinkTokenRequest): Request to create a link token for embedded authentication. Raises: @@ -234,8 +202,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -248,8 +214,6 @@ async def asyncio( *, client: AuthenticatedClient, body: LinkTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Create Link Token @@ -269,9 +233,7 @@ async def asyncio( No credits are consumed for creating link tokens. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (LinkTokenRequest): Request to create a link token for embedded authentication. Raises: @@ -287,7 +249,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/delete_connection.py b/robosystems_client/api/connections/delete_connection.py index 51445a6..9209083 100644 --- a/robosystems_client/api/connections/delete_connection.py +++ b/robosystems_client/api/connections/delete_connection.py @@ -8,38 +8,18 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.success_response import SuccessResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, connection_id: str, - *, - 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": "delete", "url": f"/v1/graphs/{graph_id}/connections/{connection_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -93,8 +73,6 @@ def sync_detailed( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Delete Connection @@ -112,10 +90,8 @@ def sync_detailed( Only users with admin role can delete connections. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -128,8 +104,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, connection_id=connection_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -144,8 +118,6 @@ def sync( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Delete Connection @@ -163,10 +135,8 @@ def sync( Only users with admin role can delete connections. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -180,8 +150,6 @@ def sync( graph_id=graph_id, connection_id=connection_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -190,8 +158,6 @@ async def asyncio_detailed( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Delete Connection @@ -209,10 +175,8 @@ async def asyncio_detailed( Only users with admin role can delete connections. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -225,8 +189,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, connection_id=connection_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -239,8 +201,6 @@ async def asyncio( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Delete Connection @@ -258,10 +218,8 @@ async def asyncio( Only users with admin role can delete connections. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -276,7 +234,5 @@ async def asyncio( graph_id=graph_id, connection_id=connection_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/exchange_link_token.py b/robosystems_client/api/connections/exchange_link_token.py index efb4d37..df3f0a9 100644 --- a/robosystems_client/api/connections/exchange_link_token.py +++ b/robosystems_client/api/connections/exchange_link_token.py @@ -8,35 +8,19 @@ from ...models.error_response import ErrorResponse from ...models.exchange_token_request import ExchangeTokenRequest from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: ExchangeTokenRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections/link/exchange", - "params": params, } _kwargs["json"] = body.to_dict() @@ -96,8 +80,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: ExchangeTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Exchange Link Token @@ -122,9 +104,7 @@ def sync_detailed( No credits are consumed for token exchange. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (ExchangeTokenRequest): Exchange temporary token for permanent credentials. Raises: @@ -138,8 +118,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -154,8 +132,6 @@ def sync( *, client: AuthenticatedClient, body: ExchangeTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Exchange Link Token @@ -180,9 +156,7 @@ def sync( No credits are consumed for token exchange. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (ExchangeTokenRequest): Exchange temporary token for permanent credentials. Raises: @@ -197,8 +171,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -207,8 +179,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: ExchangeTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Exchange Link Token @@ -233,9 +203,7 @@ async def asyncio_detailed( No credits are consumed for token exchange. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (ExchangeTokenRequest): Exchange temporary token for permanent credentials. Raises: @@ -249,8 +217,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -263,8 +229,6 @@ async def asyncio( *, client: AuthenticatedClient, body: ExchangeTokenRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Exchange Link Token @@ -289,9 +253,7 @@ async def asyncio( No credits are consumed for token exchange. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (ExchangeTokenRequest): Exchange temporary token for permanent credentials. Raises: @@ -307,7 +269,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/get_connection.py b/robosystems_client/api/connections/get_connection.py index 0c1b9f1..fb5dd1d 100644 --- a/robosystems_client/api/connections/get_connection.py +++ b/robosystems_client/api/connections/get_connection.py @@ -8,38 +8,18 @@ from ...models.connection_response import ConnectionResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, connection_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/connections/{connection_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -93,8 +73,6 @@ def sync_detailed( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Get Connection @@ -110,10 +88,8 @@ def sync_detailed( No credits are consumed for viewing connection details. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Unique connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -126,8 +102,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, connection_id=connection_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -142,8 +116,6 @@ def sync( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Get Connection @@ -159,10 +131,8 @@ def sync( No credits are consumed for viewing connection details. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Unique connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -176,8 +146,6 @@ def sync( graph_id=graph_id, connection_id=connection_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -186,8 +154,6 @@ async def asyncio_detailed( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Get Connection @@ -203,10 +169,8 @@ async def asyncio_detailed( No credits are consumed for viewing connection details. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Unique connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -219,8 +183,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, connection_id=connection_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -233,8 +195,6 @@ async def asyncio( connection_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]: """Get Connection @@ -250,10 +210,8 @@ async def asyncio( No credits are consumed for viewing connection details. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Unique connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -268,7 +226,5 @@ async def asyncio( graph_id=graph_id, connection_id=connection_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/get_connection_options.py b/robosystems_client/api/connections/get_connection_options.py index 1e53f0e..a92a8fe 100644 --- a/robosystems_client/api/connections/get_connection_options.py +++ b/robosystems_client/api/connections/get_connection_options.py @@ -8,37 +8,17 @@ from ...models.connection_options_response import ConnectionOptionsResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/connections/options", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -86,8 +66,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]: """List Connection Options @@ -113,9 +91,7 @@ def sync_detailed( No credits are consumed for viewing connection options. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -127,8 +103,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -142,8 +116,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]: """List Connection Options @@ -169,9 +141,7 @@ def sync( No credits are consumed for viewing connection options. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -184,8 +154,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -193,8 +161,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]: """List Connection Options @@ -220,9 +186,7 @@ async def asyncio_detailed( No credits are consumed for viewing connection options. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -234,8 +198,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -247,8 +209,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]: """List Connection Options @@ -274,9 +234,7 @@ async def asyncio( No credits are consumed for viewing connection options. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -290,7 +248,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/init_o_auth.py b/robosystems_client/api/connections/init_o_auth.py index 04f1c71..108f88c 100644 --- a/robosystems_client/api/connections/init_o_auth.py +++ b/robosystems_client/api/connections/init_o_auth.py @@ -8,35 +8,19 @@ from ...models.http_validation_error import HTTPValidationError from ...models.o_auth_init_request import OAuthInitRequest from ...models.o_auth_init_response import OAuthInitResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: OAuthInitRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections/oauth/init", - "params": params, } _kwargs["json"] = body.to_dict() @@ -82,8 +66,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: OAuthInitRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]: """Init Oauth @@ -93,9 +75,7 @@ def sync_detailed( Currently supports: QuickBooks Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (OAuthInitRequest): Request to initiate OAuth flow. Raises: @@ -109,8 +89,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -125,8 +103,6 @@ def sync( *, client: AuthenticatedClient, body: OAuthInitRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]: """Init Oauth @@ -136,9 +112,7 @@ def sync( Currently supports: QuickBooks Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (OAuthInitRequest): Request to initiate OAuth flow. Raises: @@ -153,8 +127,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -163,8 +135,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: OAuthInitRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]: """Init Oauth @@ -174,9 +144,7 @@ async def asyncio_detailed( Currently supports: QuickBooks Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (OAuthInitRequest): Request to initiate OAuth flow. Raises: @@ -190,8 +158,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -204,8 +170,6 @@ async def asyncio( *, client: AuthenticatedClient, body: OAuthInitRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]: """Init Oauth @@ -215,9 +179,7 @@ async def asyncio( Currently supports: QuickBooks Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (OAuthInitRequest): Request to initiate OAuth flow. Raises: @@ -233,7 +195,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/list_connections.py b/robosystems_client/api/connections/list_connections.py index 47a6470..e690c9d 100644 --- a/robosystems_client/api/connections/list_connections.py +++ b/robosystems_client/api/connections/list_connections.py @@ -17,13 +17,7 @@ def _get_kwargs( *, entity_id: Union[None, Unset, str] = UNSET, provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET, - 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_entity_id: Union[None, Unset, str] @@ -42,13 +36,6 @@ def _get_kwargs( json_provider = provider params["provider"] = json_provider - 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] = { @@ -57,7 +44,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -112,8 +98,6 @@ def sync_detailed( client: AuthenticatedClient, entity_id: Union[None, Unset, str] = UNSET, provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]: """List Connections @@ -133,11 +117,9 @@ def sync_detailed( No credits are consumed for listing connections. Args: - graph_id (str): Graph database identifier + graph_id (str): entity_id (Union[None, Unset, str]): Filter by entity ID provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -151,8 +133,6 @@ def sync_detailed( graph_id=graph_id, entity_id=entity_id, provider=provider, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -168,8 +148,6 @@ def sync( client: AuthenticatedClient, entity_id: Union[None, Unset, str] = UNSET, provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]: """List Connections @@ -189,11 +167,9 @@ def sync( No credits are consumed for listing connections. Args: - graph_id (str): Graph database identifier + graph_id (str): entity_id (Union[None, Unset, str]): Filter by entity ID provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -208,8 +184,6 @@ def sync( client=client, entity_id=entity_id, provider=provider, - token=token, - authorization=authorization, ).parsed @@ -219,8 +193,6 @@ async def asyncio_detailed( client: AuthenticatedClient, entity_id: Union[None, Unset, str] = UNSET, provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]: """List Connections @@ -240,11 +212,9 @@ async def asyncio_detailed( No credits are consumed for listing connections. Args: - graph_id (str): Graph database identifier + graph_id (str): entity_id (Union[None, Unset, str]): Filter by entity ID provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -258,8 +228,6 @@ async def asyncio_detailed( graph_id=graph_id, entity_id=entity_id, provider=provider, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -273,8 +241,6 @@ async def asyncio( client: AuthenticatedClient, entity_id: Union[None, Unset, str] = UNSET, provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]: """List Connections @@ -294,11 +260,9 @@ async def asyncio( No credits are consumed for listing connections. Args: - graph_id (str): Graph database identifier + graph_id (str): entity_id (Union[None, Unset, str]): Filter by entity ID provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -314,7 +278,5 @@ async def asyncio( client=client, entity_id=entity_id, provider=provider, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/oauth_callback.py b/robosystems_client/api/connections/oauth_callback.py index d68b93f..e8361e8 100644 --- a/robosystems_client/api/connections/oauth_callback.py +++ b/robosystems_client/api/connections/oauth_callback.py @@ -8,7 +8,7 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.o_auth_callback_request import OAuthCallbackRequest -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -16,28 +16,12 @@ def _get_kwargs( provider: str, *, body: OAuthCallbackRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections/oauth/callback/{provider}", - "params": params, } _kwargs["json"] = body.to_dict() @@ -103,8 +87,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: OAuthCallbackRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """OAuth Callback @@ -129,10 +111,8 @@ def sync_detailed( No credits are consumed for OAuth callbacks. Args: - graph_id (str): Graph database identifier + graph_id (str): provider (str): OAuth provider name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (OAuthCallbackRequest): OAuth callback parameters. Raises: @@ -147,8 +127,6 @@ def sync_detailed( graph_id=graph_id, provider=provider, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -164,8 +142,6 @@ def sync( *, client: AuthenticatedClient, body: OAuthCallbackRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """OAuth Callback @@ -190,10 +166,8 @@ def sync( No credits are consumed for OAuth callbacks. Args: - graph_id (str): Graph database identifier + graph_id (str): provider (str): OAuth provider name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (OAuthCallbackRequest): OAuth callback parameters. Raises: @@ -209,8 +183,6 @@ def sync( provider=provider, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -220,8 +192,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: OAuthCallbackRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """OAuth Callback @@ -246,10 +216,8 @@ async def asyncio_detailed( No credits are consumed for OAuth callbacks. Args: - graph_id (str): Graph database identifier + graph_id (str): provider (str): OAuth provider name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (OAuthCallbackRequest): OAuth callback parameters. Raises: @@ -264,8 +232,6 @@ async def asyncio_detailed( graph_id=graph_id, provider=provider, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -279,8 +245,6 @@ async def asyncio( *, client: AuthenticatedClient, body: OAuthCallbackRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """OAuth Callback @@ -305,10 +269,8 @@ async def asyncio( No credits are consumed for OAuth callbacks. Args: - graph_id (str): Graph database identifier + graph_id (str): provider (str): OAuth provider name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (OAuthCallbackRequest): OAuth callback parameters. Raises: @@ -325,7 +287,5 @@ async def asyncio( provider=provider, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/connections/sync_connection.py b/robosystems_client/api/connections/sync_connection.py index 5591945..3b800e4 100644 --- a/robosystems_client/api/connections/sync_connection.py +++ b/robosystems_client/api/connections/sync_connection.py @@ -11,7 +11,7 @@ from ...models.sync_connection_response_syncconnection import ( SyncConnectionResponseSyncconnection, ) -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -19,28 +19,12 @@ def _get_kwargs( connection_id: str, *, body: SyncConnectionRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/connections/{connection_id}/sync", - "params": params, } _kwargs["json"] = body.to_dict() @@ -106,8 +90,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: SyncConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection] ]: @@ -139,10 +121,8 @@ def sync_detailed( Returns a task ID for monitoring sync progress. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SyncConnectionRequest): Request to sync a connection. Raises: @@ -157,8 +137,6 @@ def sync_detailed( graph_id=graph_id, connection_id=connection_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -174,8 +152,6 @@ def sync( *, client: AuthenticatedClient, body: SyncConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection] ]: @@ -207,10 +183,8 @@ def sync( Returns a task ID for monitoring sync progress. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SyncConnectionRequest): Request to sync a connection. Raises: @@ -226,8 +200,6 @@ def sync( connection_id=connection_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -237,8 +209,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: SyncConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection] ]: @@ -270,10 +240,8 @@ async def asyncio_detailed( Returns a task ID for monitoring sync progress. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SyncConnectionRequest): Request to sync a connection. Raises: @@ -288,8 +256,6 @@ async def asyncio_detailed( graph_id=graph_id, connection_id=connection_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -303,8 +269,6 @@ async def asyncio( *, client: AuthenticatedClient, body: SyncConnectionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection] ]: @@ -336,10 +300,8 @@ async def asyncio( Returns a task ID for monitoring sync progress. Args: - graph_id (str): Graph database identifier + graph_id (str): connection_id (str): Connection identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SyncConnectionRequest): Request to sync a connection. Raises: @@ -356,7 +318,5 @@ async def asyncio( connection_id=connection_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_analytics/get_graph_metrics.py b/robosystems_client/api/graph_analytics/get_graph_metrics.py index f4e84c6..682abed 100644 --- a/robosystems_client/api/graph_analytics/get_graph_metrics.py +++ b/robosystems_client/api/graph_analytics/get_graph_metrics.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.graph_metrics_response import GraphMetricsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/analytics", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -91,8 +71,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]: """Get Graph Metrics @@ -115,9 +93,7 @@ def sync_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get metrics for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -129,8 +105,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -144,8 +118,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]: """Get Graph Metrics @@ -168,9 +140,7 @@ def sync( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get metrics for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -183,8 +153,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -192,8 +160,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]: """Get Graph Metrics @@ -216,9 +182,7 @@ async def asyncio_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get metrics for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -230,8 +194,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -243,8 +205,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]: """Get Graph Metrics @@ -267,9 +227,7 @@ async def asyncio( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get metrics for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -283,7 +241,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_analytics/get_graph_usage_stats.py b/robosystems_client/api/graph_analytics/get_graph_usage_stats.py index b0c516a..ac97422 100644 --- a/robosystems_client/api/graph_analytics/get_graph_usage_stats.py +++ b/robosystems_client/api/graph_analytics/get_graph_usage_stats.py @@ -15,24 +15,11 @@ def _get_kwargs( graph_id: str, *, include_details: Union[Unset, bool] = False, - 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] = {} params["include_details"] = include_details - 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] = { @@ -41,7 +28,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -90,8 +76,6 @@ def sync_detailed( *, client: AuthenticatedClient, include_details: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]: """Get Usage Statistics @@ -120,11 +104,9 @@ def sync_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get usage stats for + graph_id (str): include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -137,8 +119,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, include_details=include_details, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -153,8 +133,6 @@ def sync( *, client: AuthenticatedClient, include_details: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]: """Get Usage Statistics @@ -183,11 +161,9 @@ def sync( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get usage stats for + graph_id (str): include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -201,8 +177,6 @@ def sync( graph_id=graph_id, client=client, include_details=include_details, - token=token, - authorization=authorization, ).parsed @@ -211,8 +185,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, include_details: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]: """Get Usage Statistics @@ -241,11 +213,9 @@ async def asyncio_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get usage stats for + graph_id (str): include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -258,8 +228,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, include_details=include_details, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -272,8 +240,6 @@ async def asyncio( *, client: AuthenticatedClient, include_details: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]: """Get Usage Statistics @@ -302,11 +268,9 @@ async def asyncio( This operation is included - no credit consumption required. Args: - graph_id (str): The graph ID to get usage stats for + graph_id (str): include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -321,7 +285,5 @@ async def asyncio( graph_id=graph_id, client=client, include_details=include_details, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_billing/get_current_graph_bill.py b/robosystems_client/api/graph_billing/get_current_graph_bill.py index a3d003c..2cd41cf 100644 --- a/robosystems_client/api/graph_billing/get_current_graph_bill.py +++ b/robosystems_client/api/graph_billing/get_current_graph_bill.py @@ -10,37 +10,17 @@ GetCurrentGraphBillResponseGetcurrentgraphbill, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/billing/current", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -103,8 +83,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetCurrentGraphBillResponseGetcurrentgraphbill, HTTPValidationError @@ -126,9 +104,7 @@ def sync_detailed( ā„¹ļø No credits are consumed for viewing billing information. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -140,8 +116,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -155,8 +129,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetCurrentGraphBillResponseGetcurrentgraphbill, HTTPValidationError @@ -178,9 +150,7 @@ def sync( ā„¹ļø No credits are consumed for viewing billing information. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -193,8 +163,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -202,8 +170,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetCurrentGraphBillResponseGetcurrentgraphbill, HTTPValidationError @@ -225,9 +191,7 @@ async def asyncio_detailed( ā„¹ļø No credits are consumed for viewing billing information. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -239,8 +203,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -252,8 +214,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetCurrentGraphBillResponseGetcurrentgraphbill, HTTPValidationError @@ -275,9 +235,7 @@ async def asyncio( ā„¹ļø No credits are consumed for viewing billing information. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -291,7 +249,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_billing/get_graph_billing_history.py b/robosystems_client/api/graph_billing/get_graph_billing_history.py index 61424af..019ca82 100644 --- a/robosystems_client/api/graph_billing/get_graph_billing_history.py +++ b/robosystems_client/api/graph_billing/get_graph_billing_history.py @@ -17,24 +17,11 @@ def _get_kwargs( graph_id: str, *, months: Union[Unset, int] = 6, - 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] = {} params["months"] = months - 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] = { @@ -43,7 +30,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -111,8 +97,6 @@ def sync_detailed( *, client: AuthenticatedClient, months: Union[Unset, int] = 6, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, @@ -139,10 +123,8 @@ def sync_detailed( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -155,8 +137,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, months=months, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -171,8 +151,6 @@ def sync( *, client: AuthenticatedClient, months: Union[Unset, int] = 6, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, @@ -199,10 +177,8 @@ def sync( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -216,8 +192,6 @@ def sync( graph_id=graph_id, client=client, months=months, - token=token, - authorization=authorization, ).parsed @@ -226,8 +200,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, months: Union[Unset, int] = 6, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, @@ -254,10 +226,8 @@ async def asyncio_detailed( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -270,8 +240,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, months=months, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -284,8 +252,6 @@ async def asyncio( *, client: AuthenticatedClient, months: Union[Unset, int] = 6, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, @@ -312,10 +278,8 @@ async def asyncio( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -330,7 +294,5 @@ async def asyncio( graph_id=graph_id, client=client, months=months, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_billing/get_graph_monthly_bill.py b/robosystems_client/api/graph_billing/get_graph_monthly_bill.py index 371728c..2628f34 100644 --- a/robosystems_client/api/graph_billing/get_graph_monthly_bill.py +++ b/robosystems_client/api/graph_billing/get_graph_monthly_bill.py @@ -10,39 +10,19 @@ GetGraphMonthlyBillResponseGetgraphmonthlybill, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, year: int, month: int, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/billing/history/{year}/{month}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -112,8 +92,6 @@ def sync_detailed( month: int, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError @@ -135,11 +113,9 @@ def sync_detailed( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): year (int): Year (2024-2030) month (int): Month (1-12) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -153,8 +129,6 @@ def sync_detailed( graph_id=graph_id, year=year, month=month, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -170,8 +144,6 @@ def sync( month: int, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError @@ -193,11 +165,9 @@ def sync( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): year (int): Year (2024-2030) month (int): Month (1-12) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -212,8 +182,6 @@ def sync( year=year, month=month, client=client, - token=token, - authorization=authorization, ).parsed @@ -223,8 +191,6 @@ async def asyncio_detailed( month: int, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError @@ -246,11 +212,9 @@ async def asyncio_detailed( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): year (int): Year (2024-2030) month (int): Month (1-12) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -264,8 +228,6 @@ async def asyncio_detailed( graph_id=graph_id, year=year, month=month, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -279,8 +241,6 @@ async def asyncio( month: int, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError @@ -302,11 +262,9 @@ async def asyncio( ā„¹ļø No credits are consumed for viewing billing history. Args: - graph_id (str): Graph database identifier + graph_id (str): year (int): Year (2024-2030) month (int): Month (1-12) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -322,7 +280,5 @@ async def asyncio( year=year, month=month, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_billing/get_graph_usage_details.py b/robosystems_client/api/graph_billing/get_graph_usage_details.py index 98357c9..1abc088 100644 --- a/robosystems_client/api/graph_billing/get_graph_usage_details.py +++ b/robosystems_client/api/graph_billing/get_graph_usage_details.py @@ -18,13 +18,7 @@ def _get_kwargs( *, year: Union[None, Unset, int] = UNSET, month: Union[None, Unset, int] = UNSET, - 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_year: Union[None, Unset, int] @@ -41,13 +35,6 @@ def _get_kwargs( json_month = month params["month"] = json_month - 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] = { @@ -56,7 +43,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -126,8 +112,6 @@ def sync_detailed( client: AuthenticatedClient, year: Union[None, Unset, int] = UNSET, month: Union[None, Unset, int] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError @@ -153,11 +137,9 @@ def sync_detailed( ā„¹ļø No credits are consumed for viewing usage details. Args: - graph_id (str): Graph database identifier + graph_id (str): year (Union[None, Unset, int]): Year (defaults to current) month (Union[None, Unset, int]): Month (defaults to current) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -171,8 +153,6 @@ def sync_detailed( graph_id=graph_id, year=year, month=month, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -188,8 +168,6 @@ def sync( client: AuthenticatedClient, year: Union[None, Unset, int] = UNSET, month: Union[None, Unset, int] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError @@ -215,11 +193,9 @@ def sync( ā„¹ļø No credits are consumed for viewing usage details. Args: - graph_id (str): Graph database identifier + graph_id (str): year (Union[None, Unset, int]): Year (defaults to current) month (Union[None, Unset, int]): Month (defaults to current) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -234,8 +210,6 @@ def sync( client=client, year=year, month=month, - token=token, - authorization=authorization, ).parsed @@ -245,8 +219,6 @@ async def asyncio_detailed( client: AuthenticatedClient, year: Union[None, Unset, int] = UNSET, month: Union[None, Unset, int] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError @@ -272,11 +244,9 @@ async def asyncio_detailed( ā„¹ļø No credits are consumed for viewing usage details. Args: - graph_id (str): Graph database identifier + graph_id (str): year (Union[None, Unset, int]): Year (defaults to current) month (Union[None, Unset, int]): Month (defaults to current) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -290,8 +260,6 @@ async def asyncio_detailed( graph_id=graph_id, year=year, month=month, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -305,8 +273,6 @@ async def asyncio( client: AuthenticatedClient, year: Union[None, Unset, int] = UNSET, month: Union[None, Unset, int] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError @@ -332,11 +298,9 @@ async def asyncio( ā„¹ļø No credits are consumed for viewing usage details. Args: - graph_id (str): Graph database identifier + graph_id (str): year (Union[None, Unset, int]): Year (defaults to current) month (Union[None, Unset, int]): Month (defaults to current) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -352,7 +316,5 @@ async def asyncio( client=client, year=year, month=month, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_credits/check_credit_balance.py b/robosystems_client/api/graph_credits/check_credit_balance.py index 53b5b3e..4e5c70a 100644 --- a/robosystems_client/api/graph_credits/check_credit_balance.py +++ b/robosystems_client/api/graph_credits/check_credit_balance.py @@ -18,13 +18,7 @@ def _get_kwargs( *, operation_type: str, base_cost: Union[None, Unset, float, str] = UNSET, - 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] = {} params["operation_type"] = operation_type @@ -36,13 +30,6 @@ def _get_kwargs( json_base_cost = base_cost params["base_cost"] = json_base_cost - 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] = { @@ -51,7 +38,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -116,8 +102,6 @@ def sync_detailed( client: AuthenticatedClient, operation_type: str, base_cost: Union[None, Unset, float, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError @@ -142,8 +126,6 @@ def sync_detailed( operation_type (str): Type of operation to check base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not provided) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -157,8 +139,6 @@ def sync_detailed( graph_id=graph_id, operation_type=operation_type, base_cost=base_cost, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -174,8 +154,6 @@ def sync( client: AuthenticatedClient, operation_type: str, base_cost: Union[None, Unset, float, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError @@ -200,8 +178,6 @@ def sync( operation_type (str): Type of operation to check base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not provided) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -216,8 +192,6 @@ def sync( client=client, operation_type=operation_type, base_cost=base_cost, - token=token, - authorization=authorization, ).parsed @@ -227,8 +201,6 @@ async def asyncio_detailed( client: AuthenticatedClient, operation_type: str, base_cost: Union[None, Unset, float, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError @@ -253,8 +225,6 @@ async def asyncio_detailed( operation_type (str): Type of operation to check base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not provided) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -268,8 +238,6 @@ async def asyncio_detailed( graph_id=graph_id, operation_type=operation_type, base_cost=base_cost, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -283,8 +251,6 @@ async def asyncio( client: AuthenticatedClient, operation_type: str, base_cost: Union[None, Unset, float, str] = UNSET, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError @@ -309,8 +275,6 @@ async def asyncio( operation_type (str): Type of operation to check base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not provided) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -326,7 +290,5 @@ async def asyncio( client=client, operation_type=operation_type, base_cost=base_cost, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_credits/check_storage_limits.py b/robosystems_client/api/graph_credits/check_storage_limits.py index 1ec089e..0986333 100644 --- a/robosystems_client/api/graph_credits/check_storage_limits.py +++ b/robosystems_client/api/graph_credits/check_storage_limits.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.storage_limit_response import StorageLimitResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/credits/storage/limits", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -91,8 +71,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]: """Check Storage Limits @@ -109,8 +87,6 @@ def sync_detailed( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -122,8 +98,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -137,8 +111,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]: """Check Storage Limits @@ -155,8 +127,6 @@ def sync( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -169,8 +139,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -178,8 +146,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]: """Check Storage Limits @@ -196,8 +162,6 @@ async def asyncio_detailed( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -209,8 +173,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -222,8 +184,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]: """Check Storage Limits @@ -240,8 +200,6 @@ async def asyncio( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -255,7 +213,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_credits/get_credit_summary.py b/robosystems_client/api/graph_credits/get_credit_summary.py index a83495c..2e6a952 100644 --- a/robosystems_client/api/graph_credits/get_credit_summary.py +++ b/robosystems_client/api/graph_credits/get_credit_summary.py @@ -8,37 +8,17 @@ from ...models.credit_summary_response import CreditSummaryResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/credits/summary", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -91,8 +71,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]: """Get Credit Summary @@ -108,8 +86,6 @@ def sync_detailed( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -121,8 +97,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -136,8 +110,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]: """Get Credit Summary @@ -153,8 +125,6 @@ def sync( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -167,8 +137,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -176,8 +144,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]: """Get Credit Summary @@ -193,8 +159,6 @@ async def asyncio_detailed( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -206,8 +170,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -219,8 +181,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]: """Get Credit Summary @@ -236,8 +196,6 @@ async def asyncio( Args: graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -251,7 +209,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_credits/get_storage_usage.py b/robosystems_client/api/graph_credits/get_storage_usage.py index 4554910..05e2e8b 100644 --- a/robosystems_client/api/graph_credits/get_storage_usage.py +++ b/robosystems_client/api/graph_credits/get_storage_usage.py @@ -17,24 +17,11 @@ def _get_kwargs( graph_id: str, *, days: Union[Unset, int] = 30, - 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] = {} params["days"] = days - 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] = { @@ -43,7 +30,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -96,8 +82,6 @@ def sync_detailed( *, client: AuthenticatedClient, days: Union[Unset, int] = 30, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError] ]: @@ -117,8 +101,6 @@ def sync_detailed( Args: graph_id (str): Graph database identifier days (Union[Unset, int]): Number of days of history to return Default: 30. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -131,8 +113,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, days=days, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -147,8 +127,6 @@ def sync( *, client: AuthenticatedClient, days: Union[Unset, int] = 30, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError] ]: @@ -168,8 +146,6 @@ def sync( Args: graph_id (str): Graph database identifier days (Union[Unset, int]): Number of days of history to return Default: 30. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -183,8 +159,6 @@ def sync( graph_id=graph_id, client=client, days=days, - token=token, - authorization=authorization, ).parsed @@ -193,8 +167,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, days: Union[Unset, int] = 30, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError] ]: @@ -214,8 +186,6 @@ async def asyncio_detailed( Args: graph_id (str): Graph database identifier days (Union[Unset, int]): Number of days of history to return Default: 30. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -228,8 +198,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, days=days, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -242,8 +210,6 @@ async def asyncio( *, client: AuthenticatedClient, days: Union[Unset, int] = 30, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError] ]: @@ -263,8 +229,6 @@ async def asyncio( Args: graph_id (str): Graph database identifier days (Union[Unset, int]): Number of days of history to return Default: 30. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -279,7 +243,5 @@ async def asyncio( graph_id=graph_id, client=client, days=days, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_credits/list_credit_transactions.py b/robosystems_client/api/graph_credits/list_credit_transactions.py index 094a3ce..b0aa6c1 100644 --- a/robosystems_client/api/graph_credits/list_credit_transactions.py +++ b/robosystems_client/api/graph_credits/list_credit_transactions.py @@ -20,13 +20,7 @@ def _get_kwargs( end_date: Union[None, Unset, str] = UNSET, limit: Union[Unset, int] = 100, offset: Union[Unset, int] = 0, - 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_transaction_type: Union[None, Unset, str] @@ -61,13 +55,6 @@ def _get_kwargs( params["offset"] = offset - 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] = { @@ -76,7 +63,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -135,8 +121,6 @@ def sync_detailed( end_date: Union[None, Unset, str] = UNSET, limit: Union[Unset, int] = 100, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]: """List Credit Transactions @@ -157,7 +141,7 @@ def sync_detailed( No credits are consumed for viewing transaction history. Args: - graph_id (str): Graph database identifier + graph_id (str): transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation, consumption, bonus, refund) operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup, @@ -166,8 +150,6 @@ def sync_detailed( end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD) limit (Union[Unset, int]): Maximum number of transactions to return Default: 100. offset (Union[Unset, int]): Number of transactions to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -185,8 +167,6 @@ def sync_detailed( end_date=end_date, limit=limit, offset=offset, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -206,8 +186,6 @@ def sync( end_date: Union[None, Unset, str] = UNSET, limit: Union[Unset, int] = 100, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]: """List Credit Transactions @@ -228,7 +206,7 @@ def sync( No credits are consumed for viewing transaction history. Args: - graph_id (str): Graph database identifier + graph_id (str): transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation, consumption, bonus, refund) operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup, @@ -237,8 +215,6 @@ def sync( end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD) limit (Union[Unset, int]): Maximum number of transactions to return Default: 100. offset (Union[Unset, int]): Number of transactions to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -257,8 +233,6 @@ def sync( end_date=end_date, limit=limit, offset=offset, - token=token, - authorization=authorization, ).parsed @@ -272,8 +246,6 @@ async def asyncio_detailed( end_date: Union[None, Unset, str] = UNSET, limit: Union[Unset, int] = 100, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]: """List Credit Transactions @@ -294,7 +266,7 @@ async def asyncio_detailed( No credits are consumed for viewing transaction history. Args: - graph_id (str): Graph database identifier + graph_id (str): transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation, consumption, bonus, refund) operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup, @@ -303,8 +275,6 @@ async def asyncio_detailed( end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD) limit (Union[Unset, int]): Maximum number of transactions to return Default: 100. offset (Union[Unset, int]): Number of transactions to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -322,8 +292,6 @@ async def asyncio_detailed( end_date=end_date, limit=limit, offset=offset, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -341,8 +309,6 @@ async def asyncio( end_date: Union[None, Unset, str] = UNSET, limit: Union[Unset, int] = 100, offset: Union[Unset, int] = 0, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]: """List Credit Transactions @@ -363,7 +329,7 @@ async def asyncio( No credits are consumed for viewing transaction history. Args: - graph_id (str): Graph database identifier + graph_id (str): transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation, consumption, bonus, refund) operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup, @@ -372,8 +338,6 @@ async def asyncio( end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD) limit (Union[Unset, int]): Maximum number of transactions to return Default: 100. offset (Union[Unset, int]): Number of transactions to skip Default: 0. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -393,7 +357,5 @@ async def asyncio( end_date=end_date, limit=limit, offset=offset, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_health/get_database_health.py b/robosystems_client/api/graph_health/get_database_health.py index c944c40..32542f9 100644 --- a/robosystems_client/api/graph_health/get_database_health.py +++ b/robosystems_client/api/graph_health/get_database_health.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.database_health_response import DatabaseHealthResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/health", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -87,8 +67,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DatabaseHealthResponse, HTTPValidationError]]: """Database Health Check @@ -111,9 +89,7 @@ def sync_detailed( This endpoint provides essential monitoring data for operational visibility. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -125,8 +101,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -140,8 +114,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]: """Database Health Check @@ -164,9 +136,7 @@ def sync( This endpoint provides essential monitoring data for operational visibility. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -179,8 +149,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -188,8 +156,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DatabaseHealthResponse, HTTPValidationError]]: """Database Health Check @@ -212,9 +178,7 @@ async def asyncio_detailed( This endpoint provides essential monitoring data for operational visibility. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -226,8 +190,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -239,8 +201,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]: """Database Health Check @@ -263,9 +223,7 @@ async def asyncio( This endpoint provides essential monitoring data for operational visibility. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -279,7 +237,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_info/get_database_info.py b/robosystems_client/api/graph_info/get_database_info.py index cfd9c22..5865c5d 100644 --- a/robosystems_client/api/graph_info/get_database_info.py +++ b/robosystems_client/api/graph_info/get_database_info.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.database_info_response import DatabaseInfoResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/info", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -87,8 +67,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DatabaseInfoResponse, HTTPValidationError]]: """Database Information @@ -112,9 +90,7 @@ def sync_detailed( This endpoint provides essential database information for capacity planning and monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -126,8 +102,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -141,8 +115,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DatabaseInfoResponse, HTTPValidationError]]: """Database Information @@ -166,9 +138,7 @@ def sync( This endpoint provides essential database information for capacity planning and monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -181,8 +151,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -190,8 +158,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DatabaseInfoResponse, HTTPValidationError]]: """Database Information @@ -215,9 +181,7 @@ async def asyncio_detailed( This endpoint provides essential database information for capacity planning and monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -229,8 +193,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -242,8 +204,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DatabaseInfoResponse, HTTPValidationError]]: """Database Information @@ -267,9 +227,7 @@ async def asyncio( This endpoint provides essential database information for capacity planning and monitoring. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -283,7 +241,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graph_limits/get_graph_limits.py b/robosystems_client/api/graph_limits/get_graph_limits.py index 962bce4..8628679 100644 --- a/robosystems_client/api/graph_limits/get_graph_limits.py +++ b/robosystems_client/api/graph_limits/get_graph_limits.py @@ -9,37 +9,17 @@ GetGraphLimitsResponseGetgraphlimits, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/limits", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -89,8 +69,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]: """Get Graph Operational Limits @@ -109,9 +87,7 @@ def sync_detailed( **Note**: Limits vary based on subscription tier (Standard, Enterprise, Premium). Args: - graph_id (str): Graph database identifier (user graph or shared repository) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -123,8 +99,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -138,8 +112,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]: """Get Graph Operational Limits @@ -158,9 +130,7 @@ def sync( **Note**: Limits vary based on subscription tier (Standard, Enterprise, Premium). Args: - graph_id (str): Graph database identifier (user graph or shared repository) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -173,8 +143,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -182,8 +150,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]: """Get Graph Operational Limits @@ -202,9 +168,7 @@ async def asyncio_detailed( **Note**: Limits vary based on subscription tier (Standard, Enterprise, Premium). Args: - graph_id (str): Graph database identifier (user graph or shared repository) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -216,8 +180,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -229,8 +191,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]: """Get Graph Operational Limits @@ -249,9 +209,7 @@ async def asyncio( **Note**: Limits vary based on subscription tier (Standard, Enterprise, Premium). Args: - graph_id (str): Graph database identifier (user graph or shared repository) - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -265,7 +223,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graphs/create_graph.py b/robosystems_client/api/graphs/create_graph.py index 33157dc..ffda325 100644 --- a/robosystems_client/api/graphs/create_graph.py +++ b/robosystems_client/api/graphs/create_graph.py @@ -7,34 +7,18 @@ from ...client import AuthenticatedClient, Client from ...models.create_graph_request import CreateGraphRequest from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( *, body: CreateGraphRequest, - 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": "post", "url": "/v1/graphs", - "params": params, } _kwargs["json"] = body.to_dict() @@ -78,8 +62,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: CreateGraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError]]: """Create New Graph Database @@ -125,8 +107,6 @@ def sync_detailed( - `_links.status`: Point-in-time status check endpoint Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (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', @@ -143,8 +123,6 @@ def sync_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -158,8 +136,6 @@ def sync( *, client: AuthenticatedClient, body: CreateGraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError]]: """Create New Graph Database @@ -205,8 +181,6 @@ def sync( - `_links.status`: Point-in-time status check endpoint Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (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', @@ -224,8 +198,6 @@ def sync( return sync_detailed( client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -233,8 +205,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: CreateGraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError]]: """Create New Graph Database @@ -280,8 +250,6 @@ async def asyncio_detailed( - `_links.status`: Point-in-time status check endpoint Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (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', @@ -298,8 +266,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -311,8 +277,6 @@ async def asyncio( *, client: AuthenticatedClient, body: CreateGraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError]]: """Create New Graph Database @@ -358,8 +322,6 @@ async def asyncio( - `_links.status`: Point-in-time status check endpoint Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (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', @@ -378,7 +340,5 @@ async def asyncio( await asyncio_detailed( client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graphs/get_graphs.py b/robosystems_client/api/graphs/get_graphs.py index 8475343..2bc5acd 100644 --- a/robosystems_client/api/graphs/get_graphs.py +++ b/robosystems_client/api/graphs/get_graphs.py @@ -5,54 +5,27 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.user_graphs_response import UserGraphsResponse -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/graphs", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Optional[UserGraphsResponse]: if response.status_code == 200: response_200 = UserGraphsResponse.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: @@ -61,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Response[UserGraphsResponse]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,29 +46,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Response[UserGraphsResponse]: """Get User Graphs Get all graph databases accessible to the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserGraphsResponse]] + Response[UserGraphsResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -107,58 +71,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Optional[UserGraphsResponse]: """Get User Graphs Get all graph databases accessible to the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserGraphsResponse] + UserGraphsResponse """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Response[UserGraphsResponse]: """Get User Graphs Get all graph databases accessible to the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserGraphsResponse]] + Response[UserGraphsResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -168,29 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserGraphsResponse]]: +) -> Optional[UserGraphsResponse]: """Get User Graphs Get all graph databases accessible to the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserGraphsResponse] + UserGraphsResponse """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/graphs/select_graph.py b/robosystems_client/api/graphs/select_graph.py index 5d7b54d..41aff1a 100644 --- a/robosystems_client/api/graphs/select_graph.py +++ b/robosystems_client/api/graphs/select_graph.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.success_response import SuccessResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "post", "url": f"/v1/graphs/{graph_id}/select", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -91,8 +71,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph @@ -100,8 +78,6 @@ def sync_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -113,8 +89,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -128,8 +102,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph @@ -137,8 +109,6 @@ def sync( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -151,8 +121,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -160,8 +128,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph @@ -169,8 +135,6 @@ async def asyncio_detailed( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -182,8 +146,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -195,8 +157,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph @@ -204,8 +164,6 @@ async def asyncio( Args: graph_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -219,7 +177,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/mcp/call_mcp_tool.py b/robosystems_client/api/mcp/call_mcp_tool.py index 6ce89c3..1aab4fa 100644 --- a/robosystems_client/api/mcp/call_mcp_tool.py +++ b/robosystems_client/api/mcp/call_mcp_tool.py @@ -17,12 +17,8 @@ def _get_kwargs( body: MCPToolCall, format_: Union[None, Unset, str] = UNSET, test_mode: Union[Unset, bool] = False, - 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] = {} @@ -35,13 +31,6 @@ def _get_kwargs( params["test_mode"] = test_mode - 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] = { @@ -133,8 +122,6 @@ def sync_detailed( body: MCPToolCall, format_: Union[None, Unset, str] = UNSET, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Execute MCP Tool @@ -174,11 +161,9 @@ def sync_detailed( MCP tool calls are included and do not consume credits. Args: - graph_id (str): Graph database identifier + graph_id (str): format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson) test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (MCPToolCall): Request model for MCP tool execution. Raises: @@ -194,8 +179,6 @@ def sync_detailed( body=body, format_=format_, test_mode=test_mode, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -212,8 +195,6 @@ def sync( body: MCPToolCall, format_: Union[None, Unset, str] = UNSET, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Execute MCP Tool @@ -253,11 +234,9 @@ def sync( MCP tool calls are included and do not consume credits. Args: - graph_id (str): Graph database identifier + graph_id (str): format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson) test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (MCPToolCall): Request model for MCP tool execution. Raises: @@ -274,8 +253,6 @@ def sync( body=body, format_=format_, test_mode=test_mode, - token=token, - authorization=authorization, ).parsed @@ -286,8 +263,6 @@ async def asyncio_detailed( body: MCPToolCall, format_: Union[None, Unset, str] = UNSET, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]: """Execute MCP Tool @@ -327,11 +302,9 @@ async def asyncio_detailed( MCP tool calls are included and do not consume credits. Args: - graph_id (str): Graph database identifier + graph_id (str): format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson) test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (MCPToolCall): Request model for MCP tool execution. Raises: @@ -347,8 +320,6 @@ async def asyncio_detailed( body=body, format_=format_, test_mode=test_mode, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -363,8 +334,6 @@ async def asyncio( body: MCPToolCall, format_: Union[None, Unset, str] = UNSET, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]: """Execute MCP Tool @@ -404,11 +373,9 @@ async def asyncio( MCP tool calls are included and do not consume credits. Args: - graph_id (str): Graph database identifier + graph_id (str): format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson) test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (MCPToolCall): Request model for MCP tool execution. Raises: @@ -426,7 +393,5 @@ async def asyncio( body=body, format_=format_, test_mode=test_mode, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/mcp/list_mcp_tools.py b/robosystems_client/api/mcp/list_mcp_tools.py index caf43f9..b13bbea 100644 --- a/robosystems_client/api/mcp/list_mcp_tools.py +++ b/robosystems_client/api/mcp/list_mcp_tools.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.mcp_tools_response import MCPToolsResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/mcp/tools", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -86,8 +66,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]: """List MCP Tools @@ -108,9 +86,7 @@ def sync_detailed( - Tool execution costs vary by operation complexity Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -122,8 +98,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -137,8 +111,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]: """List MCP Tools @@ -159,9 +131,7 @@ def sync( - Tool execution costs vary by operation complexity Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -174,8 +144,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -183,8 +151,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]: """List MCP Tools @@ -205,9 +171,7 @@ async def asyncio_detailed( - Tool execution costs vary by operation complexity Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -219,8 +183,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -232,8 +194,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]: """List MCP Tools @@ -254,9 +214,7 @@ async def asyncio( - Tool execution costs vary by operation complexity Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -270,7 +228,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/operations/cancel_operation.py b/robosystems_client/api/operations/cancel_operation.py index c78cfad..f152e96 100644 --- a/robosystems_client/api/operations/cancel_operation.py +++ b/robosystems_client/api/operations/cancel_operation.py @@ -9,37 +9,17 @@ CancelOperationResponseCanceloperation, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( operation_id: str, - *, - 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": "delete", "url": f"/v1/operations/{operation_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -93,8 +73,6 @@ def sync_detailed( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]: """Cancel Operation @@ -110,8 +88,6 @@ def sync_detailed( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -123,8 +99,6 @@ def sync_detailed( kwargs = _get_kwargs( operation_id=operation_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -138,8 +112,6 @@ def sync( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]: """Cancel Operation @@ -155,8 +127,6 @@ def sync( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -169,8 +139,6 @@ def sync( return sync_detailed( operation_id=operation_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -178,8 +146,6 @@ async def asyncio_detailed( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]: """Cancel Operation @@ -195,8 +161,6 @@ async def asyncio_detailed( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -208,8 +172,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( operation_id=operation_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -221,8 +183,6 @@ async def asyncio( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]: """Cancel Operation @@ -238,8 +198,6 @@ async def asyncio( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -253,7 +211,5 @@ async def asyncio( await asyncio_detailed( operation_id=operation_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/operations/get_operation_status.py b/robosystems_client/api/operations/get_operation_status.py index dbfccce..116d66c 100644 --- a/robosystems_client/api/operations/get_operation_status.py +++ b/robosystems_client/api/operations/get_operation_status.py @@ -9,37 +9,17 @@ GetOperationStatusResponseGetoperationstatus, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( operation_id: str, - *, - 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": "get", "url": f"/v1/operations/{operation_id}/status", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -95,8 +75,6 @@ def sync_detailed( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError] ]: @@ -118,8 +96,6 @@ def sync_detailed( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -131,8 +107,6 @@ def sync_detailed( kwargs = _get_kwargs( operation_id=operation_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -146,8 +120,6 @@ def sync( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError] ]: @@ -169,8 +141,6 @@ def sync( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -183,8 +153,6 @@ def sync( return sync_detailed( operation_id=operation_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -192,8 +160,6 @@ async def asyncio_detailed( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError] ]: @@ -215,8 +181,6 @@ async def asyncio_detailed( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -228,8 +192,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( operation_id=operation_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -241,8 +203,6 @@ async def asyncio( operation_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError] ]: @@ -264,8 +224,6 @@ async def asyncio( Args: operation_id (str): Operation identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -279,7 +237,5 @@ async def asyncio( await asyncio_detailed( operation_id=operation_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/query/execute_cypher_query.py b/robosystems_client/api/query/execute_cypher_query.py index c34a26d..07c6568 100644 --- a/robosystems_client/api/query/execute_cypher_query.py +++ b/robosystems_client/api/query/execute_cypher_query.py @@ -18,12 +18,8 @@ def _get_kwargs( mode: Union[None, ResponseMode, Unset] = UNSET, chunk_size: Union[Unset, int] = 1000, test_mode: Union[Unset, bool] = False, - 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] = {} @@ -40,13 +36,6 @@ def _get_kwargs( params["test_mode"] = test_mode - 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] = { @@ -128,8 +117,6 @@ def sync_detailed( mode: Union[None, ResponseMode, Unset] = UNSET, chunk_size: Union[Unset, int] = 1000, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -188,12 +175,10 @@ def sync_detailed( Queue position is based on subscription tier for priority. Args: - graph_id (str): Graph database identifier + graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CypherQueryRequest): Request model for Cypher query execution. Raises: @@ -210,8 +195,6 @@ def sync_detailed( mode=mode, chunk_size=chunk_size, test_mode=test_mode, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -229,8 +212,6 @@ def sync( mode: Union[None, ResponseMode, Unset] = UNSET, chunk_size: Union[Unset, int] = 1000, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -289,12 +270,10 @@ def sync( Queue position is based on subscription tier for priority. Args: - graph_id (str): Graph database identifier + graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CypherQueryRequest): Request model for Cypher query execution. Raises: @@ -312,8 +291,6 @@ def sync( mode=mode, chunk_size=chunk_size, test_mode=test_mode, - token=token, - authorization=authorization, ).parsed @@ -325,8 +302,6 @@ async def asyncio_detailed( mode: Union[None, ResponseMode, Unset] = UNSET, chunk_size: Union[Unset, int] = 1000, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -385,12 +360,10 @@ async def asyncio_detailed( Queue position is based on subscription tier for priority. Args: - graph_id (str): Graph database identifier + graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CypherQueryRequest): Request model for Cypher query execution. Raises: @@ -407,8 +380,6 @@ async def asyncio_detailed( mode=mode, chunk_size=chunk_size, test_mode=test_mode, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -424,8 +395,6 @@ async def asyncio( mode: Union[None, ResponseMode, Unset] = UNSET, chunk_size: Union[Unset, int] = 1000, test_mode: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -484,12 +453,10 @@ async def asyncio( Queue position is based on subscription tier for priority. Args: - graph_id (str): Graph database identifier + graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CypherQueryRequest): Request model for Cypher query execution. Raises: @@ -508,7 +475,5 @@ async def asyncio( mode=mode, chunk_size=chunk_size, test_mode=test_mode, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/schema/export_graph_schema.py b/robosystems_client/api/schema/export_graph_schema.py index 47c751c..c1b9fdb 100644 --- a/robosystems_client/api/schema/export_graph_schema.py +++ b/robosystems_client/api/schema/export_graph_schema.py @@ -15,26 +15,13 @@ def _get_kwargs( *, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, - 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] = {} params["format"] = format_ params["include_data_stats"] = include_data_stats - 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] = { @@ -43,7 +30,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -83,20 +69,16 @@ def sync_detailed( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, SchemaExportResponse]]: """Export Graph Schema Export the schema of an existing graph in JSON, YAML, or Cypher format Args: - graph_id (str): The graph ID to export schema from + graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -110,8 +92,6 @@ def sync_detailed( graph_id=graph_id, format_=format_, include_data_stats=include_data_stats, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -127,20 +107,16 @@ def sync( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]: """Export Graph Schema Export the schema of an existing graph in JSON, YAML, or Cypher format Args: - graph_id (str): The graph ID to export schema from + graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -155,8 +131,6 @@ def sync( client=client, format_=format_, include_data_stats=include_data_stats, - token=token, - authorization=authorization, ).parsed @@ -166,20 +140,16 @@ async def asyncio_detailed( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, SchemaExportResponse]]: """Export Graph Schema Export the schema of an existing graph in JSON, YAML, or Cypher format Args: - graph_id (str): The graph ID to export schema from + graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -193,8 +163,6 @@ async def asyncio_detailed( graph_id=graph_id, format_=format_, include_data_stats=include_data_stats, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -208,20 +176,16 @@ async def asyncio( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]: """Export Graph Schema Export the schema of an existing graph in JSON, YAML, or Cypher format Args: - graph_id (str): The graph ID to export schema from + graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph Default: False. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -237,7 +201,5 @@ async def asyncio( client=client, format_=format_, include_data_stats=include_data_stats, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/schema/get_graph_schema.py b/robosystems_client/api/schema/get_graph_schema.py index 32d5837..c68f5d6 100644 --- a/robosystems_client/api/schema/get_graph_schema.py +++ b/robosystems_client/api/schema/get_graph_schema.py @@ -9,37 +9,17 @@ GetGraphSchemaResponseGetgraphschema, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/schema", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -85,8 +65,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: """Get Runtime Graph Schema @@ -103,9 +81,7 @@ def sync_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph database to get schema for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -117,8 +93,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -132,8 +106,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: """Get Runtime Graph Schema @@ -150,9 +122,7 @@ def sync( This operation is included - no credit consumption required. Args: - graph_id (str): The graph database to get schema for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -165,8 +135,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -174,8 +142,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: """Get Runtime Graph Schema @@ -192,9 +158,7 @@ async def asyncio_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): The graph database to get schema for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -206,8 +170,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -219,8 +181,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: """Get Runtime Graph Schema @@ -237,9 +197,7 @@ async def asyncio( This operation is included - no credit consumption required. Args: - graph_id (str): The graph database to get schema for - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -253,7 +211,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/schema/validate_schema.py b/robosystems_client/api/schema/validate_schema.py index 453ae7b..1e774b7 100644 --- a/robosystems_client/api/schema/validate_schema.py +++ b/robosystems_client/api/schema/validate_schema.py @@ -8,35 +8,19 @@ from ...models.error_response import ErrorResponse from ...models.schema_validation_request import SchemaValidationRequest from ...models.schema_validation_response import SchemaValidationResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: SchemaValidationRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/schema/validate", - "params": params, } _kwargs["json"] = body.to_dict() @@ -97,8 +81,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: SchemaValidationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, SchemaValidationResponse]]: """Validate Schema @@ -126,9 +108,7 @@ def sync_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (SchemaValidationRequest): Request model for schema validation. Raises: @@ -142,8 +122,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -158,8 +136,6 @@ def sync( *, client: AuthenticatedClient, body: SchemaValidationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, SchemaValidationResponse]]: """Validate Schema @@ -187,9 +163,7 @@ def sync( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (SchemaValidationRequest): Request model for schema validation. Raises: @@ -204,8 +178,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -214,8 +186,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: SchemaValidationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, SchemaValidationResponse]]: """Validate Schema @@ -243,9 +213,7 @@ async def asyncio_detailed( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (SchemaValidationRequest): Request model for schema validation. Raises: @@ -259,8 +227,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -273,8 +239,6 @@ async def asyncio( *, client: AuthenticatedClient, body: SchemaValidationRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, SchemaValidationResponse]]: """Validate Schema @@ -302,9 +266,7 @@ async def asyncio( This operation is included - no credit consumption required. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (SchemaValidationRequest): Request model for schema validation. Raises: @@ -320,7 +282,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/subgraphs/create_subgraph.py b/robosystems_client/api/subgraphs/create_subgraph.py index 038127c..f615b6d 100644 --- a/robosystems_client/api/subgraphs/create_subgraph.py +++ b/robosystems_client/api/subgraphs/create_subgraph.py @@ -8,35 +8,19 @@ from ...models.create_subgraph_request import CreateSubgraphRequest from ...models.http_validation_error import HTTPValidationError from ...models.subgraph_response import SubgraphResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: CreateSubgraphRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/subgraphs", - "params": params, } _kwargs["json"] = body.to_dict() @@ -82,8 +66,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: CreateSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, SubgraphResponse]]: """Create Subgraph @@ -101,9 +83,7 @@ def sync_detailed( - Created subgraph details including its unique ID Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateSubgraphRequest): Request model for creating a subgraph. Raises: @@ -117,8 +97,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -133,8 +111,6 @@ def sync( *, client: AuthenticatedClient, body: CreateSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, SubgraphResponse]]: """Create Subgraph @@ -152,9 +128,7 @@ def sync( - Created subgraph details including its unique ID Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateSubgraphRequest): Request model for creating a subgraph. Raises: @@ -169,8 +143,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -179,8 +151,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: CreateSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, SubgraphResponse]]: """Create Subgraph @@ -198,9 +168,7 @@ async def asyncio_detailed( - Created subgraph details including its unique ID Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateSubgraphRequest): Request model for creating a subgraph. Raises: @@ -214,8 +182,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -228,8 +194,6 @@ async def asyncio( *, client: AuthenticatedClient, body: CreateSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, SubgraphResponse]]: """Create Subgraph @@ -247,9 +211,7 @@ async def asyncio( - Created subgraph details including its unique ID Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (CreateSubgraphRequest): Request model for creating a subgraph. Raises: @@ -265,7 +227,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/subgraphs/delete_subgraph.py b/robosystems_client/api/subgraphs/delete_subgraph.py index c862458..3cbb1a8 100644 --- a/robosystems_client/api/subgraphs/delete_subgraph.py +++ b/robosystems_client/api/subgraphs/delete_subgraph.py @@ -8,7 +8,7 @@ from ...models.delete_subgraph_request import DeleteSubgraphRequest from ...models.delete_subgraph_response import DeleteSubgraphResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -16,28 +16,12 @@ def _get_kwargs( subgraph_id: str, *, body: DeleteSubgraphRequest, - 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": "delete", "url": f"/v1/graphs/{graph_id}/subgraphs/{subgraph_id}", - "params": params, } _kwargs["json"] = body.to_dict() @@ -108,8 +92,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: DeleteSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]: """Delete Subgraph @@ -133,10 +115,8 @@ def sync_detailed( `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup` Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier to delete - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (DeleteSubgraphRequest): Request model for deleting a subgraph. Raises: @@ -151,8 +131,6 @@ def sync_detailed( graph_id=graph_id, subgraph_id=subgraph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -168,8 +146,6 @@ def sync( *, client: AuthenticatedClient, body: DeleteSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]: """Delete Subgraph @@ -193,10 +169,8 @@ def sync( `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup` Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier to delete - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (DeleteSubgraphRequest): Request model for deleting a subgraph. Raises: @@ -212,8 +186,6 @@ def sync( subgraph_id=subgraph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -223,8 +195,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: DeleteSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]: """Delete Subgraph @@ -248,10 +218,8 @@ async def asyncio_detailed( `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup` Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier to delete - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (DeleteSubgraphRequest): Request model for deleting a subgraph. Raises: @@ -266,8 +234,6 @@ async def asyncio_detailed( graph_id=graph_id, subgraph_id=subgraph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -281,8 +247,6 @@ async def asyncio( *, client: AuthenticatedClient, body: DeleteSubgraphRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]: """Delete Subgraph @@ -306,10 +270,8 @@ async def asyncio( `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup` Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier to delete - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (DeleteSubgraphRequest): Request model for deleting a subgraph. Raises: @@ -326,7 +288,5 @@ async def asyncio( subgraph_id=subgraph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/subgraphs/get_subgraph_info.py b/robosystems_client/api/subgraphs/get_subgraph_info.py index e3753f3..937b889 100644 --- a/robosystems_client/api/subgraphs/get_subgraph_info.py +++ b/robosystems_client/api/subgraphs/get_subgraph_info.py @@ -7,38 +7,18 @@ from ...client import AuthenticatedClient, Client from ...models.http_validation_error import HTTPValidationError from ...models.subgraph_response import SubgraphResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, subgraph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/subgraphs/{subgraph_id}/info", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -97,8 +77,6 @@ def sync_detailed( subgraph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubgraphResponse]]: """Get Subgraph Details @@ -123,10 +101,8 @@ def sync_detailed( - Schema information Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -139,8 +115,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, subgraph_id=subgraph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -155,8 +129,6 @@ def sync( subgraph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubgraphResponse]]: """Get Subgraph Details @@ -181,10 +153,8 @@ def sync( - Schema information Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -198,8 +168,6 @@ def sync( graph_id=graph_id, subgraph_id=subgraph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -208,8 +176,6 @@ async def asyncio_detailed( subgraph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubgraphResponse]]: """Get Subgraph Details @@ -234,10 +200,8 @@ async def asyncio_detailed( - Schema information Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -250,8 +214,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, subgraph_id=subgraph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -264,8 +226,6 @@ async def asyncio( subgraph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubgraphResponse]]: """Get Subgraph Details @@ -290,10 +250,8 @@ async def asyncio( - Schema information Args: - graph_id (str): Parent graph identifier + graph_id (str): subgraph_id (str): Subgraph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -308,7 +266,5 @@ async def asyncio( graph_id=graph_id, subgraph_id=subgraph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/subgraphs/get_subgraph_quota.py b/robosystems_client/api/subgraphs/get_subgraph_quota.py index 957b13b..0d8cd2b 100644 --- a/robosystems_client/api/subgraphs/get_subgraph_quota.py +++ b/robosystems_client/api/subgraphs/get_subgraph_quota.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.http_validation_error import HTTPValidationError from ...models.subgraph_quota_response import SubgraphQuotaResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/subgraphs/quota", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -91,8 +71,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]: """Get Subgraph Quota @@ -115,9 +93,7 @@ def sync_detailed( Individual subgraph sizes shown in list endpoint. Args: - graph_id (str): Parent graph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -129,8 +105,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -144,8 +118,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]: """Get Subgraph Quota @@ -168,9 +140,7 @@ def sync( Individual subgraph sizes shown in list endpoint. Args: - graph_id (str): Parent graph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -183,8 +153,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -192,8 +160,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]: """Get Subgraph Quota @@ -216,9 +182,7 @@ async def asyncio_detailed( Individual subgraph sizes shown in list endpoint. Args: - graph_id (str): Parent graph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -230,8 +194,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -243,8 +205,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]: """Get Subgraph Quota @@ -267,9 +227,7 @@ async def asyncio( Individual subgraph sizes shown in list endpoint. Args: - graph_id (str): Parent graph identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -283,7 +241,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/subgraphs/list_subgraphs.py b/robosystems_client/api/subgraphs/list_subgraphs.py index 4c36721..7e06b09 100644 --- a/robosystems_client/api/subgraphs/list_subgraphs.py +++ b/robosystems_client/api/subgraphs/list_subgraphs.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.http_validation_error import HTTPValidationError from ...models.list_subgraphs_response import ListSubgraphsResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/subgraphs", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -75,8 +55,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, ListSubgraphsResponse]]: """List Subgraphs @@ -92,9 +70,7 @@ def sync_detailed( - Each subgraph includes its ID, name, description, type, status, and creation date Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -106,8 +82,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -121,8 +95,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, ListSubgraphsResponse]]: """List Subgraphs @@ -138,9 +110,7 @@ def sync( - Each subgraph includes its ID, name, description, type, status, and creation date Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -153,8 +123,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -162,8 +130,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, ListSubgraphsResponse]]: """List Subgraphs @@ -179,9 +145,7 @@ async def asyncio_detailed( - Each subgraph includes its ID, name, description, type, status, and creation date Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -193,8 +157,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -206,8 +168,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, ListSubgraphsResponse]]: """List Subgraphs @@ -223,9 +183,7 @@ async def asyncio( - Each subgraph includes its ID, name, description, type, status, and creation date Args: - graph_id (str): Parent graph ID (e.g., 'kg1a2b3c4d5') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -239,7 +197,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/delete_file.py b/robosystems_client/api/tables/delete_file.py index 408f86a..a9d6083 100644 --- a/robosystems_client/api/tables/delete_file.py +++ b/robosystems_client/api/tables/delete_file.py @@ -8,38 +8,18 @@ from ...models.delete_file_response import DeleteFileResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, file_id: str, - *, - 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": "delete", "url": f"/v1/graphs/{graph_id}/tables/files/{file_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -96,8 +76,6 @@ def sync_detailed( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: r""" Delete File from Staging @@ -153,10 +131,8 @@ def sync_detailed( File deletion is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -169,8 +145,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, file_id=file_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -185,8 +159,6 @@ def sync( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: r""" Delete File from Staging @@ -242,10 +214,8 @@ def sync( File deletion is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -259,8 +229,6 @@ def sync( graph_id=graph_id, file_id=file_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -269,8 +237,6 @@ async def asyncio_detailed( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: r""" Delete File from Staging @@ -326,10 +292,8 @@ async def asyncio_detailed( File deletion is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -342,8 +306,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, file_id=file_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -356,8 +318,6 @@ async def asyncio( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: r""" Delete File from Staging @@ -413,10 +373,8 @@ async def asyncio( File deletion is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -431,7 +389,5 @@ async def asyncio( graph_id=graph_id, file_id=file_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/get_file_info.py b/robosystems_client/api/tables/get_file_info.py index 557134f..c148c52 100644 --- a/robosystems_client/api/tables/get_file_info.py +++ b/robosystems_client/api/tables/get_file_info.py @@ -8,38 +8,18 @@ from ...models.error_response import ErrorResponse from ...models.get_file_info_response import GetFileInfoResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, file_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/tables/files/{file_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -92,8 +72,6 @@ def sync_detailed( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: r""" Get File Information @@ -140,10 +118,8 @@ def sync_detailed( File info retrieval is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -156,8 +132,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, file_id=file_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -172,8 +146,6 @@ def sync( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: r""" Get File Information @@ -220,10 +192,8 @@ def sync( File info retrieval is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -237,8 +207,6 @@ def sync( graph_id=graph_id, file_id=file_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -247,8 +215,6 @@ async def asyncio_detailed( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: r""" Get File Information @@ -295,10 +261,8 @@ async def asyncio_detailed( File info retrieval is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -311,8 +275,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, file_id=file_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -325,8 +287,6 @@ async def asyncio( file_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: r""" Get File Information @@ -373,10 +333,8 @@ async def asyncio( File info retrieval is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File ID - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -391,7 +349,5 @@ async def asyncio( graph_id=graph_id, file_id=file_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/get_upload_url.py b/robosystems_client/api/tables/get_upload_url.py index c778acb..065db74 100644 --- a/robosystems_client/api/tables/get_upload_url.py +++ b/robosystems_client/api/tables/get_upload_url.py @@ -9,7 +9,7 @@ from ...models.file_upload_request import FileUploadRequest from ...models.file_upload_response import FileUploadResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -17,28 +17,12 @@ def _get_kwargs( table_name: str, *, body: FileUploadRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/tables/{table_name}/files", - "params": params, } _kwargs["json"] = body.to_dict() @@ -108,8 +92,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: FileUploadRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: r""" Get File Upload URL @@ -187,10 +169,8 @@ def sync_detailed( Upload URL generation is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileUploadRequest): Raises: @@ -205,8 +185,6 @@ def sync_detailed( graph_id=graph_id, table_name=table_name, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -222,8 +200,6 @@ def sync( *, client: AuthenticatedClient, body: FileUploadRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: r""" Get File Upload URL @@ -301,10 +277,8 @@ def sync( Upload URL generation is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileUploadRequest): Raises: @@ -320,8 +294,6 @@ def sync( table_name=table_name, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -331,8 +303,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: FileUploadRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: r""" Get File Upload URL @@ -410,10 +380,8 @@ async def asyncio_detailed( Upload URL generation is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileUploadRequest): Raises: @@ -428,8 +396,6 @@ async def asyncio_detailed( graph_id=graph_id, table_name=table_name, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -443,8 +409,6 @@ async def asyncio( *, client: AuthenticatedClient, body: FileUploadRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: r""" Get File Upload URL @@ -522,10 +486,8 @@ async def asyncio( Upload URL generation is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileUploadRequest): Raises: @@ -542,7 +504,5 @@ async def asyncio( table_name=table_name, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/ingest_tables.py b/robosystems_client/api/tables/ingest_tables.py index 4cd04ed..fe9d890 100644 --- a/robosystems_client/api/tables/ingest_tables.py +++ b/robosystems_client/api/tables/ingest_tables.py @@ -9,35 +9,19 @@ from ...models.bulk_ingest_response import BulkIngestResponse from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: BulkIngestRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/tables/ingest", - "params": params, } _kwargs["json"] = body.to_dict() @@ -107,8 +91,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: BulkIngestRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: r""" Ingest Tables to Graph @@ -206,9 +188,7 @@ def sync_detailed( Table ingestion is included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BulkIngestRequest): Raises: @@ -222,8 +202,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -238,8 +216,6 @@ def sync( *, client: AuthenticatedClient, body: BulkIngestRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: r""" Ingest Tables to Graph @@ -337,9 +313,7 @@ def sync( Table ingestion is included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BulkIngestRequest): Raises: @@ -354,8 +328,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -364,8 +336,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: BulkIngestRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: r""" Ingest Tables to Graph @@ -463,9 +433,7 @@ async def asyncio_detailed( Table ingestion is included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BulkIngestRequest): Raises: @@ -479,8 +447,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -493,8 +459,6 @@ async def asyncio( *, client: AuthenticatedClient, body: BulkIngestRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: r""" Ingest Tables to Graph @@ -592,9 +556,7 @@ async def asyncio( Table ingestion is included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (BulkIngestRequest): Raises: @@ -610,7 +572,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/list_table_files.py b/robosystems_client/api/tables/list_table_files.py index f969357..2a408f6 100644 --- a/robosystems_client/api/tables/list_table_files.py +++ b/robosystems_client/api/tables/list_table_files.py @@ -8,38 +8,18 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.list_table_files_response import ListTableFilesResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, table_name: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/tables/{table_name}/files", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -96,8 +76,6 @@ def sync_detailed( table_name: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: r""" List Files in Staging Table @@ -171,10 +149,8 @@ def sync_detailed( File listing is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -187,8 +163,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, table_name=table_name, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -203,8 +177,6 @@ def sync( table_name: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: r""" List Files in Staging Table @@ -278,10 +250,8 @@ def sync( File listing is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -295,8 +265,6 @@ def sync( graph_id=graph_id, table_name=table_name, client=client, - token=token, - authorization=authorization, ).parsed @@ -305,8 +273,6 @@ async def asyncio_detailed( table_name: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: r""" List Files in Staging Table @@ -380,10 +346,8 @@ async def asyncio_detailed( File listing is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -396,8 +360,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, table_name=table_name, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -410,8 +372,6 @@ async def asyncio( table_name: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: r""" List Files in Staging Table @@ -485,10 +445,8 @@ async def asyncio( File listing is included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): table_name (str): Table name - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -503,7 +461,5 @@ async def asyncio( graph_id=graph_id, table_name=table_name, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/list_tables.py b/robosystems_client/api/tables/list_tables.py index 64255ee..72e90c8 100644 --- a/robosystems_client/api/tables/list_tables.py +++ b/robosystems_client/api/tables/list_tables.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.table_list_response import TableListResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, - *, - 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": "get", "url": f"/v1/graphs/{graph_id}/tables", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -94,8 +74,6 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: r""" List Staging Tables @@ -167,9 +145,7 @@ def sync_detailed( Table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -181,8 +157,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -196,8 +170,6 @@ def sync( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: r""" List Staging Tables @@ -269,9 +241,7 @@ def sync( Table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -284,8 +254,6 @@ def sync( return sync_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -293,8 +261,6 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: r""" List Staging Tables @@ -366,9 +332,7 @@ async def asyncio_detailed( Table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -380,8 +344,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -393,8 +355,6 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: r""" List Staging Tables @@ -466,9 +426,7 @@ async def asyncio( Table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -482,7 +440,5 @@ async def asyncio( await asyncio_detailed( graph_id=graph_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/query_tables.py b/robosystems_client/api/tables/query_tables.py index 9d07827..1a351ef 100644 --- a/robosystems_client/api/tables/query_tables.py +++ b/robosystems_client/api/tables/query_tables.py @@ -9,35 +9,19 @@ from ...models.http_validation_error import HTTPValidationError from ...models.table_query_request import TableQueryRequest from ...models.table_query_response import TableQueryResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( graph_id: str, *, body: TableQueryRequest, - 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": "post", "url": f"/v1/graphs/{graph_id}/tables/query", - "params": params, } _kwargs["json"] = body.to_dict() @@ -110,8 +94,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: TableQueryRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: """Query Staging Tables with SQL @@ -176,9 +158,7 @@ def sync_detailed( Staging table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (TableQueryRequest): Raises: @@ -192,8 +172,6 @@ def sync_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -208,8 +186,6 @@ def sync( *, client: AuthenticatedClient, body: TableQueryRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: """Query Staging Tables with SQL @@ -274,9 +250,7 @@ def sync( Staging table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (TableQueryRequest): Raises: @@ -291,8 +265,6 @@ def sync( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -301,8 +273,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: TableQueryRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: """Query Staging Tables with SQL @@ -367,9 +337,7 @@ async def asyncio_detailed( Staging table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (TableQueryRequest): Raises: @@ -383,8 +351,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( graph_id=graph_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -397,8 +363,6 @@ async def asyncio( *, client: AuthenticatedClient, body: TableQueryRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: """Query Staging Tables with SQL @@ -463,9 +427,7 @@ async def asyncio( Staging table queries are included - no credit consumption. Args: - graph_id (str): Graph database identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): + graph_id (str): body (TableQueryRequest): Raises: @@ -481,7 +443,5 @@ async def asyncio( graph_id=graph_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/tables/update_file_status.py b/robosystems_client/api/tables/update_file_status.py index a0d0ad7..7d4fa4f 100644 --- a/robosystems_client/api/tables/update_file_status.py +++ b/robosystems_client/api/tables/update_file_status.py @@ -11,7 +11,7 @@ from ...models.update_file_status_response_updatefilestatus import ( UpdateFileStatusResponseUpdatefilestatus, ) -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( @@ -19,28 +19,12 @@ def _get_kwargs( file_id: str, *, body: FileStatusUpdate, - 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": "patch", "url": f"/v1/graphs/{graph_id}/tables/files/{file_id}", - "params": params, } _kwargs["json"] = body.to_dict() @@ -123,8 +107,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: FileStatusUpdate, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus @@ -196,10 +178,8 @@ def sync_detailed( Status updates are included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileStatusUpdate): Raises: @@ -214,8 +194,6 @@ def sync_detailed( graph_id=graph_id, file_id=file_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -231,8 +209,6 @@ def sync( *, client: AuthenticatedClient, body: FileStatusUpdate, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus @@ -304,10 +280,8 @@ def sync( Status updates are included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileStatusUpdate): Raises: @@ -323,8 +297,6 @@ def sync( file_id=file_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -334,8 +306,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: FileStatusUpdate, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[ Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus @@ -407,10 +377,8 @@ async def asyncio_detailed( Status updates are included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileStatusUpdate): Raises: @@ -425,8 +393,6 @@ async def asyncio_detailed( graph_id=graph_id, file_id=file_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -440,8 +406,6 @@ async def asyncio( *, client: AuthenticatedClient, body: FileStatusUpdate, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[ Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus @@ -513,10 +477,8 @@ async def asyncio( Status updates are included - no credit consumption. Args: - graph_id (str): Graph database identifier + graph_id (str): file_id (str): File identifier - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (FileStatusUpdate): Raises: @@ -533,7 +495,5 @@ async def asyncio( file_id=file_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/create_user_api_key.py b/robosystems_client/api/user/create_user_api_key.py index 5404825..806835c 100644 --- a/robosystems_client/api/user/create_user_api_key.py +++ b/robosystems_client/api/user/create_user_api_key.py @@ -8,34 +8,18 @@ from ...models.create_api_key_request import CreateAPIKeyRequest from ...models.create_api_key_response import CreateAPIKeyResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( *, body: CreateAPIKeyRequest, - 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": "post", "url": "/v1/user/api-keys", - "params": params, } _kwargs["json"] = body.to_dict() @@ -80,16 +64,12 @@ def sync_detailed( *, client: AuthenticatedClient, body: CreateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[CreateAPIKeyResponse, HTTPValidationError]]: """Create API Key Create a new API key for the current user. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CreateAPIKeyRequest): Request model for creating a new API key. Raises: @@ -102,8 +82,6 @@ def sync_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -117,16 +95,12 @@ def sync( *, client: AuthenticatedClient, body: CreateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[CreateAPIKeyResponse, HTTPValidationError]]: """Create API Key Create a new API key for the current user. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CreateAPIKeyRequest): Request model for creating a new API key. Raises: @@ -140,8 +114,6 @@ def sync( return sync_detailed( client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -149,16 +121,12 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: CreateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[CreateAPIKeyResponse, HTTPValidationError]]: """Create API Key Create a new API key for the current user. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CreateAPIKeyRequest): Request model for creating a new API key. Raises: @@ -171,8 +139,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -184,16 +150,12 @@ async def asyncio( *, client: AuthenticatedClient, body: CreateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[CreateAPIKeyResponse, HTTPValidationError]]: """Create API Key Create a new API key for the current user. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (CreateAPIKeyRequest): Request model for creating a new API key. Raises: @@ -208,7 +170,5 @@ async def asyncio( await asyncio_detailed( client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/get_all_credit_summaries.py b/robosystems_client/api/user/get_all_credit_summaries.py index 7d8a6eb..cb37909 100644 --- a/robosystems_client/api/user/get_all_credit_summaries.py +++ b/robosystems_client/api/user/get_all_credit_summaries.py @@ -9,49 +9,21 @@ from ...models.get_all_credit_summaries_response_getallcreditsummaries import ( GetAllCreditSummariesResponseGetallcreditsummaries, ) -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/credits", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Optional[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: if response.status_code == 200: response_200 = GetAllCreditSummariesResponseGetallcreditsummaries.from_dict( response.json() @@ -59,11 +31,6 @@ def _parse_response( return response_200 - 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()) @@ -77,13 +44,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -95,15 +56,7 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: """Get All Credit Summaries Get credit summaries for all graphs owned by the user. @@ -114,22 +67,15 @@ def sync_detailed( No credits are consumed for viewing summaries. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries, HTTPValidationError]] + Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -141,15 +87,7 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Optional[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: """Get All Credit Summaries Get credit summaries for all graphs owned by the user. @@ -160,37 +98,23 @@ def sync( No credits are consumed for viewing summaries. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries, HTTPValidationError] + Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries] """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: """Get All Credit Summaries Get credit summaries for all graphs owned by the user. @@ -201,22 +125,15 @@ async def asyncio_detailed( No credits are consumed for viewing summaries. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries, HTTPValidationError]] + Response[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -226,15 +143,7 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - ErrorResponse, - GetAllCreditSummariesResponseGetallcreditsummaries, - HTTPValidationError, - ] -]: +) -> Optional[Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries]]: """Get All Credit Summaries Get credit summaries for all graphs owned by the user. @@ -245,22 +154,16 @@ async def asyncio( No credits are consumed for viewing summaries. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries, HTTPValidationError] + Union[ErrorResponse, GetAllCreditSummariesResponseGetallcreditsummaries] """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/get_current_user.py b/robosystems_client/api/user/get_current_user.py index bb6ae60..a7fc919 100644 --- a/robosystems_client/api/user/get_current_user.py +++ b/robosystems_client/api/user/get_current_user.py @@ -5,54 +5,27 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.user_response import UserResponse -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserResponse]]: +) -> Optional[UserResponse]: if response.status_code == 200: response_200 = UserResponse.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: @@ -61,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, UserResponse]]: +) -> Response[UserResponse]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,29 +46,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserResponse]]: +) -> Response[UserResponse]: """Get Current User Returns information about the currently authenticated user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserResponse]] + Response[UserResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -107,58 +71,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserResponse]]: +) -> Optional[UserResponse]: """Get Current User Returns information about the currently authenticated user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserResponse] + UserResponse """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserResponse]]: +) -> Response[UserResponse]: """Get Current User Returns information about the currently authenticated user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserResponse]] + Response[UserResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -168,29 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserResponse]]: +) -> Optional[UserResponse]: """Get Current User Returns information about the currently authenticated user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserResponse] + UserResponse """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/list_user_api_keys.py b/robosystems_client/api/user/list_user_api_keys.py index cfa0e11..f230b12 100644 --- a/robosystems_client/api/user/list_user_api_keys.py +++ b/robosystems_client/api/user/list_user_api_keys.py @@ -6,53 +6,26 @@ from ... import errors from ...client import AuthenticatedClient, Client from ...models.api_keys_response import APIKeysResponse -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/api-keys", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[APIKeysResponse, HTTPValidationError]]: +) -> Optional[APIKeysResponse]: if response.status_code == 200: response_200 = APIKeysResponse.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: @@ -61,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[APIKeysResponse, HTTPValidationError]]: +) -> Response[APIKeysResponse]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,29 +46,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[APIKeysResponse, HTTPValidationError]]: +) -> Response[APIKeysResponse]: """List API Keys Get all API keys for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[APIKeysResponse, HTTPValidationError]] + Response[APIKeysResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -107,58 +71,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[APIKeysResponse, HTTPValidationError]]: +) -> Optional[APIKeysResponse]: """List API Keys Get all API keys for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[APIKeysResponse, HTTPValidationError] + APIKeysResponse """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[APIKeysResponse, HTTPValidationError]]: +) -> Response[APIKeysResponse]: """List API Keys Get all API keys for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[APIKeysResponse, HTTPValidationError]] + Response[APIKeysResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -168,29 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[APIKeysResponse, HTTPValidationError]]: +) -> Optional[APIKeysResponse]: """List API Keys Get all API keys for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[APIKeysResponse, HTTPValidationError] + APIKeysResponse """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/revoke_user_api_key.py b/robosystems_client/api/user/revoke_user_api_key.py index 198c233..698bf20 100644 --- a/robosystems_client/api/user/revoke_user_api_key.py +++ b/robosystems_client/api/user/revoke_user_api_key.py @@ -8,37 +8,17 @@ from ...models.error_response import ErrorResponse from ...models.http_validation_error import HTTPValidationError from ...models.success_response import SuccessResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( api_key_id: str, - *, - 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": "delete", "url": f"/v1/user/api-keys/{api_key_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -86,8 +66,6 @@ def sync_detailed( api_key_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Revoke API Key @@ -95,8 +73,6 @@ def sync_detailed( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -108,8 +84,6 @@ def sync_detailed( kwargs = _get_kwargs( api_key_id=api_key_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -123,8 +97,6 @@ def sync( api_key_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Revoke API Key @@ -132,8 +104,6 @@ def sync( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -146,8 +116,6 @@ def sync( return sync_detailed( api_key_id=api_key_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -155,8 +123,6 @@ async def asyncio_detailed( api_key_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Revoke API Key @@ -164,8 +130,6 @@ async def asyncio_detailed( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -177,8 +141,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( api_key_id=api_key_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -190,8 +152,6 @@ async def asyncio( api_key_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Revoke API Key @@ -199,8 +159,6 @@ async def asyncio( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -214,7 +172,5 @@ async def asyncio( await asyncio_detailed( api_key_id=api_key_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/update_user.py b/robosystems_client/api/user/update_user.py index 77b9405..e5b7cad 100644 --- a/robosystems_client/api/user/update_user.py +++ b/robosystems_client/api/user/update_user.py @@ -8,34 +8,18 @@ from ...models.http_validation_error import HTTPValidationError from ...models.update_user_request import UpdateUserRequest from ...models.user_response import UserResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( *, body: UpdateUserRequest, - 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": "/v1/user", - "params": params, } _kwargs["json"] = body.to_dict() @@ -80,16 +64,12 @@ def sync_detailed( *, client: AuthenticatedClient, body: UpdateUserRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, UserResponse]]: """Update User Profile Update the current user's profile information. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateUserRequest): Request model for updating user profile. Raises: @@ -102,8 +82,6 @@ def sync_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -117,16 +95,12 @@ def sync( *, client: AuthenticatedClient, body: UpdateUserRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, UserResponse]]: """Update User Profile Update the current user's profile information. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateUserRequest): Request model for updating user profile. Raises: @@ -140,8 +114,6 @@ def sync( return sync_detailed( client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -149,16 +121,12 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: UpdateUserRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, UserResponse]]: """Update User Profile Update the current user's profile information. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateUserRequest): Request model for updating user profile. Raises: @@ -171,8 +139,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -184,16 +150,12 @@ async def asyncio( *, client: AuthenticatedClient, body: UpdateUserRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, UserResponse]]: """Update User Profile Update the current user's profile information. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateUserRequest): Request model for updating user profile. Raises: @@ -208,7 +170,5 @@ async def asyncio( await asyncio_detailed( client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/update_user_api_key.py b/robosystems_client/api/user/update_user_api_key.py index 37e7009..a9450d2 100644 --- a/robosystems_client/api/user/update_user_api_key.py +++ b/robosystems_client/api/user/update_user_api_key.py @@ -8,35 +8,19 @@ from ...models.api_key_info import APIKeyInfo from ...models.http_validation_error import HTTPValidationError from ...models.update_api_key_request import UpdateAPIKeyRequest -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( api_key_id: str, *, body: UpdateAPIKeyRequest, - 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/api-keys/{api_key_id}", - "params": params, } _kwargs["json"] = body.to_dict() @@ -82,8 +66,6 @@ def sync_detailed( *, client: AuthenticatedClient, body: UpdateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[APIKeyInfo, HTTPValidationError]]: """Update API Key @@ -91,8 +73,6 @@ def sync_detailed( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateAPIKeyRequest): Request model for updating an API key. Raises: @@ -106,8 +86,6 @@ def sync_detailed( kwargs = _get_kwargs( api_key_id=api_key_id, body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -122,8 +100,6 @@ def sync( *, client: AuthenticatedClient, body: UpdateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[APIKeyInfo, HTTPValidationError]]: """Update API Key @@ -131,8 +107,6 @@ def sync( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateAPIKeyRequest): Request model for updating an API key. Raises: @@ -147,8 +121,6 @@ def sync( api_key_id=api_key_id, client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -157,8 +129,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: UpdateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[APIKeyInfo, HTTPValidationError]]: """Update API Key @@ -166,8 +136,6 @@ async def asyncio_detailed( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateAPIKeyRequest): Request model for updating an API key. Raises: @@ -181,8 +149,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( api_key_id=api_key_id, body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -195,8 +161,6 @@ async def asyncio( *, client: AuthenticatedClient, body: UpdateAPIKeyRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[APIKeyInfo, HTTPValidationError]]: """Update API Key @@ -204,8 +168,6 @@ async def asyncio( Args: api_key_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdateAPIKeyRequest): Request model for updating an API key. Raises: @@ -221,7 +183,5 @@ async def asyncio( api_key_id=api_key_id, client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user/update_user_password.py b/robosystems_client/api/user/update_user_password.py index b39a29f..c39aca2 100644 --- a/robosystems_client/api/user/update_user_password.py +++ b/robosystems_client/api/user/update_user_password.py @@ -9,34 +9,18 @@ from ...models.http_validation_error import HTTPValidationError from ...models.success_response import SuccessResponse from ...models.update_password_request import UpdatePasswordRequest -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( *, body: UpdatePasswordRequest, - 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": "/v1/user/password", - "params": params, } _kwargs["json"] = body.to_dict() @@ -96,16 +80,12 @@ def sync_detailed( *, client: AuthenticatedClient, body: UpdatePasswordRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Update Password Update the current user's password. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdatePasswordRequest): Request model for updating user password. Raises: @@ -118,8 +98,6 @@ def sync_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -133,16 +111,12 @@ def sync( *, client: AuthenticatedClient, body: UpdatePasswordRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Update Password Update the current user's password. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdatePasswordRequest): Request model for updating user password. Raises: @@ -156,8 +130,6 @@ def sync( return sync_detailed( client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -165,16 +137,12 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: UpdatePasswordRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Update Password Update the current user's password. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdatePasswordRequest): Request model for updating user password. Raises: @@ -187,8 +155,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -200,16 +166,12 @@ async def asyncio( *, client: AuthenticatedClient, body: UpdatePasswordRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Update Password Update the current user's password. Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (UpdatePasswordRequest): Request model for updating user password. Raises: @@ -224,7 +186,5 @@ async def asyncio( await asyncio_detailed( client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_analytics/get_detailed_user_analytics.py b/robosystems_client/api/user_analytics/get_detailed_user_analytics.py index 47b4c30..0f83166 100644 --- a/robosystems_client/api/user_analytics/get_detailed_user_analytics.py +++ b/robosystems_client/api/user_analytics/get_detailed_user_analytics.py @@ -14,26 +14,13 @@ def _get_kwargs( *, include_api_stats: Union[Unset, bool] = True, include_recent_activity: Union[Unset, bool] = True, - 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] = {} params["include_api_stats"] = include_api_stats params["include_recent_activity"] = include_recent_activity - 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] = { @@ -42,7 +29,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -81,8 +67,6 @@ def sync_detailed( client: AuthenticatedClient, include_api_stats: Union[Unset, bool] = True, include_recent_activity: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, UserAnalyticsResponse]]: """Get Detailed User Analytics @@ -91,8 +75,6 @@ def sync_detailed( Args: include_api_stats (Union[Unset, bool]): Include API usage statistics Default: True. include_recent_activity (Union[Unset, bool]): Include recent activity Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -105,8 +87,6 @@ def sync_detailed( kwargs = _get_kwargs( include_api_stats=include_api_stats, include_recent_activity=include_recent_activity, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -121,8 +101,6 @@ def sync( client: AuthenticatedClient, include_api_stats: Union[Unset, bool] = True, include_recent_activity: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, UserAnalyticsResponse]]: """Get Detailed User Analytics @@ -131,8 +109,6 @@ def sync( Args: include_api_stats (Union[Unset, bool]): Include API usage statistics Default: True. include_recent_activity (Union[Unset, bool]): Include recent activity Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -146,8 +122,6 @@ def sync( client=client, include_api_stats=include_api_stats, include_recent_activity=include_recent_activity, - token=token, - authorization=authorization, ).parsed @@ -156,8 +130,6 @@ async def asyncio_detailed( client: AuthenticatedClient, include_api_stats: Union[Unset, bool] = True, include_recent_activity: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[HTTPValidationError, UserAnalyticsResponse]]: """Get Detailed User Analytics @@ -166,8 +138,6 @@ async def asyncio_detailed( Args: include_api_stats (Union[Unset, bool]): Include API usage statistics Default: True. include_recent_activity (Union[Unset, bool]): Include recent activity Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -180,8 +150,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( include_api_stats=include_api_stats, include_recent_activity=include_recent_activity, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -194,8 +162,6 @@ async def asyncio( client: AuthenticatedClient, include_api_stats: Union[Unset, bool] = True, include_recent_activity: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[HTTPValidationError, UserAnalyticsResponse]]: """Get Detailed User Analytics @@ -204,8 +170,6 @@ async def asyncio( Args: include_api_stats (Union[Unset, bool]): Include API usage statistics Default: True. include_recent_activity (Union[Unset, bool]): Include recent activity Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -220,7 +184,5 @@ async def asyncio( client=client, include_api_stats=include_api_stats, include_recent_activity=include_recent_activity, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_analytics/get_user_usage_overview.py b/robosystems_client/api/user_analytics/get_user_usage_overview.py index 4977a53..5776822 100644 --- a/robosystems_client/api/user_analytics/get_user_usage_overview.py +++ b/robosystems_client/api/user_analytics/get_user_usage_overview.py @@ -5,54 +5,27 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.user_usage_summary_response import UserUsageSummaryResponse -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/analytics/overview", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Optional[UserUsageSummaryResponse]: if response.status_code == 200: 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: @@ -61,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Response[UserUsageSummaryResponse]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,29 +46,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Response[UserUsageSummaryResponse]: """Get User Usage Overview Get a high-level overview of usage statistics for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserUsageSummaryResponse]] + Response[UserUsageSummaryResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -107,58 +71,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Optional[UserUsageSummaryResponse]: """Get User Usage Overview Get a high-level overview of usage statistics for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserUsageSummaryResponse] + UserUsageSummaryResponse """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Response[UserUsageSummaryResponse]: """Get User Usage Overview Get a high-level overview of usage statistics for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserUsageSummaryResponse]] + Response[UserUsageSummaryResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -168,29 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserUsageSummaryResponse]]: +) -> Optional[UserUsageSummaryResponse]: """Get User Usage Overview Get a high-level overview of usage statistics for the current user. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserUsageSummaryResponse] + UserUsageSummaryResponse """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_limits/get_all_shared_repository_limits.py b/robosystems_client/api/user_limits/get_all_shared_repository_limits.py index 9e0f1d0..ccd711b 100644 --- a/robosystems_client/api/user_limits/get_all_shared_repository_limits.py +++ b/robosystems_client/api/user_limits/get_all_shared_repository_limits.py @@ -8,48 +8,21 @@ from ...models.get_all_shared_repository_limits_response_getallsharedrepositorylimits import ( GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, ) -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/limits/shared-repositories/summary", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Optional[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: if response.status_code == 200: response_200 = ( GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits.from_dict( @@ -59,11 +32,6 @@ 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: @@ -72,12 +40,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Response[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -89,34 +52,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Response[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: """Get all shared repository limits Get rate limit status for all shared repositories the user has access to. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, HTTPValidationError]] + Response[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -128,68 +77,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Optional[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: """Get all shared repository limits Get rate limit status for all shared repositories the user has access to. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, HTTPValidationError] + GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Response[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: """Get all shared repository limits Get rate limit status for all shared repositories the user has access to. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, HTTPValidationError]] + Response[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -199,34 +121,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[ - Union[ - GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, - HTTPValidationError, - ] -]: +) -> Optional[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits]: """Get all shared repository limits Get rate limit status for all shared repositories the user has access to. - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits, HTTPValidationError] + GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_limits/get_shared_repository_limits.py b/robosystems_client/api/user_limits/get_shared_repository_limits.py index f1dd015..45d3d52 100644 --- a/robosystems_client/api/user_limits/get_shared_repository_limits.py +++ b/robosystems_client/api/user_limits/get_shared_repository_limits.py @@ -9,37 +9,17 @@ GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, ) from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( repository: str, - *, - 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": "get", "url": f"/v1/user/limits/shared-repositories/{repository}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -83,8 +63,6 @@ def sync_detailed( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError] ]: @@ -102,8 +80,6 @@ def sync_detailed( Args: repository (str): Repository name (e.g., 'sec') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -115,8 +91,6 @@ def sync_detailed( kwargs = _get_kwargs( repository=repository, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -130,8 +104,6 @@ def sync( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError] ]: @@ -149,8 +121,6 @@ def sync( Args: repository (str): Repository name (e.g., 'sec') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -163,8 +133,6 @@ def sync( return sync_detailed( repository=repository, client=client, - token=token, - authorization=authorization, ).parsed @@ -172,8 +140,6 @@ async def asyncio_detailed( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[ Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError] ]: @@ -191,8 +157,6 @@ async def asyncio_detailed( Args: repository (str): Repository name (e.g., 'sec') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -204,8 +168,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( repository=repository, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -217,8 +179,6 @@ async def asyncio( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[ Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError] ]: @@ -236,8 +196,6 @@ async def asyncio( Args: repository (str): Repository name (e.g., 'sec') - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -251,7 +209,5 @@ async def asyncio( await asyncio_detailed( repository=repository, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_limits/get_user_limits.py b/robosystems_client/api/user_limits/get_user_limits.py index 76f3b40..a44e004 100644 --- a/robosystems_client/api/user_limits/get_user_limits.py +++ b/robosystems_client/api/user_limits/get_user_limits.py @@ -5,44 +5,22 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.user_limits_response import UserLimitsResponse -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/limits", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Optional[Union[Any, UserLimitsResponse]]: if response.status_code == 200: response_200 = UserLimitsResponse.from_dict(response.json()) @@ -52,11 +30,6 @@ def _parse_response( 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: @@ -65,7 +38,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Response[Union[Any, UserLimitsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -77,29 +50,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Response[Union[Any, UserLimitsResponse]]: """Get user limits Retrieve current limits and restrictions for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, HTTPValidationError, UserLimitsResponse]] + Response[Union[Any, UserLimitsResponse]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -111,58 +75,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Optional[Union[Any, UserLimitsResponse]]: """Get user limits Retrieve current limits and restrictions for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, HTTPValidationError, UserLimitsResponse] + Union[Any, UserLimitsResponse] """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Response[Union[Any, UserLimitsResponse]]: """Get user limits Retrieve current limits and restrictions for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, HTTPValidationError, UserLimitsResponse]] + Response[Union[Any, UserLimitsResponse]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -172,29 +119,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[Any, HTTPValidationError, UserLimitsResponse]]: +) -> Optional[Union[Any, UserLimitsResponse]]: """Get user limits Retrieve current limits and restrictions for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, HTTPValidationError, UserLimitsResponse] + Union[Any, UserLimitsResponse] """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_limits/get_user_usage.py b/robosystems_client/api/user_limits/get_user_usage.py index 1d57807..8efbdaa 100644 --- a/robosystems_client/api/user_limits/get_user_usage.py +++ b/robosystems_client/api/user_limits/get_user_usage.py @@ -5,54 +5,27 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.http_validation_error import HTTPValidationError from ...models.user_usage_response import UserUsageResponse -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/limits/usage", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserUsageResponse]]: +) -> Optional[UserUsageResponse]: if response.status_code == 200: 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: @@ -61,7 +34,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, UserUsageResponse]]: +) -> Response[UserUsageResponse]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,29 +46,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserUsageResponse]]: +) -> Response[UserUsageResponse]: """Get user usage statistics Retrieve current usage statistics and remaining limits for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserUsageResponse]] + Response[UserUsageResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -107,58 +71,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserUsageResponse]]: +) -> Optional[UserUsageResponse]: """Get user usage statistics Retrieve current usage statistics and remaining limits for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserUsageResponse] + UserUsageResponse """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[HTTPValidationError, UserUsageResponse]]: +) -> Response[UserUsageResponse]: """Get user usage statistics Retrieve current usage statistics and remaining limits for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, UserUsageResponse]] + Response[UserUsageResponse] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -168,29 +115,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, UserUsageResponse]]: +) -> Optional[UserUsageResponse]: """Get user usage statistics Retrieve current usage statistics and remaining limits for the authenticated user - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, UserUsageResponse] + UserUsageResponse """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py b/robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py index 1b7eb55..5f8080c 100644 --- a/robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +++ b/robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.cancellation_response import CancellationResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( subscription_id: str, - *, - 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": "delete", "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -87,8 +67,6 @@ def sync_detailed( subscription_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, CancellationResponse, HTTPValidationError]]: """Cancel Subscription @@ -96,8 +74,6 @@ def sync_detailed( Args: subscription_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -109,8 +85,6 @@ def sync_detailed( kwargs = _get_kwargs( subscription_id=subscription_id, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -124,8 +98,6 @@ def sync( subscription_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, CancellationResponse, HTTPValidationError]]: """Cancel Subscription @@ -133,8 +105,6 @@ def sync( Args: subscription_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -147,8 +117,6 @@ def sync( return sync_detailed( subscription_id=subscription_id, client=client, - token=token, - authorization=authorization, ).parsed @@ -156,8 +124,6 @@ async def asyncio_detailed( subscription_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, CancellationResponse, HTTPValidationError]]: """Cancel Subscription @@ -165,8 +131,6 @@ async def asyncio_detailed( Args: subscription_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -178,8 +142,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( subscription_id=subscription_id, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -191,8 +153,6 @@ async def asyncio( subscription_id: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, CancellationResponse, HTTPValidationError]]: """Cancel Subscription @@ -200,8 +160,6 @@ async def asyncio( Args: subscription_id (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -215,7 +173,5 @@ async def asyncio( await asyncio_detailed( subscription_id=subscription_id, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/get_repository_credits.py b/robosystems_client/api/user_subscriptions/get_repository_credits.py index 27929b1..e39036b 100644 --- a/robosystems_client/api/user_subscriptions/get_repository_credits.py +++ b/robosystems_client/api/user_subscriptions/get_repository_credits.py @@ -7,37 +7,17 @@ from ...client import AuthenticatedClient, Client from ...models.http_validation_error import HTTPValidationError from ...models.repository_credits_response import RepositoryCreditsResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( repository: str, - *, - 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": "get", "url": f"/v1/user/subscriptions/shared-repositories/credits/{repository}", - "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -83,8 +63,6 @@ def sync_detailed( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]: """Get Repository Credits @@ -92,8 +70,6 @@ def sync_detailed( Args: repository (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -105,8 +81,6 @@ def sync_detailed( kwargs = _get_kwargs( repository=repository, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -120,8 +94,6 @@ def sync( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]: """Get Repository Credits @@ -129,8 +101,6 @@ def sync( Args: repository (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -143,8 +113,6 @@ def sync( return sync_detailed( repository=repository, client=client, - token=token, - authorization=authorization, ).parsed @@ -152,8 +120,6 @@ async def asyncio_detailed( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]: """Get Repository Credits @@ -161,8 +127,6 @@ async def asyncio_detailed( Args: repository (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -174,8 +138,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( repository=repository, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -187,8 +149,6 @@ async def asyncio( repository: str, *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]: """Get Repository Credits @@ -196,8 +156,6 @@ async def asyncio( Args: repository (str): - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -211,7 +169,5 @@ async def asyncio( await asyncio_detailed( repository=repository, client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/get_shared_repository_credits.py b/robosystems_client/api/user_subscriptions/get_shared_repository_credits.py index 3879d2e..ccd2ca7 100644 --- a/robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +++ b/robosystems_client/api/user_subscriptions/get_shared_repository_credits.py @@ -6,43 +6,21 @@ from ... import errors from ...client import AuthenticatedClient, Client from ...models.credits_summary_response import CreditsSummaryResponse -from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import Response -def _get_kwargs( - *, - 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} - +def _get_kwargs() -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": "get", "url": "/v1/user/subscriptions/shared-repositories/credits", - "params": params, } - _kwargs["headers"] = headers return _kwargs def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Optional[Union[Any, CreditsSummaryResponse]]: if response.status_code == 200: response_200 = CreditsSummaryResponse.from_dict(response.json()) @@ -52,11 +30,6 @@ def _parse_response( response_401 = cast(Any, None) return response_401 - 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 @@ -69,7 +42,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Response[Union[Any, CreditsSummaryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -81,29 +54,20 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Response[Union[Any, CreditsSummaryResponse]]: """Get Credit Balances Retrieve credit balances for all shared repository subscriptions - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]] + Response[Union[Any, CreditsSummaryResponse]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = client.get_httpx_client().request( **kwargs, @@ -115,58 +79,41 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Optional[Union[Any, CreditsSummaryResponse]]: """Get Credit Balances Retrieve credit balances for all shared repository subscriptions - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, CreditsSummaryResponse, HTTPValidationError] + Union[Any, CreditsSummaryResponse] """ return sync_detailed( client=client, - token=token, - authorization=authorization, ).parsed async def asyncio_detailed( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Response[Union[Any, CreditsSummaryResponse]]: """Get Credit Balances Retrieve credit balances for all shared repository subscriptions - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]] + Response[Union[Any, CreditsSummaryResponse]] """ - kwargs = _get_kwargs( - token=token, - authorization=authorization, - ) + kwargs = _get_kwargs() response = await client.get_async_httpx_client().request(**kwargs) @@ -176,29 +123,21 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, -) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]: +) -> Optional[Union[Any, CreditsSummaryResponse]]: """Get Credit Balances Retrieve credit balances for all shared repository subscriptions - Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): - Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, CreditsSummaryResponse, HTTPValidationError] + Union[Any, CreditsSummaryResponse] """ return ( await asyncio_detailed( client=client, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py b/robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py index 6c612e5..440ee52 100644 --- a/robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +++ b/robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py @@ -13,24 +13,11 @@ def _get_kwargs( *, active_only: Union[Unset, bool] = True, - 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] = {} params["active_only"] = active_only - 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] = { @@ -39,7 +26,6 @@ def _get_kwargs( "params": params, } - _kwargs["headers"] = headers return _kwargs @@ -85,8 +71,6 @@ def sync_detailed( *, client: AuthenticatedClient, active_only: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]: """Get User Subscriptions @@ -94,8 +78,6 @@ def sync_detailed( Args: active_only (Union[Unset, bool]): Only return active subscriptions Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -107,8 +89,6 @@ def sync_detailed( kwargs = _get_kwargs( active_only=active_only, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -122,8 +102,6 @@ def sync( *, client: AuthenticatedClient, active_only: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]: """Get User Subscriptions @@ -131,8 +109,6 @@ def sync( Args: active_only (Union[Unset, bool]): Only return active subscriptions Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -145,8 +121,6 @@ def sync( return sync_detailed( client=client, active_only=active_only, - token=token, - authorization=authorization, ).parsed @@ -154,8 +128,6 @@ async def asyncio_detailed( *, client: AuthenticatedClient, active_only: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]: """Get User Subscriptions @@ -163,8 +135,6 @@ async def asyncio_detailed( Args: active_only (Union[Unset, bool]): Only return active subscriptions Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -176,8 +146,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( active_only=active_only, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -189,8 +157,6 @@ async def asyncio( *, client: AuthenticatedClient, active_only: Union[Unset, bool] = True, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]: """Get User Subscriptions @@ -198,8 +164,6 @@ async def asyncio( Args: active_only (Union[Unset, bool]): Only return active subscriptions Default: True. - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -213,7 +177,5 @@ async def asyncio( await asyncio_detailed( client=client, active_only=active_only, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py b/robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py index 89df698..5fa18db 100644 --- a/robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +++ b/robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py @@ -8,34 +8,18 @@ from ...models.http_validation_error import HTTPValidationError from ...models.subscription_request import SubscriptionRequest from ...models.subscription_response import SubscriptionResponse -from ...types import UNSET, Response, Unset +from ...types import Response def _get_kwargs( *, body: SubscriptionRequest, - 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": "post", "url": "/v1/user/subscriptions/shared-repositories/subscribe", - "params": params, } _kwargs["json"] = body.to_dict() @@ -92,16 +76,12 @@ def sync_detailed( *, client: AuthenticatedClient, body: SubscriptionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]: """Subscribe to Shared Repository Create a new subscription to a shared repository add-on with specified tier Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SubscriptionRequest): Request to create a new subscription. Raises: @@ -114,8 +94,6 @@ def sync_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = client.get_httpx_client().request( @@ -129,16 +107,12 @@ def sync( *, client: AuthenticatedClient, body: SubscriptionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]: """Subscribe to Shared Repository Create a new subscription to a shared repository add-on with specified tier Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SubscriptionRequest): Request to create a new subscription. Raises: @@ -152,8 +126,6 @@ def sync( return sync_detailed( client=client, body=body, - token=token, - authorization=authorization, ).parsed @@ -161,16 +133,12 @@ async def asyncio_detailed( *, client: AuthenticatedClient, body: SubscriptionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]: """Subscribe to Shared Repository Create a new subscription to a shared repository add-on with specified tier Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SubscriptionRequest): Request to create a new subscription. Raises: @@ -183,8 +151,6 @@ async def asyncio_detailed( kwargs = _get_kwargs( body=body, - token=token, - authorization=authorization, ) response = await client.get_async_httpx_client().request(**kwargs) @@ -196,16 +162,12 @@ async def asyncio( *, client: AuthenticatedClient, body: SubscriptionRequest, - token: Union[None, Unset, str] = UNSET, - authorization: Union[None, Unset, str] = UNSET, ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]: """Subscribe to Shared Repository Create a new subscription to a shared repository add-on with specified tier Args: - token (Union[None, Unset, str]): JWT token for SSE authentication - authorization (Union[None, Unset, str]): body (SubscriptionRequest): Request to create a new subscription. Raises: @@ -220,7 +182,5 @@ async def asyncio( await asyncio_detailed( client=client, body=body, - token=token, - authorization=authorization, ) ).parsed diff --git a/robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py b/robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py index f164970..fe861a7 100644 --- a/robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +++ b/robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py @@ -7,35 +7,19 @@ 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 diff --git a/robosystems_client/extensions/__init__.py b/robosystems_client/extensions/__init__.py index 718f86d..270f1d7 100644 --- a/robosystems_client/extensions/__init__.py +++ b/robosystems_client/extensions/__init__.py @@ -94,6 +94,7 @@ 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) diff --git a/robosystems_client/extensions/auth_integration.py b/robosystems_client/extensions/auth_integration.py index ca82f42..c4252ba 100644 --- a/robosystems_client/extensions/auth_integration.py +++ b/robosystems_client/extensions/auth_integration.py @@ -60,12 +60,11 @@ def execute_cypher_query( 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: diff --git a/robosystems_client/extensions/query_client.py b/robosystems_client/extensions/query_client.py index a064250..9dfb5d4 100644 --- a/robosystems_client/extensions/query_client.py +++ b/robosystems_client/extensions/query_client.py @@ -476,7 +476,7 @@ def query_batch( 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 @@ def query_batch( ... ]) """ 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") diff --git a/robosystems_client/extensions/sse_client.py b/robosystems_client/extensions/sse_client.py index 77e27c0..6251b51 100644 --- a/robosystems_client/extensions/sse_client.py +++ b/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: diff --git a/robosystems_client/extensions/table_ingest_client.py b/robosystems_client/extensions/table_ingest_client.py index d5f839f..de3ad27 100644 --- a/robosystems_client/extensions/table_ingest_client.py +++ b/robosystems_client/extensions/table_ingest_client.py @@ -114,6 +114,9 @@ def upload_parquet_file( 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 @@ def upload_parquet_file( 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() diff --git a/robosystems_client/extensions/utils.py b/robosystems_client/extensions/utils.py index a198a33..521942b 100644 --- a/robosystems_client/extensions/utils.py +++ b/robosystems_client/extensions/utils.py @@ -144,7 +144,7 @@ def to_csv( 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 @@ def to_csv( 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) From 2be86e064767b43cbb99f801e353250133c09b6b Mon Sep 17 00:00:00 2001 From: Joey French Date: Wed, 29 Oct 2025 00:34:03 -0500 Subject: [PATCH 2/4] Update API documentation and enhance file management features ## Summary This commit updates the API documentation for various endpoints and enhances file management functionalities, improving clarity and usability. ## Key Changes - Revised example requests and responses in the `create_graph`, `get_available_extensions`, `get_graphs`, and `select_graph` endpoints to reflect updated entity names and descriptions. - Enhanced documentation for file management operations, including `delete_file`, `get_file_info`, `get_upload_url`, and `update_file_status`, to provide clearer usage instructions and examples. - Improved the handling of file statuses and upload processes, ensuring better validation and error reporting. ## Testing Notes - Verify that the updated documentation accurately reflects the API behavior. - Test file management operations to ensure they function as expected with the new documentation. ## Infrastructure Considerations The documentation updates and enhancements to file management features will improve developer experience and reduce confusion during API integration. --- robosystems_client/api/graphs/create_graph.py | 36 +- .../api/graphs/get_available_extensions.py | 146 +++- robosystems_client/api/graphs/get_graphs.py | 174 ++++- robosystems_client/api/graphs/select_graph.py | 120 ++- robosystems_client/api/mcp/call_mcp_tool.py | 28 +- robosystems_client/api/mcp/list_mcp_tools.py | 20 +- .../api/query/execute_cypher_query.py | 25 +- robosystems_client/api/tables/delete_file.py | 436 +++++------ .../api/tables/get_file_info.py | 336 +++------ .../api/tables/get_upload_url.py | 540 +++++--------- .../api/tables/ingest_tables.py | 700 +++++++----------- .../api/tables/list_table_files.py | 520 +++++-------- robosystems_client/api/tables/list_tables.py | 504 +++++-------- robosystems_client/api/tables/query_tables.py | 124 +--- .../api/tables/update_file_status.py | 512 +++++-------- .../models/create_graph_request.py | 7 +- .../models/cypher_query_request.py | 21 +- 17 files changed, 1850 insertions(+), 2399 deletions(-) diff --git a/robosystems_client/api/graphs/create_graph.py b/robosystems_client/api/graphs/create_graph.py index ffda325..1bac42e 100644 --- a/robosystems_client/api/graphs/create_graph.py +++ b/robosystems_client/api/graphs/create_graph.py @@ -108,10 +108,11 @@ def sync_detailed( Args: body (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']}. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -182,10 +183,11 @@ def sync( Args: body (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']}. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -251,10 +253,11 @@ async def asyncio_detailed( Args: body (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']}. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -323,10 +326,11 @@ async def asyncio( Args: body (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']}. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. diff --git a/robosystems_client/api/graphs/get_available_extensions.py b/robosystems_client/api/graphs/get_available_extensions.py index 22323de..44e2797 100644 --- a/robosystems_client/api/graphs/get_available_extensions.py +++ b/robosystems_client/api/graphs/get_available_extensions.py @@ -1,5 +1,5 @@ from http import HTTPStatus -from typing import Any, Optional, Union +from typing import Any, Optional, Union, cast import httpx @@ -20,12 +20,16 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[AvailableExtensionsResponse]: +) -> Optional[Union[Any, AvailableExtensionsResponse]]: if response.status_code == 200: response_200 = AvailableExtensionsResponse.from_dict(response.json()) return response_200 + 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: @@ -34,7 +38,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[AvailableExtensionsResponse]: +) -> Response[Union[Any, AvailableExtensionsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -46,17 +50,45 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, -) -> Response[AvailableExtensionsResponse]: +) -> Response[Union[Any, AvailableExtensionsResponse]]: """Get Available Schema Extensions - List all available schema extensions for graph creation + List all available schema extensions for graph creation. + + Schema extensions provide pre-built industry-specific data models that extend + the base graph schema with specialized nodes, relationships, and properties. + + **Available Extensions:** + - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial + statements + - **RoboInvestor**: Investment portfolio management and tracking + - **RoboSCM**: Supply chain management and logistics + - **RoboFO**: Front office operations and CRM + - **RoboHRM**: Human resources management + - **RoboEPM**: Enterprise performance management + - **RoboReport**: Business intelligence and reporting + + **Extension Information:** + Each extension includes: + - Display name and description + - Node and relationship counts + - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs) + + **Use Cases:** + - Browse available extensions before creating a graph + - Understand extension capabilities and data models + - Plan graph schema based on business requirements + - Combine multiple extensions for comprehensive data modeling + + **Note:** + Extension listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[AvailableExtensionsResponse] + Response[Union[Any, AvailableExtensionsResponse]] """ kwargs = _get_kwargs() @@ -71,17 +103,45 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, -) -> Optional[AvailableExtensionsResponse]: +) -> Optional[Union[Any, AvailableExtensionsResponse]]: """Get Available Schema Extensions - List all available schema extensions for graph creation + List all available schema extensions for graph creation. + + Schema extensions provide pre-built industry-specific data models that extend + the base graph schema with specialized nodes, relationships, and properties. + + **Available Extensions:** + - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial + statements + - **RoboInvestor**: Investment portfolio management and tracking + - **RoboSCM**: Supply chain management and logistics + - **RoboFO**: Front office operations and CRM + - **RoboHRM**: Human resources management + - **RoboEPM**: Enterprise performance management + - **RoboReport**: Business intelligence and reporting + + **Extension Information:** + Each extension includes: + - Display name and description + - Node and relationship counts + - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs) + + **Use Cases:** + - Browse available extensions before creating a graph + - Understand extension capabilities and data models + - Plan graph schema based on business requirements + - Combine multiple extensions for comprehensive data modeling + + **Note:** + Extension listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - AvailableExtensionsResponse + Union[Any, AvailableExtensionsResponse] """ return sync_detailed( @@ -92,17 +152,45 @@ def sync( async def asyncio_detailed( *, client: AuthenticatedClient, -) -> Response[AvailableExtensionsResponse]: +) -> Response[Union[Any, AvailableExtensionsResponse]]: """Get Available Schema Extensions - List all available schema extensions for graph creation + List all available schema extensions for graph creation. + + Schema extensions provide pre-built industry-specific data models that extend + the base graph schema with specialized nodes, relationships, and properties. + + **Available Extensions:** + - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial + statements + - **RoboInvestor**: Investment portfolio management and tracking + - **RoboSCM**: Supply chain management and logistics + - **RoboFO**: Front office operations and CRM + - **RoboHRM**: Human resources management + - **RoboEPM**: Enterprise performance management + - **RoboReport**: Business intelligence and reporting + + **Extension Information:** + Each extension includes: + - Display name and description + - Node and relationship counts + - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs) + + **Use Cases:** + - Browse available extensions before creating a graph + - Understand extension capabilities and data models + - Plan graph schema based on business requirements + - Combine multiple extensions for comprehensive data modeling + + **Note:** + Extension listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[AvailableExtensionsResponse] + Response[Union[Any, AvailableExtensionsResponse]] """ kwargs = _get_kwargs() @@ -115,17 +203,45 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, -) -> Optional[AvailableExtensionsResponse]: +) -> Optional[Union[Any, AvailableExtensionsResponse]]: """Get Available Schema Extensions - List all available schema extensions for graph creation + List all available schema extensions for graph creation. + + Schema extensions provide pre-built industry-specific data models that extend + the base graph schema with specialized nodes, relationships, and properties. + + **Available Extensions:** + - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial + statements + - **RoboInvestor**: Investment portfolio management and tracking + - **RoboSCM**: Supply chain management and logistics + - **RoboFO**: Front office operations and CRM + - **RoboHRM**: Human resources management + - **RoboEPM**: Enterprise performance management + - **RoboReport**: Business intelligence and reporting + + **Extension Information:** + Each extension includes: + - Display name and description + - Node and relationship counts + - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs) + + **Use Cases:** + - Browse available extensions before creating a graph + - Understand extension capabilities and data models + - Plan graph schema based on business requirements + - Combine multiple extensions for comprehensive data modeling + + **Note:** + Extension listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - AvailableExtensionsResponse + Union[Any, AvailableExtensionsResponse] """ return ( diff --git a/robosystems_client/api/graphs/get_graphs.py b/robosystems_client/api/graphs/get_graphs.py index 2bc5acd..7ca98a7 100644 --- a/robosystems_client/api/graphs/get_graphs.py +++ b/robosystems_client/api/graphs/get_graphs.py @@ -1,5 +1,5 @@ from http import HTTPStatus -from typing import Any, Optional, Union +from typing import Any, Optional, Union, cast import httpx @@ -20,12 +20,16 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[UserGraphsResponse]: +) -> Optional[Union[Any, UserGraphsResponse]]: if response.status_code == 200: response_200 = UserGraphsResponse.from_dict(response.json()) return response_200 + 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: @@ -34,7 +38,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[UserGraphsResponse]: +) -> Response[Union[Any, UserGraphsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -46,17 +50,50 @@ def _build_response( def sync_detailed( *, client: AuthenticatedClient, -) -> Response[UserGraphsResponse]: - """Get User Graphs +) -> Response[Union[Any, UserGraphsResponse]]: + r"""Get User Graphs + + List all graph databases accessible to the current user with roles and selection status. + + Returns a comprehensive list of all graphs the user can access, including their + role in each graph (admin or member) and which graph is currently selected as + the active workspace. + + **Returned Information:** + - Graph ID and display name for each accessible graph + - User's role (admin/member) indicating permission level + - Selection status (one graph can be marked as \"selected\") + - Creation timestamp for each graph + + **Graph Roles:** + - `admin`: Full access - can manage graph settings, invite users, delete graph + - `member`: Read/write access - can query and modify data, cannot manage settings + + **Selected Graph Concept:** + The \"selected\" graph is the user's currently active workspace. Many API operations + default to the selected graph if no graph_id is provided. Users can change their + selected graph via the `POST /v1/graphs/{graph_id}/select` endpoint. + + **Use Cases:** + - Display graph selector in UI + - Show user's accessible workspaces + - Identify which graph is currently active + - Filter graphs by role for permission-based features + + **Empty Response:** + New users or users without graph access will receive an empty list with + `selectedGraphId: null`. Users should create a new graph or request access + to an existing graph. - Get all graph databases accessible to the current user. + **Note:** + Graph listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[UserGraphsResponse] + Response[Union[Any, UserGraphsResponse]] """ kwargs = _get_kwargs() @@ -71,17 +108,50 @@ def sync_detailed( def sync( *, client: AuthenticatedClient, -) -> Optional[UserGraphsResponse]: - """Get User Graphs +) -> Optional[Union[Any, UserGraphsResponse]]: + r"""Get User Graphs + + List all graph databases accessible to the current user with roles and selection status. + + Returns a comprehensive list of all graphs the user can access, including their + role in each graph (admin or member) and which graph is currently selected as + the active workspace. + + **Returned Information:** + - Graph ID and display name for each accessible graph + - User's role (admin/member) indicating permission level + - Selection status (one graph can be marked as \"selected\") + - Creation timestamp for each graph + + **Graph Roles:** + - `admin`: Full access - can manage graph settings, invite users, delete graph + - `member`: Read/write access - can query and modify data, cannot manage settings + + **Selected Graph Concept:** + The \"selected\" graph is the user's currently active workspace. Many API operations + default to the selected graph if no graph_id is provided. Users can change their + selected graph via the `POST /v1/graphs/{graph_id}/select` endpoint. + + **Use Cases:** + - Display graph selector in UI + - Show user's accessible workspaces + - Identify which graph is currently active + - Filter graphs by role for permission-based features + + **Empty Response:** + New users or users without graph access will receive an empty list with + `selectedGraphId: null`. Users should create a new graph or request access + to an existing graph. - Get all graph databases accessible to the current user. + **Note:** + Graph listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - UserGraphsResponse + Union[Any, UserGraphsResponse] """ return sync_detailed( @@ -92,17 +162,50 @@ def sync( async def asyncio_detailed( *, client: AuthenticatedClient, -) -> Response[UserGraphsResponse]: - """Get User Graphs +) -> Response[Union[Any, UserGraphsResponse]]: + r"""Get User Graphs - Get all graph databases accessible to the current user. + List all graph databases accessible to the current user with roles and selection status. + + Returns a comprehensive list of all graphs the user can access, including their + role in each graph (admin or member) and which graph is currently selected as + the active workspace. + + **Returned Information:** + - Graph ID and display name for each accessible graph + - User's role (admin/member) indicating permission level + - Selection status (one graph can be marked as \"selected\") + - Creation timestamp for each graph + + **Graph Roles:** + - `admin`: Full access - can manage graph settings, invite users, delete graph + - `member`: Read/write access - can query and modify data, cannot manage settings + + **Selected Graph Concept:** + The \"selected\" graph is the user's currently active workspace. Many API operations + default to the selected graph if no graph_id is provided. Users can change their + selected graph via the `POST /v1/graphs/{graph_id}/select` endpoint. + + **Use Cases:** + - Display graph selector in UI + - Show user's accessible workspaces + - Identify which graph is currently active + - Filter graphs by role for permission-based features + + **Empty Response:** + New users or users without graph access will receive an empty list with + `selectedGraphId: null`. Users should create a new graph or request access + to an existing graph. + + **Note:** + Graph listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[UserGraphsResponse] + Response[Union[Any, UserGraphsResponse]] """ kwargs = _get_kwargs() @@ -115,17 +218,50 @@ async def asyncio_detailed( async def asyncio( *, client: AuthenticatedClient, -) -> Optional[UserGraphsResponse]: - """Get User Graphs +) -> Optional[Union[Any, UserGraphsResponse]]: + r"""Get User Graphs + + List all graph databases accessible to the current user with roles and selection status. + + Returns a comprehensive list of all graphs the user can access, including their + role in each graph (admin or member) and which graph is currently selected as + the active workspace. + + **Returned Information:** + - Graph ID and display name for each accessible graph + - User's role (admin/member) indicating permission level + - Selection status (one graph can be marked as \"selected\") + - Creation timestamp for each graph + + **Graph Roles:** + - `admin`: Full access - can manage graph settings, invite users, delete graph + - `member`: Read/write access - can query and modify data, cannot manage settings + + **Selected Graph Concept:** + The \"selected\" graph is the user's currently active workspace. Many API operations + default to the selected graph if no graph_id is provided. Users can change their + selected graph via the `POST /v1/graphs/{graph_id}/select` endpoint. + + **Use Cases:** + - Display graph selector in UI + - Show user's accessible workspaces + - Identify which graph is currently active + - Filter graphs by role for permission-based features + + **Empty Response:** + New users or users without graph access will receive an empty list with + `selectedGraphId: null`. Users should create a new graph or request access + to an existing graph. - Get all graph databases accessible to the current user. + **Note:** + Graph listing is included - no credit consumption required. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - UserGraphsResponse + Union[Any, UserGraphsResponse] """ return ( diff --git a/robosystems_client/api/graphs/select_graph.py b/robosystems_client/api/graphs/select_graph.py index 41aff1a..2b35685 100644 --- a/robosystems_client/api/graphs/select_graph.py +++ b/robosystems_client/api/graphs/select_graph.py @@ -74,7 +74,35 @@ def sync_detailed( ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph - Select a specific graph as the active graph for the user. + Select a specific graph as the active workspace for the user. + + The selected graph becomes the default context for operations in client applications + and can be used to maintain user workspace preferences across sessions. + + **Functionality:** + - Sets the specified graph as the user's currently selected graph + - Deselects any previously selected graph (only one can be selected at a time) + - Persists selection across sessions until changed + - Returns confirmation with the selected graph ID + + **Requirements:** + - User must have access to the graph (as admin or member) + - Graph must exist and not be deleted + - User can only select graphs they have permission to access + + **Use Cases:** + - Switch between multiple graphs in a multi-graph environment + - Set default workspace after creating a new graph + - Restore user's preferred workspace on login + - Support graph context switching in client applications + + **Client Integration:** + Many client operations can default to the selected graph, simplifying API calls + by eliminating the need to specify graph_id repeatedly. Check the selected + graph with `GET /v1/graphs` which returns `selectedGraphId`. + + **Note:** + Graph selection is included - no credit consumption required. Args: graph_id (str): @@ -105,7 +133,35 @@ def sync( ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph - Select a specific graph as the active graph for the user. + Select a specific graph as the active workspace for the user. + + The selected graph becomes the default context for operations in client applications + and can be used to maintain user workspace preferences across sessions. + + **Functionality:** + - Sets the specified graph as the user's currently selected graph + - Deselects any previously selected graph (only one can be selected at a time) + - Persists selection across sessions until changed + - Returns confirmation with the selected graph ID + + **Requirements:** + - User must have access to the graph (as admin or member) + - Graph must exist and not be deleted + - User can only select graphs they have permission to access + + **Use Cases:** + - Switch between multiple graphs in a multi-graph environment + - Set default workspace after creating a new graph + - Restore user's preferred workspace on login + - Support graph context switching in client applications + + **Client Integration:** + Many client operations can default to the selected graph, simplifying API calls + by eliminating the need to specify graph_id repeatedly. Check the selected + graph with `GET /v1/graphs` which returns `selectedGraphId`. + + **Note:** + Graph selection is included - no credit consumption required. Args: graph_id (str): @@ -131,7 +187,35 @@ async def asyncio_detailed( ) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph - Select a specific graph as the active graph for the user. + Select a specific graph as the active workspace for the user. + + The selected graph becomes the default context for operations in client applications + and can be used to maintain user workspace preferences across sessions. + + **Functionality:** + - Sets the specified graph as the user's currently selected graph + - Deselects any previously selected graph (only one can be selected at a time) + - Persists selection across sessions until changed + - Returns confirmation with the selected graph ID + + **Requirements:** + - User must have access to the graph (as admin or member) + - Graph must exist and not be deleted + - User can only select graphs they have permission to access + + **Use Cases:** + - Switch between multiple graphs in a multi-graph environment + - Set default workspace after creating a new graph + - Restore user's preferred workspace on login + - Support graph context switching in client applications + + **Client Integration:** + Many client operations can default to the selected graph, simplifying API calls + by eliminating the need to specify graph_id repeatedly. Check the selected + graph with `GET /v1/graphs` which returns `selectedGraphId`. + + **Note:** + Graph selection is included - no credit consumption required. Args: graph_id (str): @@ -160,7 +244,35 @@ async def asyncio( ) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]: """Select Graph - Select a specific graph as the active graph for the user. + Select a specific graph as the active workspace for the user. + + The selected graph becomes the default context for operations in client applications + and can be used to maintain user workspace preferences across sessions. + + **Functionality:** + - Sets the specified graph as the user's currently selected graph + - Deselects any previously selected graph (only one can be selected at a time) + - Persists selection across sessions until changed + - Returns confirmation with the selected graph ID + + **Requirements:** + - User must have access to the graph (as admin or member) + - Graph must exist and not be deleted + - User can only select graphs they have permission to access + + **Use Cases:** + - Switch between multiple graphs in a multi-graph environment + - Set default workspace after creating a new graph + - Restore user's preferred workspace on login + - Support graph context switching in client applications + + **Client Integration:** + Many client operations can default to the selected graph, simplifying API calls + by eliminating the need to specify graph_id repeatedly. Check the selected + graph with `GET /v1/graphs` which returns `selectedGraphId`. + + **Note:** + Graph selection is included - no credit consumption required. Args: graph_id (str): diff --git a/robosystems_client/api/mcp/call_mcp_tool.py b/robosystems_client/api/mcp/call_mcp_tool.py index 1aab4fa..ee20506 100644 --- a/robosystems_client/api/mcp/call_mcp_tool.py +++ b/robosystems_client/api/mcp/call_mcp_tool.py @@ -157,8 +157,11 @@ def sync_detailed( - `408 Request Timeout`: Tool execution exceeded timeout - Clients should implement exponential backoff on errors - **Note:** - MCP tool calls are included and do not consume credits. + **Credit Model:** + MCP tool execution is included - no credit consumption required. Database + operations (queries, schema inspection, analytics) are completely free. + Only AI operations that invoke Claude or other LLM APIs consume credits, + which happens at the AI agent layer, not the MCP tool layer. Args: graph_id (str): @@ -230,8 +233,11 @@ def sync( - `408 Request Timeout`: Tool execution exceeded timeout - Clients should implement exponential backoff on errors - **Note:** - MCP tool calls are included and do not consume credits. + **Credit Model:** + MCP tool execution is included - no credit consumption required. Database + operations (queries, schema inspection, analytics) are completely free. + Only AI operations that invoke Claude or other LLM APIs consume credits, + which happens at the AI agent layer, not the MCP tool layer. Args: graph_id (str): @@ -298,8 +304,11 @@ async def asyncio_detailed( - `408 Request Timeout`: Tool execution exceeded timeout - Clients should implement exponential backoff on errors - **Note:** - MCP tool calls are included and do not consume credits. + **Credit Model:** + MCP tool execution is included - no credit consumption required. Database + operations (queries, schema inspection, analytics) are completely free. + Only AI operations that invoke Claude or other LLM APIs consume credits, + which happens at the AI agent layer, not the MCP tool layer. Args: graph_id (str): @@ -369,8 +378,11 @@ async def asyncio( - `408 Request Timeout`: Tool execution exceeded timeout - Clients should implement exponential backoff on errors - **Note:** - MCP tool calls are included and do not consume credits. + **Credit Model:** + MCP tool execution is included - no credit consumption required. Database + operations (queries, schema inspection, analytics) are completely free. + Only AI operations that invoke Claude or other LLM APIs consume credits, + which happens at the AI agent layer, not the MCP tool layer. Args: graph_id (str): diff --git a/robosystems_client/api/mcp/list_mcp_tools.py b/robosystems_client/api/mcp/list_mcp_tools.py index b13bbea..c642ebb 100644 --- a/robosystems_client/api/mcp/list_mcp_tools.py +++ b/robosystems_client/api/mcp/list_mcp_tools.py @@ -81,9 +81,8 @@ def sync_detailed( - User permissions and subscription tier - Backend capabilities (Kuzu, Neo4j, etc.) - Credit consumption: - - Listing tools is included to encourage exploration - - Tool execution costs vary by operation complexity + **Note:** + MCP tool listing is included - no credit consumption required. Args: graph_id (str): @@ -126,9 +125,8 @@ def sync( - User permissions and subscription tier - Backend capabilities (Kuzu, Neo4j, etc.) - Credit consumption: - - Listing tools is included to encourage exploration - - Tool execution costs vary by operation complexity + **Note:** + MCP tool listing is included - no credit consumption required. Args: graph_id (str): @@ -166,9 +164,8 @@ async def asyncio_detailed( - User permissions and subscription tier - Backend capabilities (Kuzu, Neo4j, etc.) - Credit consumption: - - Listing tools is included to encourage exploration - - Tool execution costs vary by operation complexity + **Note:** + MCP tool listing is included - no credit consumption required. Args: graph_id (str): @@ -209,9 +206,8 @@ async def asyncio( - User permissions and subscription tier - Backend capabilities (Kuzu, Neo4j, etc.) - Credit consumption: - - Listing tools is included to encourage exploration - - Tool execution costs vary by operation complexity + **Note:** + MCP tool listing is included - no credit consumption required. Args: graph_id (str): diff --git a/robosystems_client/api/query/execute_cypher_query.py b/robosystems_client/api/query/execute_cypher_query.py index 07c6568..d9b0661 100644 --- a/robosystems_client/api/query/execute_cypher_query.py +++ b/robosystems_client/api/query/execute_cypher_query.py @@ -16,7 +16,7 @@ def _get_kwargs( *, body: CypherQueryRequest, mode: Union[None, ResponseMode, Unset] = UNSET, - chunk_size: Union[Unset, int] = 1000, + chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> dict[str, Any]: headers: dict[str, Any] = {} @@ -32,7 +32,12 @@ def _get_kwargs( json_mode = mode params["mode"] = json_mode - params["chunk_size"] = chunk_size + json_chunk_size: Union[None, Unset, int] + if isinstance(chunk_size, Unset): + json_chunk_size = UNSET + else: + json_chunk_size = chunk_size + params["chunk_size"] = json_chunk_size params["test_mode"] = test_mode @@ -115,7 +120,7 @@ def sync_detailed( client: AuthenticatedClient, body: CypherQueryRequest, mode: Union[None, ResponseMode, Unset] = UNSET, - chunk_size: Union[Unset, int] = 1000, + chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Response[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -177,7 +182,7 @@ def sync_detailed( Args: graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override - chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. + chunk_size (Union[None, Unset, int]): Rows per chunk for streaming test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. body (CypherQueryRequest): Request model for Cypher query execution. @@ -210,7 +215,7 @@ def sync( client: AuthenticatedClient, body: CypherQueryRequest, mode: Union[None, ResponseMode, Unset] = UNSET, - chunk_size: Union[Unset, int] = 1000, + chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Optional[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -272,7 +277,7 @@ def sync( Args: graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override - chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. + chunk_size (Union[None, Unset, int]): Rows per chunk for streaming test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. body (CypherQueryRequest): Request model for Cypher query execution. @@ -300,7 +305,7 @@ async def asyncio_detailed( client: AuthenticatedClient, body: CypherQueryRequest, mode: Union[None, ResponseMode, Unset] = UNSET, - chunk_size: Union[Unset, int] = 1000, + chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Response[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -362,7 +367,7 @@ async def asyncio_detailed( Args: graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override - chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. + chunk_size (Union[None, Unset, int]): Rows per chunk for streaming test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. body (CypherQueryRequest): Request model for Cypher query execution. @@ -393,7 +398,7 @@ async def asyncio( client: AuthenticatedClient, body: CypherQueryRequest, mode: Union[None, ResponseMode, Unset] = UNSET, - chunk_size: Union[Unset, int] = 1000, + chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Optional[Union[Any, HTTPValidationError]]: """Execute Cypher Query (Read-Only) @@ -455,7 +460,7 @@ async def asyncio( Args: graph_id (str): mode (Union[None, ResponseMode, Unset]): Response mode override - chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000. + chunk_size (Union[None, Unset, int]): Rows per chunk for streaming test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False. body (CypherQueryRequest): Request model for Cypher query execution. diff --git a/robosystems_client/api/tables/delete_file.py b/robosystems_client/api/tables/delete_file.py index a9d6083..e50096f 100644 --- a/robosystems_client/api/tables/delete_file.py +++ b/robosystems_client/api/tables/delete_file.py @@ -77,70 +77,51 @@ def sync_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: - r""" Delete File from Staging - - Delete a file from S3 storage and database tracking. - - **Purpose:** - Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. - The file is deleted from both S3 and database tracking, and table statistics - are automatically recalculated. - - **Use Cases:** - - Remove duplicate uploads - - Delete files with incorrect data - - Clean up failed uploads - - Fix data quality issues before ingestion - - Manage storage usage - - **What Happens:** - 1. File deleted from S3 storage - 2. Database tracking record removed - 3. Table statistics recalculated (file count, size, row count) - 4. DuckDB automatically excludes file from future queries - - **Security:** - - Write access required (verified via auth) - - Shared repositories block file deletions - - Full audit trail of deletion operations - - Cannot delete after ingestion to graph - - **Example Response:** - ```json - { - \"status\": \"deleted\", - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"message\": \"File deleted successfully. DuckDB will automatically exclude it from queries.\" - } - ``` - - **Example Usage:** - ```bash - curl -X DELETE -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Tips:** - - Delete files before ingestion for best results - - Table statistics update automatically - - No need to refresh DuckDB - exclusion is automatic - - Consider re-uploading corrected version after deletion - - **Note:** - File deletion is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]] - """ + """Delete File from Staging + + Delete a file from S3 storage and database tracking. + + Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. + The file is deleted from both S3 and database tracking, and table statistics + are automatically recalculated. + + **Use Cases:** + - Remove duplicate uploads + - Delete files with incorrect data + - Clean up failed uploads + - Fix data quality issues before ingestion + - Manage storage usage + + **What Happens:** + 1. File deleted from S3 storage + 2. Database tracking record removed + 3. Table statistics recalculated (file count, size, row count) + 4. DuckDB automatically excludes file from future queries + + **Security:** + - Write access required (verified via auth) + - Shared repositories block file deletions + - Full audit trail of deletion operations + - Cannot delete after ingestion to graph + + **Important Notes:** + - Delete files before ingestion for best results + - Table statistics update automatically + - No need to refresh DuckDB - exclusion is automatic + - Consider re-uploading corrected version after deletion + - File deletion is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -160,70 +141,51 @@ def sync( *, client: AuthenticatedClient, ) -> Optional[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: - r""" Delete File from Staging - - Delete a file from S3 storage and database tracking. - - **Purpose:** - Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. - The file is deleted from both S3 and database tracking, and table statistics - are automatically recalculated. - - **Use Cases:** - - Remove duplicate uploads - - Delete files with incorrect data - - Clean up failed uploads - - Fix data quality issues before ingestion - - Manage storage usage - - **What Happens:** - 1. File deleted from S3 storage - 2. Database tracking record removed - 3. Table statistics recalculated (file count, size, row count) - 4. DuckDB automatically excludes file from future queries - - **Security:** - - Write access required (verified via auth) - - Shared repositories block file deletions - - Full audit trail of deletion operations - - Cannot delete after ingestion to graph - - **Example Response:** - ```json - { - \"status\": \"deleted\", - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"message\": \"File deleted successfully. DuckDB will automatically exclude it from queries.\" - } - ``` - - **Example Usage:** - ```bash - curl -X DELETE -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Tips:** - - Delete files before ingestion for best results - - Table statistics update automatically - - No need to refresh DuckDB - exclusion is automatic - - Consider re-uploading corrected version after deletion - - **Note:** - File deletion is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError] - """ + """Delete File from Staging + + Delete a file from S3 storage and database tracking. + + Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. + The file is deleted from both S3 and database tracking, and table statistics + are automatically recalculated. + + **Use Cases:** + - Remove duplicate uploads + - Delete files with incorrect data + - Clean up failed uploads + - Fix data quality issues before ingestion + - Manage storage usage + + **What Happens:** + 1. File deleted from S3 storage + 2. Database tracking record removed + 3. Table statistics recalculated (file count, size, row count) + 4. DuckDB automatically excludes file from future queries + + **Security:** + - Write access required (verified via auth) + - Shared repositories block file deletions + - Full audit trail of deletion operations + - Cannot delete after ingestion to graph + + **Important Notes:** + - Delete files before ingestion for best results + - Table statistics update automatically + - No need to refresh DuckDB - exclusion is automatic + - Consider re-uploading corrected version after deletion + - File deletion is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError] + """ return sync_detailed( graph_id=graph_id, @@ -238,70 +200,51 @@ async def asyncio_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: - r""" Delete File from Staging - - Delete a file from S3 storage and database tracking. - - **Purpose:** - Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. - The file is deleted from both S3 and database tracking, and table statistics - are automatically recalculated. - - **Use Cases:** - - Remove duplicate uploads - - Delete files with incorrect data - - Clean up failed uploads - - Fix data quality issues before ingestion - - Manage storage usage - - **What Happens:** - 1. File deleted from S3 storage - 2. Database tracking record removed - 3. Table statistics recalculated (file count, size, row count) - 4. DuckDB automatically excludes file from future queries - - **Security:** - - Write access required (verified via auth) - - Shared repositories block file deletions - - Full audit trail of deletion operations - - Cannot delete after ingestion to graph - - **Example Response:** - ```json - { - \"status\": \"deleted\", - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"message\": \"File deleted successfully. DuckDB will automatically exclude it from queries.\" - } - ``` - - **Example Usage:** - ```bash - curl -X DELETE -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Tips:** - - Delete files before ingestion for best results - - Table statistics update automatically - - No need to refresh DuckDB - exclusion is automatic - - Consider re-uploading corrected version after deletion - - **Note:** - File deletion is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]] - """ + """Delete File from Staging + + Delete a file from S3 storage and database tracking. + + Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. + The file is deleted from both S3 and database tracking, and table statistics + are automatically recalculated. + + **Use Cases:** + - Remove duplicate uploads + - Delete files with incorrect data + - Clean up failed uploads + - Fix data quality issues before ingestion + - Manage storage usage + + **What Happens:** + 1. File deleted from S3 storage + 2. Database tracking record removed + 3. Table statistics recalculated (file count, size, row count) + 4. DuckDB automatically excludes file from future queries + + **Security:** + - Write access required (verified via auth) + - Shared repositories block file deletions + - Full audit trail of deletion operations + - Cannot delete after ingestion to graph + + **Important Notes:** + - Delete files before ingestion for best results + - Table statistics update automatically + - No need to refresh DuckDB - exclusion is automatic + - Consider re-uploading corrected version after deletion + - File deletion is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -319,70 +262,51 @@ async def asyncio( *, client: AuthenticatedClient, ) -> Optional[Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError]]: - r""" Delete File from Staging - - Delete a file from S3 storage and database tracking. - - **Purpose:** - Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. - The file is deleted from both S3 and database tracking, and table statistics - are automatically recalculated. - - **Use Cases:** - - Remove duplicate uploads - - Delete files with incorrect data - - Clean up failed uploads - - Fix data quality issues before ingestion - - Manage storage usage - - **What Happens:** - 1. File deleted from S3 storage - 2. Database tracking record removed - 3. Table statistics recalculated (file count, size, row count) - 4. DuckDB automatically excludes file from future queries - - **Security:** - - Write access required (verified via auth) - - Shared repositories block file deletions - - Full audit trail of deletion operations - - Cannot delete after ingestion to graph - - **Example Response:** - ```json - { - \"status\": \"deleted\", - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"message\": \"File deleted successfully. DuckDB will automatically exclude it from queries.\" - } - ``` - - **Example Usage:** - ```bash - curl -X DELETE -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Tips:** - - Delete files before ingestion for best results - - Table statistics update automatically - - No need to refresh DuckDB - exclusion is automatic - - Consider re-uploading corrected version after deletion - - **Note:** - File deletion is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError] - """ + """Delete File from Staging + + Delete a file from S3 storage and database tracking. + + Remove unwanted, duplicate, or incorrect files from staging tables before ingestion. + The file is deleted from both S3 and database tracking, and table statistics + are automatically recalculated. + + **Use Cases:** + - Remove duplicate uploads + - Delete files with incorrect data + - Clean up failed uploads + - Fix data quality issues before ingestion + - Manage storage usage + + **What Happens:** + 1. File deleted from S3 storage + 2. Database tracking record removed + 3. Table statistics recalculated (file count, size, row count) + 4. DuckDB automatically excludes file from future queries + + **Security:** + - Write access required (verified via auth) + - Shared repositories block file deletions + - Full audit trail of deletion operations + - Cannot delete after ingestion to graph + + **Important Notes:** + - Delete files before ingestion for best results + - Table statistics update automatically + - No need to refresh DuckDB - exclusion is automatic + - Consider re-uploading corrected version after deletion + - File deletion is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, DeleteFileResponse, ErrorResponse, HTTPValidationError] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/get_file_info.py b/robosystems_client/api/tables/get_file_info.py index c148c52..1601b64 100644 --- a/robosystems_client/api/tables/get_file_info.py +++ b/robosystems_client/api/tables/get_file_info.py @@ -73,61 +73,35 @@ def sync_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: - r""" Get File Information - - Get detailed information about a specific file. - - **Purpose:** - Retrieve comprehensive metadata for a single file, including upload status, - size, row count, and timestamps. Useful for validating individual files - before ingestion. - - **Use Cases:** - - Validate file upload completion - - Check file metadata before ingestion - - Debug upload issues - - Verify file format and size - - Track file lifecycle - - **Example Response:** - ```json - { - \"file_id\": \"f123\", - \"graph_id\": \"kg123\", - \"table_id\": \"t456\", - \"table_name\": \"Entity\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Note:** - File info retrieval is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]] - """ + """Get File Information + + Get detailed information about a specific file. + + Retrieve comprehensive metadata for a single file, including upload status, + size, row count, and timestamps. Useful for validating individual files + before ingestion. + + **Use Cases:** + - Validate file upload completion + - Check file metadata before ingestion + - Debug upload issues + - Verify file format and size + - Track file lifecycle + + **Note:** + File info retrieval is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -147,61 +121,35 @@ def sync( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: - r""" Get File Information - - Get detailed information about a specific file. - - **Purpose:** - Retrieve comprehensive metadata for a single file, including upload status, - size, row count, and timestamps. Useful for validating individual files - before ingestion. - - **Use Cases:** - - Validate file upload completion - - Check file metadata before ingestion - - Debug upload issues - - Verify file format and size - - Track file lifecycle - - **Example Response:** - ```json - { - \"file_id\": \"f123\", - \"graph_id\": \"kg123\", - \"table_id\": \"t456\", - \"table_name\": \"Entity\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Note:** - File info retrieval is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError] - """ + """Get File Information + + Get detailed information about a specific file. + + Retrieve comprehensive metadata for a single file, including upload status, + size, row count, and timestamps. Useful for validating individual files + before ingestion. + + **Use Cases:** + - Validate file upload completion + - Check file metadata before ingestion + - Debug upload issues + - Verify file format and size + - Track file lifecycle + + **Note:** + File info retrieval is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError] + """ return sync_detailed( graph_id=graph_id, @@ -216,61 +164,35 @@ async def asyncio_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: - r""" Get File Information - - Get detailed information about a specific file. - - **Purpose:** - Retrieve comprehensive metadata for a single file, including upload status, - size, row count, and timestamps. Useful for validating individual files - before ingestion. - - **Use Cases:** - - Validate file upload completion - - Check file metadata before ingestion - - Debug upload issues - - Verify file format and size - - Track file lifecycle - - **Example Response:** - ```json - { - \"file_id\": \"f123\", - \"graph_id\": \"kg123\", - \"table_id\": \"t456\", - \"table_name\": \"Entity\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Note:** - File info retrieval is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]] - """ + """Get File Information + + Get detailed information about a specific file. + + Retrieve comprehensive metadata for a single file, including upload status, + size, row count, and timestamps. Useful for validating individual files + before ingestion. + + **Use Cases:** + - Validate file upload completion + - Check file metadata before ingestion + - Debug upload issues + - Verify file format and size + - Track file lifecycle + + **Note:** + File info retrieval is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -288,61 +210,35 @@ async def asyncio( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError]]: - r""" Get File Information - - Get detailed information about a specific file. - - **Purpose:** - Retrieve comprehensive metadata for a single file, including upload status, - size, row count, and timestamps. Useful for validating individual files - before ingestion. - - **Use Cases:** - - Validate file upload completion - - Check file metadata before ingestion - - Debug upload issues - - Verify file format and size - - Track file lifecycle - - **Example Response:** - ```json - { - \"file_id\": \"f123\", - \"graph_id\": \"kg123\", - \"table_id\": \"t456\", - \"table_name\": \"Entity\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123 - ``` - - **Note:** - File info retrieval is included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File ID - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError] - """ + """Get File Information + + Get detailed information about a specific file. + + Retrieve comprehensive metadata for a single file, including upload status, + size, row count, and timestamps. Useful for validating individual files + before ingestion. + + **Use Cases:** + - Validate file upload completion + - Check file metadata before ingestion + - Debug upload issues + - Verify file format and size + - Track file lifecycle + + **Note:** + File info retrieval is included - no credit consumption + + Args: + graph_id (str): + file_id (str): File ID + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, GetFileInfoResponse, HTTPValidationError] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/get_upload_url.py b/robosystems_client/api/tables/get_upload_url.py index 065db74..b48ad79 100644 --- a/robosystems_client/api/tables/get_upload_url.py +++ b/robosystems_client/api/tables/get_upload_url.py @@ -93,93 +93,54 @@ def sync_detailed( client: AuthenticatedClient, body: FileUploadRequest, ) -> Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: - r""" Get File Upload URL - - Generate a presigned S3 URL for secure file upload. - - **Purpose:** - Initiate file upload to a staging table by generating a secure, time-limited - presigned S3 URL. Files are uploaded directly to S3, bypassing the API for - optimal performance. - - **Upload Workflow:** - 1. Call this endpoint to get presigned URL - 2. PUT file directly to S3 URL (using curl, axios, etc.) - 3. Call PATCH /tables/files/{file_id} with status='uploaded' - 4. Backend validates file and calculates metrics - 5. File ready for ingestion - - **Supported Formats:** - - Parquet (`application/x-parquet` with `.parquet` extension) - - CSV (`text/csv` with `.csv` extension) - - JSON (`application/json` with `.json` extension) - - **Validation:** - - File extension must match content type - - File name 1-255 characters - - No path traversal characters (.. / \) - - Auto-creates table if it doesn't exist - - **Auto-Table Creation:** - If the table doesn't exist, it's automatically created with: - - Type inferred from name (e.g., \"Transaction\" ā†’ relationship) - - Empty schema (populated on ingestion) - - Ready for file uploads - - **Example Response:** - ```json - { - \"upload_url\": \"https://bucket.s3.amazonaws.com/path?X-Amz-Algorithm=...\", - \"expires_in\": 3600, - \"file_id\": \"f123-456-789\", - \"s3_key\": \"user-staging/user123/kg456/Entity/f123.../data.parquet\" - } - ``` - - **Example Usage:** - ```bash - # Step 1: Get upload URL - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"file_name\": \"entities.parquet\", - \"content_type\": \"application/x-parquet\" - }' - - # Step 2: Upload file directly to S3 - curl -X PUT \"$UPLOAD_URL\" \ - -H \"Content-Type: application/x-parquet\" \ - --data-binary \"@entities.parquet\" - - # Step 3: Mark as uploaded - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/$FILE_ID\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Presigned URLs expire (default: 1 hour) - - Use appropriate Content-Type header when uploading to S3 - - File extension must match content type - - Large files benefit from direct S3 upload - - **Note:** - Upload URL generation is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - body (FileUploadRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]] - """ + r"""Get File Upload URL + + Generate a presigned S3 URL for secure file upload. + + Initiates file upload to a staging table by generating a secure, time-limited + presigned S3 URL. Files are uploaded directly to S3, bypassing the API for + optimal performance. + + **Upload Workflow:** + 1. Call this endpoint to get presigned URL + 2. PUT file directly to S3 URL + 3. Call PATCH /tables/files/{file_id} with status='uploaded' + 4. Backend validates file and calculates metrics + 5. File ready for ingestion + + **Supported Formats:** + - Parquet (`application/x-parquet` with `.parquet` extension) + - CSV (`text/csv` with `.csv` extension) + - JSON (`application/json` with `.json` extension) + + **Validation:** + - File extension must match content type + - File name 1-255 characters + - No path traversal characters (.. / \) + - Auto-creates table if it doesn't exist + + **Auto-Table Creation:** + Tables are automatically created on first file upload with type inferred from name + (e.g., \"Transaction\" ā†’ relationship) and empty schema populated during ingestion. + + **Important Notes:** + - Presigned URLs expire (default: 1 hour) + - Use appropriate Content-Type header when uploading to S3 + - File extension must match content type + - Upload URL generation is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + body (FileUploadRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -201,93 +162,54 @@ def sync( client: AuthenticatedClient, body: FileUploadRequest, ) -> Optional[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: - r""" Get File Upload URL - - Generate a presigned S3 URL for secure file upload. - - **Purpose:** - Initiate file upload to a staging table by generating a secure, time-limited - presigned S3 URL. Files are uploaded directly to S3, bypassing the API for - optimal performance. - - **Upload Workflow:** - 1. Call this endpoint to get presigned URL - 2. PUT file directly to S3 URL (using curl, axios, etc.) - 3. Call PATCH /tables/files/{file_id} with status='uploaded' - 4. Backend validates file and calculates metrics - 5. File ready for ingestion - - **Supported Formats:** - - Parquet (`application/x-parquet` with `.parquet` extension) - - CSV (`text/csv` with `.csv` extension) - - JSON (`application/json` with `.json` extension) - - **Validation:** - - File extension must match content type - - File name 1-255 characters - - No path traversal characters (.. / \) - - Auto-creates table if it doesn't exist - - **Auto-Table Creation:** - If the table doesn't exist, it's automatically created with: - - Type inferred from name (e.g., \"Transaction\" ā†’ relationship) - - Empty schema (populated on ingestion) - - Ready for file uploads - - **Example Response:** - ```json - { - \"upload_url\": \"https://bucket.s3.amazonaws.com/path?X-Amz-Algorithm=...\", - \"expires_in\": 3600, - \"file_id\": \"f123-456-789\", - \"s3_key\": \"user-staging/user123/kg456/Entity/f123.../data.parquet\" - } - ``` - - **Example Usage:** - ```bash - # Step 1: Get upload URL - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"file_name\": \"entities.parquet\", - \"content_type\": \"application/x-parquet\" - }' - - # Step 2: Upload file directly to S3 - curl -X PUT \"$UPLOAD_URL\" \ - -H \"Content-Type: application/x-parquet\" \ - --data-binary \"@entities.parquet\" - - # Step 3: Mark as uploaded - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/$FILE_ID\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Presigned URLs expire (default: 1 hour) - - Use appropriate Content-Type header when uploading to S3 - - File extension must match content type - - Large files benefit from direct S3 upload - - **Note:** - Upload URL generation is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - body (FileUploadRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError] - """ + r"""Get File Upload URL + + Generate a presigned S3 URL for secure file upload. + + Initiates file upload to a staging table by generating a secure, time-limited + presigned S3 URL. Files are uploaded directly to S3, bypassing the API for + optimal performance. + + **Upload Workflow:** + 1. Call this endpoint to get presigned URL + 2. PUT file directly to S3 URL + 3. Call PATCH /tables/files/{file_id} with status='uploaded' + 4. Backend validates file and calculates metrics + 5. File ready for ingestion + + **Supported Formats:** + - Parquet (`application/x-parquet` with `.parquet` extension) + - CSV (`text/csv` with `.csv` extension) + - JSON (`application/json` with `.json` extension) + + **Validation:** + - File extension must match content type + - File name 1-255 characters + - No path traversal characters (.. / \) + - Auto-creates table if it doesn't exist + + **Auto-Table Creation:** + Tables are automatically created on first file upload with type inferred from name + (e.g., \"Transaction\" ā†’ relationship) and empty schema populated during ingestion. + + **Important Notes:** + - Presigned URLs expire (default: 1 hour) + - Use appropriate Content-Type header when uploading to S3 + - File extension must match content type + - Upload URL generation is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + body (FileUploadRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError] + """ return sync_detailed( graph_id=graph_id, @@ -304,93 +226,54 @@ async def asyncio_detailed( client: AuthenticatedClient, body: FileUploadRequest, ) -> Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: - r""" Get File Upload URL - - Generate a presigned S3 URL for secure file upload. - - **Purpose:** - Initiate file upload to a staging table by generating a secure, time-limited - presigned S3 URL. Files are uploaded directly to S3, bypassing the API for - optimal performance. - - **Upload Workflow:** - 1. Call this endpoint to get presigned URL - 2. PUT file directly to S3 URL (using curl, axios, etc.) - 3. Call PATCH /tables/files/{file_id} with status='uploaded' - 4. Backend validates file and calculates metrics - 5. File ready for ingestion - - **Supported Formats:** - - Parquet (`application/x-parquet` with `.parquet` extension) - - CSV (`text/csv` with `.csv` extension) - - JSON (`application/json` with `.json` extension) - - **Validation:** - - File extension must match content type - - File name 1-255 characters - - No path traversal characters (.. / \) - - Auto-creates table if it doesn't exist - - **Auto-Table Creation:** - If the table doesn't exist, it's automatically created with: - - Type inferred from name (e.g., \"Transaction\" ā†’ relationship) - - Empty schema (populated on ingestion) - - Ready for file uploads - - **Example Response:** - ```json - { - \"upload_url\": \"https://bucket.s3.amazonaws.com/path?X-Amz-Algorithm=...\", - \"expires_in\": 3600, - \"file_id\": \"f123-456-789\", - \"s3_key\": \"user-staging/user123/kg456/Entity/f123.../data.parquet\" - } - ``` - - **Example Usage:** - ```bash - # Step 1: Get upload URL - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"file_name\": \"entities.parquet\", - \"content_type\": \"application/x-parquet\" - }' - - # Step 2: Upload file directly to S3 - curl -X PUT \"$UPLOAD_URL\" \ - -H \"Content-Type: application/x-parquet\" \ - --data-binary \"@entities.parquet\" - - # Step 3: Mark as uploaded - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/$FILE_ID\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Presigned URLs expire (default: 1 hour) - - Use appropriate Content-Type header when uploading to S3 - - File extension must match content type - - Large files benefit from direct S3 upload - - **Note:** - Upload URL generation is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - body (FileUploadRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]] - """ + r"""Get File Upload URL + + Generate a presigned S3 URL for secure file upload. + + Initiates file upload to a staging table by generating a secure, time-limited + presigned S3 URL. Files are uploaded directly to S3, bypassing the API for + optimal performance. + + **Upload Workflow:** + 1. Call this endpoint to get presigned URL + 2. PUT file directly to S3 URL + 3. Call PATCH /tables/files/{file_id} with status='uploaded' + 4. Backend validates file and calculates metrics + 5. File ready for ingestion + + **Supported Formats:** + - Parquet (`application/x-parquet` with `.parquet` extension) + - CSV (`text/csv` with `.csv` extension) + - JSON (`application/json` with `.json` extension) + + **Validation:** + - File extension must match content type + - File name 1-255 characters + - No path traversal characters (.. / \) + - Auto-creates table if it doesn't exist + + **Auto-Table Creation:** + Tables are automatically created on first file upload with type inferred from name + (e.g., \"Transaction\" ā†’ relationship) and empty schema populated during ingestion. + + **Important Notes:** + - Presigned URLs expire (default: 1 hour) + - Use appropriate Content-Type header when uploading to S3 + - File extension must match content type + - Upload URL generation is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + body (FileUploadRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -410,93 +293,54 @@ async def asyncio( client: AuthenticatedClient, body: FileUploadRequest, ) -> Optional[Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError]]: - r""" Get File Upload URL - - Generate a presigned S3 URL for secure file upload. - - **Purpose:** - Initiate file upload to a staging table by generating a secure, time-limited - presigned S3 URL. Files are uploaded directly to S3, bypassing the API for - optimal performance. - - **Upload Workflow:** - 1. Call this endpoint to get presigned URL - 2. PUT file directly to S3 URL (using curl, axios, etc.) - 3. Call PATCH /tables/files/{file_id} with status='uploaded' - 4. Backend validates file and calculates metrics - 5. File ready for ingestion - - **Supported Formats:** - - Parquet (`application/x-parquet` with `.parquet` extension) - - CSV (`text/csv` with `.csv` extension) - - JSON (`application/json` with `.json` extension) - - **Validation:** - - File extension must match content type - - File name 1-255 characters - - No path traversal characters (.. / \) - - Auto-creates table if it doesn't exist - - **Auto-Table Creation:** - If the table doesn't exist, it's automatically created with: - - Type inferred from name (e.g., \"Transaction\" ā†’ relationship) - - Empty schema (populated on ingestion) - - Ready for file uploads - - **Example Response:** - ```json - { - \"upload_url\": \"https://bucket.s3.amazonaws.com/path?X-Amz-Algorithm=...\", - \"expires_in\": 3600, - \"file_id\": \"f123-456-789\", - \"s3_key\": \"user-staging/user123/kg456/Entity/f123.../data.parquet\" - } - ``` - - **Example Usage:** - ```bash - # Step 1: Get upload URL - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"file_name\": \"entities.parquet\", - \"content_type\": \"application/x-parquet\" - }' - - # Step 2: Upload file directly to S3 - curl -X PUT \"$UPLOAD_URL\" \ - -H \"Content-Type: application/x-parquet\" \ - --data-binary \"@entities.parquet\" - - # Step 3: Mark as uploaded - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/$FILE_ID\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Presigned URLs expire (default: 1 hour) - - Use appropriate Content-Type header when uploading to S3 - - File extension must match content type - - Large files benefit from direct S3 upload - - **Note:** - Upload URL generation is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - body (FileUploadRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError] - """ + r"""Get File Upload URL + + Generate a presigned S3 URL for secure file upload. + + Initiates file upload to a staging table by generating a secure, time-limited + presigned S3 URL. Files are uploaded directly to S3, bypassing the API for + optimal performance. + + **Upload Workflow:** + 1. Call this endpoint to get presigned URL + 2. PUT file directly to S3 URL + 3. Call PATCH /tables/files/{file_id} with status='uploaded' + 4. Backend validates file and calculates metrics + 5. File ready for ingestion + + **Supported Formats:** + - Parquet (`application/x-parquet` with `.parquet` extension) + - CSV (`text/csv` with `.csv` extension) + - JSON (`application/json` with `.json` extension) + + **Validation:** + - File extension must match content type + - File name 1-255 characters + - No path traversal characters (.. / \) + - Auto-creates table if it doesn't exist + + **Auto-Table Creation:** + Tables are automatically created on first file upload with type inferred from name + (e.g., \"Transaction\" ā†’ relationship) and empty schema populated during ingestion. + + **Important Notes:** + - Presigned URLs expire (default: 1 hour) + - Use appropriate Content-Type header when uploading to S3 + - File extension must match content type + - Upload URL generation is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + body (FileUploadRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, FileUploadResponse, HTTPValidationError] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/ingest_tables.py b/robosystems_client/api/tables/ingest_tables.py index fe9d890..7191471 100644 --- a/robosystems_client/api/tables/ingest_tables.py +++ b/robosystems_client/api/tables/ingest_tables.py @@ -92,112 +92,75 @@ def sync_detailed( client: AuthenticatedClient, body: BulkIngestRequest, ) -> Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: - r""" Ingest Tables to Graph - - Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. - - **Purpose:** - Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. - Processes all tables in a single bulk operation with comprehensive error handling and metrics. - - **Use Cases:** - - Initial graph population from uploaded data - - Incremental data updates with new files - - Complete database rebuild from source files - - Recovery from failed ingestion attempts - - **Workflow:** - 1. Upload data files via `POST /tables/{table_name}/files` - 2. Files are validated and marked as 'uploaded' - 3. Trigger ingestion: `POST /tables/ingest` - 4. DuckDB staging tables created from S3 patterns - 5. Data copied row-by-row from DuckDB to Kuzu - 6. Per-table results and metrics returned - - **Rebuild Feature:** - Setting `rebuild=true` regenerates the entire graph database from scratch: - - Deletes existing Kuzu database - - Recreates with fresh schema from active GraphSchema - - Ingests all data files - - Safe operation - S3 is source of truth - - Useful for schema changes or data corrections - - Graph marked as 'rebuilding' during process - - **Error Handling:** - - Per-table error isolation with `ignore_errors` flag - - Partial success support (some tables succeed, some fail) - - Detailed error reporting per table - - Graph status tracking throughout process - - Automatic failure recovery and cleanup - - **Performance:** - - Processes all tables in sequence - - Each table timed independently - - Total execution metrics provided - - Scales to thousands of files - - Optimized for large datasets - - **Example Request:** - ```bash - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/ingest\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"ignore_errors\": true, - \"rebuild\": false - }' - ``` - - **Example Response:** - ```json - { - \"status\": \"success\", - \"graph_id\": \"kg123\", - \"total_tables\": 5, - \"successful_tables\": 5, - \"failed_tables\": 0, - \"skipped_tables\": 0, - \"total_rows_ingested\": 25000, - \"total_execution_time_ms\": 15420.5, - \"results\": [ - { - \"table_name\": \"Entity\", - \"status\": \"success\", - \"rows_ingested\": 5000, - \"execution_time_ms\": 3200.1, - \"error\": null - } - ] - } - ``` - - **Concurrency Control:** - Only one ingestion can run per graph at a time. If another ingestion is in progress, - you'll receive a 409 Conflict error. The distributed lock automatically expires after - the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. - - **Tips:** - - Only files with 'uploaded' status are processed - - Tables with no uploaded files are skipped - - Use `ignore_errors=false` for strict validation - - Monitor progress via per-table results - - Check graph metadata for rebuild status - - Wait for current ingestion to complete before starting another - - **Note:** - Table ingestion is included - no credit consumption. - - Args: - graph_id (str): - body (BulkIngestRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]] - """ + """Ingest Tables to Graph + + Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. + + Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. + Processes all tables in a single bulk operation with comprehensive error handling and metrics. + + **Use Cases:** + - Initial graph population from uploaded data + - Incremental data updates with new files + - Complete database rebuild from source files + - Recovery from failed ingestion attempts + + **Workflow:** + 1. Upload data files via `POST /tables/{table_name}/files` + 2. Files are validated and marked as 'uploaded' + 3. Trigger ingestion: `POST /tables/ingest` + 4. DuckDB staging tables created from S3 patterns + 5. Data copied row-by-row from DuckDB to Kuzu + 6. Per-table results and metrics returned + + **Rebuild Feature:** + Setting `rebuild=true` regenerates the entire graph database from scratch: + - Deletes existing Kuzu database + - Recreates with fresh schema from active GraphSchema + - Ingests all data files + - Safe operation - S3 is source of truth + - Useful for schema changes or data corrections + - Graph marked as 'rebuilding' during process + + **Error Handling:** + - Per-table error isolation with `ignore_errors` flag + - Partial success support (some tables succeed, some fail) + - Detailed error reporting per table + - Graph status tracking throughout process + - Automatic failure recovery and cleanup + + **Performance:** + - Processes all tables in sequence + - Each table timed independently + - Total execution metrics provided + - Scales to thousands of files + - Optimized for large datasets + + **Concurrency Control:** + Only one ingestion can run per graph at a time. If another ingestion is in progress, + you'll receive a 409 Conflict error. The distributed lock automatically expires after + the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. + + **Important Notes:** + - Only files with 'uploaded' status are processed + - Tables with no uploaded files are skipped + - Use `ignore_errors=false` for strict validation + - Monitor progress via per-table results + - Check graph metadata for rebuild status + - Wait for current ingestion to complete before starting another + - Table ingestion is included - no credit consumption + + Args: + graph_id (str): + body (BulkIngestRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -217,112 +180,75 @@ def sync( client: AuthenticatedClient, body: BulkIngestRequest, ) -> Optional[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: - r""" Ingest Tables to Graph - - Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. - - **Purpose:** - Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. - Processes all tables in a single bulk operation with comprehensive error handling and metrics. - - **Use Cases:** - - Initial graph population from uploaded data - - Incremental data updates with new files - - Complete database rebuild from source files - - Recovery from failed ingestion attempts - - **Workflow:** - 1. Upload data files via `POST /tables/{table_name}/files` - 2. Files are validated and marked as 'uploaded' - 3. Trigger ingestion: `POST /tables/ingest` - 4. DuckDB staging tables created from S3 patterns - 5. Data copied row-by-row from DuckDB to Kuzu - 6. Per-table results and metrics returned - - **Rebuild Feature:** - Setting `rebuild=true` regenerates the entire graph database from scratch: - - Deletes existing Kuzu database - - Recreates with fresh schema from active GraphSchema - - Ingests all data files - - Safe operation - S3 is source of truth - - Useful for schema changes or data corrections - - Graph marked as 'rebuilding' during process - - **Error Handling:** - - Per-table error isolation with `ignore_errors` flag - - Partial success support (some tables succeed, some fail) - - Detailed error reporting per table - - Graph status tracking throughout process - - Automatic failure recovery and cleanup - - **Performance:** - - Processes all tables in sequence - - Each table timed independently - - Total execution metrics provided - - Scales to thousands of files - - Optimized for large datasets - - **Example Request:** - ```bash - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/ingest\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"ignore_errors\": true, - \"rebuild\": false - }' - ``` - - **Example Response:** - ```json - { - \"status\": \"success\", - \"graph_id\": \"kg123\", - \"total_tables\": 5, - \"successful_tables\": 5, - \"failed_tables\": 0, - \"skipped_tables\": 0, - \"total_rows_ingested\": 25000, - \"total_execution_time_ms\": 15420.5, - \"results\": [ - { - \"table_name\": \"Entity\", - \"status\": \"success\", - \"rows_ingested\": 5000, - \"execution_time_ms\": 3200.1, - \"error\": null - } - ] - } - ``` - - **Concurrency Control:** - Only one ingestion can run per graph at a time. If another ingestion is in progress, - you'll receive a 409 Conflict error. The distributed lock automatically expires after - the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. - - **Tips:** - - Only files with 'uploaded' status are processed - - Tables with no uploaded files are skipped - - Use `ignore_errors=false` for strict validation - - Monitor progress via per-table results - - Check graph metadata for rebuild status - - Wait for current ingestion to complete before starting another - - **Note:** - Table ingestion is included - no credit consumption. - - Args: - graph_id (str): - body (BulkIngestRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError] - """ + """Ingest Tables to Graph + + Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. + + Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. + Processes all tables in a single bulk operation with comprehensive error handling and metrics. + + **Use Cases:** + - Initial graph population from uploaded data + - Incremental data updates with new files + - Complete database rebuild from source files + - Recovery from failed ingestion attempts + + **Workflow:** + 1. Upload data files via `POST /tables/{table_name}/files` + 2. Files are validated and marked as 'uploaded' + 3. Trigger ingestion: `POST /tables/ingest` + 4. DuckDB staging tables created from S3 patterns + 5. Data copied row-by-row from DuckDB to Kuzu + 6. Per-table results and metrics returned + + **Rebuild Feature:** + Setting `rebuild=true` regenerates the entire graph database from scratch: + - Deletes existing Kuzu database + - Recreates with fresh schema from active GraphSchema + - Ingests all data files + - Safe operation - S3 is source of truth + - Useful for schema changes or data corrections + - Graph marked as 'rebuilding' during process + + **Error Handling:** + - Per-table error isolation with `ignore_errors` flag + - Partial success support (some tables succeed, some fail) + - Detailed error reporting per table + - Graph status tracking throughout process + - Automatic failure recovery and cleanup + + **Performance:** + - Processes all tables in sequence + - Each table timed independently + - Total execution metrics provided + - Scales to thousands of files + - Optimized for large datasets + + **Concurrency Control:** + Only one ingestion can run per graph at a time. If another ingestion is in progress, + you'll receive a 409 Conflict error. The distributed lock automatically expires after + the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. + + **Important Notes:** + - Only files with 'uploaded' status are processed + - Tables with no uploaded files are skipped + - Use `ignore_errors=false` for strict validation + - Monitor progress via per-table results + - Check graph metadata for rebuild status + - Wait for current ingestion to complete before starting another + - Table ingestion is included - no credit consumption + + Args: + graph_id (str): + body (BulkIngestRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError] + """ return sync_detailed( graph_id=graph_id, @@ -337,112 +263,75 @@ async def asyncio_detailed( client: AuthenticatedClient, body: BulkIngestRequest, ) -> Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: - r""" Ingest Tables to Graph - - Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. - - **Purpose:** - Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. - Processes all tables in a single bulk operation with comprehensive error handling and metrics. - - **Use Cases:** - - Initial graph population from uploaded data - - Incremental data updates with new files - - Complete database rebuild from source files - - Recovery from failed ingestion attempts - - **Workflow:** - 1. Upload data files via `POST /tables/{table_name}/files` - 2. Files are validated and marked as 'uploaded' - 3. Trigger ingestion: `POST /tables/ingest` - 4. DuckDB staging tables created from S3 patterns - 5. Data copied row-by-row from DuckDB to Kuzu - 6. Per-table results and metrics returned - - **Rebuild Feature:** - Setting `rebuild=true` regenerates the entire graph database from scratch: - - Deletes existing Kuzu database - - Recreates with fresh schema from active GraphSchema - - Ingests all data files - - Safe operation - S3 is source of truth - - Useful for schema changes or data corrections - - Graph marked as 'rebuilding' during process - - **Error Handling:** - - Per-table error isolation with `ignore_errors` flag - - Partial success support (some tables succeed, some fail) - - Detailed error reporting per table - - Graph status tracking throughout process - - Automatic failure recovery and cleanup - - **Performance:** - - Processes all tables in sequence - - Each table timed independently - - Total execution metrics provided - - Scales to thousands of files - - Optimized for large datasets - - **Example Request:** - ```bash - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/ingest\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"ignore_errors\": true, - \"rebuild\": false - }' - ``` - - **Example Response:** - ```json - { - \"status\": \"success\", - \"graph_id\": \"kg123\", - \"total_tables\": 5, - \"successful_tables\": 5, - \"failed_tables\": 0, - \"skipped_tables\": 0, - \"total_rows_ingested\": 25000, - \"total_execution_time_ms\": 15420.5, - \"results\": [ - { - \"table_name\": \"Entity\", - \"status\": \"success\", - \"rows_ingested\": 5000, - \"execution_time_ms\": 3200.1, - \"error\": null - } - ] - } - ``` - - **Concurrency Control:** - Only one ingestion can run per graph at a time. If another ingestion is in progress, - you'll receive a 409 Conflict error. The distributed lock automatically expires after - the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. - - **Tips:** - - Only files with 'uploaded' status are processed - - Tables with no uploaded files are skipped - - Use `ignore_errors=false` for strict validation - - Monitor progress via per-table results - - Check graph metadata for rebuild status - - Wait for current ingestion to complete before starting another - - **Note:** - Table ingestion is included - no credit consumption. - - Args: - graph_id (str): - body (BulkIngestRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]] - """ + """Ingest Tables to Graph + + Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. + + Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. + Processes all tables in a single bulk operation with comprehensive error handling and metrics. + + **Use Cases:** + - Initial graph population from uploaded data + - Incremental data updates with new files + - Complete database rebuild from source files + - Recovery from failed ingestion attempts + + **Workflow:** + 1. Upload data files via `POST /tables/{table_name}/files` + 2. Files are validated and marked as 'uploaded' + 3. Trigger ingestion: `POST /tables/ingest` + 4. DuckDB staging tables created from S3 patterns + 5. Data copied row-by-row from DuckDB to Kuzu + 6. Per-table results and metrics returned + + **Rebuild Feature:** + Setting `rebuild=true` regenerates the entire graph database from scratch: + - Deletes existing Kuzu database + - Recreates with fresh schema from active GraphSchema + - Ingests all data files + - Safe operation - S3 is source of truth + - Useful for schema changes or data corrections + - Graph marked as 'rebuilding' during process + + **Error Handling:** + - Per-table error isolation with `ignore_errors` flag + - Partial success support (some tables succeed, some fail) + - Detailed error reporting per table + - Graph status tracking throughout process + - Automatic failure recovery and cleanup + + **Performance:** + - Processes all tables in sequence + - Each table timed independently + - Total execution metrics provided + - Scales to thousands of files + - Optimized for large datasets + + **Concurrency Control:** + Only one ingestion can run per graph at a time. If another ingestion is in progress, + you'll receive a 409 Conflict error. The distributed lock automatically expires after + the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. + + **Important Notes:** + - Only files with 'uploaded' status are processed + - Tables with no uploaded files are skipped + - Use `ignore_errors=false` for strict validation + - Monitor progress via per-table results + - Check graph metadata for rebuild status + - Wait for current ingestion to complete before starting another + - Table ingestion is included - no credit consumption + + Args: + graph_id (str): + body (BulkIngestRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -460,112 +349,75 @@ async def asyncio( client: AuthenticatedClient, body: BulkIngestRequest, ) -> Optional[Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError]]: - r""" Ingest Tables to Graph - - Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. - - **Purpose:** - Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. - Processes all tables in a single bulk operation with comprehensive error handling and metrics. - - **Use Cases:** - - Initial graph population from uploaded data - - Incremental data updates with new files - - Complete database rebuild from source files - - Recovery from failed ingestion attempts - - **Workflow:** - 1. Upload data files via `POST /tables/{table_name}/files` - 2. Files are validated and marked as 'uploaded' - 3. Trigger ingestion: `POST /tables/ingest` - 4. DuckDB staging tables created from S3 patterns - 5. Data copied row-by-row from DuckDB to Kuzu - 6. Per-table results and metrics returned - - **Rebuild Feature:** - Setting `rebuild=true` regenerates the entire graph database from scratch: - - Deletes existing Kuzu database - - Recreates with fresh schema from active GraphSchema - - Ingests all data files - - Safe operation - S3 is source of truth - - Useful for schema changes or data corrections - - Graph marked as 'rebuilding' during process - - **Error Handling:** - - Per-table error isolation with `ignore_errors` flag - - Partial success support (some tables succeed, some fail) - - Detailed error reporting per table - - Graph status tracking throughout process - - Automatic failure recovery and cleanup - - **Performance:** - - Processes all tables in sequence - - Each table timed independently - - Total execution metrics provided - - Scales to thousands of files - - Optimized for large datasets - - **Example Request:** - ```bash - curl -X POST \"https://api.robosystems.ai/v1/graphs/kg123/tables/ingest\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{ - \"ignore_errors\": true, - \"rebuild\": false - }' - ``` - - **Example Response:** - ```json - { - \"status\": \"success\", - \"graph_id\": \"kg123\", - \"total_tables\": 5, - \"successful_tables\": 5, - \"failed_tables\": 0, - \"skipped_tables\": 0, - \"total_rows_ingested\": 25000, - \"total_execution_time_ms\": 15420.5, - \"results\": [ - { - \"table_name\": \"Entity\", - \"status\": \"success\", - \"rows_ingested\": 5000, - \"execution_time_ms\": 3200.1, - \"error\": null - } - ] - } - ``` - - **Concurrency Control:** - Only one ingestion can run per graph at a time. If another ingestion is in progress, - you'll receive a 409 Conflict error. The distributed lock automatically expires after - the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. - - **Tips:** - - Only files with 'uploaded' status are processed - - Tables with no uploaded files are skipped - - Use `ignore_errors=false` for strict validation - - Monitor progress via per-table results - - Check graph metadata for rebuild status - - Wait for current ingestion to complete before starting another - - **Note:** - Table ingestion is included - no credit consumption. - - Args: - graph_id (str): - body (BulkIngestRequest): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError] - """ + """Ingest Tables to Graph + + Load all files from S3 into DuckDB staging tables and ingest into Kuzu graph database. + + Orchestrates the complete data pipeline from S3 staging files into the Kuzu graph database. + Processes all tables in a single bulk operation with comprehensive error handling and metrics. + + **Use Cases:** + - Initial graph population from uploaded data + - Incremental data updates with new files + - Complete database rebuild from source files + - Recovery from failed ingestion attempts + + **Workflow:** + 1. Upload data files via `POST /tables/{table_name}/files` + 2. Files are validated and marked as 'uploaded' + 3. Trigger ingestion: `POST /tables/ingest` + 4. DuckDB staging tables created from S3 patterns + 5. Data copied row-by-row from DuckDB to Kuzu + 6. Per-table results and metrics returned + + **Rebuild Feature:** + Setting `rebuild=true` regenerates the entire graph database from scratch: + - Deletes existing Kuzu database + - Recreates with fresh schema from active GraphSchema + - Ingests all data files + - Safe operation - S3 is source of truth + - Useful for schema changes or data corrections + - Graph marked as 'rebuilding' during process + + **Error Handling:** + - Per-table error isolation with `ignore_errors` flag + - Partial success support (some tables succeed, some fail) + - Detailed error reporting per table + - Graph status tracking throughout process + - Automatic failure recovery and cleanup + + **Performance:** + - Processes all tables in sequence + - Each table timed independently + - Total execution metrics provided + - Scales to thousands of files + - Optimized for large datasets + + **Concurrency Control:** + Only one ingestion can run per graph at a time. If another ingestion is in progress, + you'll receive a 409 Conflict error. The distributed lock automatically expires after + the configured TTL (default: 1 hour) to prevent deadlocks from failed ingestions. + + **Important Notes:** + - Only files with 'uploaded' status are processed + - Tables with no uploaded files are skipped + - Use `ignore_errors=false` for strict validation + - Monitor progress via per-table results + - Check graph metadata for rebuild status + - Wait for current ingestion to complete before starting another + - Table ingestion is included - no credit consumption + + Args: + graph_id (str): + body (BulkIngestRequest): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, BulkIngestResponse, ErrorResponse, HTTPValidationError] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/list_table_files.py b/robosystems_client/api/tables/list_table_files.py index 2a408f6..95a95cb 100644 --- a/robosystems_client/api/tables/list_table_files.py +++ b/robosystems_client/api/tables/list_table_files.py @@ -77,88 +77,54 @@ def sync_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: - r""" List Files in Staging Table - - List all files uploaded to a staging table with comprehensive metadata. - - **Purpose:** - Get a complete inventory of all files in a staging table, including upload status, - file sizes, row counts, and S3 locations. Essential for monitoring upload progress - and validating data before ingestion. - - **Use Cases:** - - Monitor file upload progress - - Verify files are ready for ingestion - - Check file formats and sizes - - Track storage usage per table - - Identify failed or incomplete uploads - - Pre-ingestion validation - - **What You Get:** - - File ID and name - - File format (parquet, csv, etc.) - - Size in bytes - - Row count (if available) - - Upload status and method - - Creation and upload timestamps - - S3 key for reference - - **Upload Status Values:** - - `created`: File record created, not yet uploaded - - `uploading`: Upload in progress - - `uploaded`: Successfully uploaded, ready for ingestion - - `failed`: Upload failed - - **Example Response:** - ```json - { - \"graph_id\": \"kg123\", - \"table_name\": \"Entity\", - \"files\": [ - { - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ], - \"total_files\": 1, - \"total_size_bytes\": 1048576 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files - ``` - - **Tips:** - - Only `uploaded` files are ingested - - Check `row_count` to estimate data volume - - Use `total_size_bytes` for storage monitoring - - Files with `failed` status should be deleted and re-uploaded - - **Note:** - File listing is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]] - """ + """List Files in Staging Table + + List all files uploaded to a staging table with comprehensive metadata. + + Get a complete inventory of all files in a staging table, including upload status, + file sizes, row counts, and S3 locations. Essential for monitoring upload progress + and validating data before ingestion. + + **Use Cases:** + - Monitor file upload progress + - Verify files are ready for ingestion + - Check file formats and sizes + - Track storage usage per table + - Identify failed or incomplete uploads + - Pre-ingestion validation + + **Returned Metadata:** + - File ID, name, and format (parquet, csv, json) + - Size in bytes and row count (if available) + - Upload status and method + - Creation and upload timestamps + - S3 key for reference + + **Upload Status Values:** + - `pending`: Upload URL generated, awaiting upload + - `uploaded`: Successfully uploaded, ready for ingestion + - `disabled`: Excluded from ingestion + - `archived`: Soft deleted + - `failed`: Upload failed + + **Important Notes:** + - Only `uploaded` files are ingested + - Check `row_count` to estimate data volume + - Use `total_size_bytes` for storage monitoring + - Files with `failed` status should be deleted and re-uploaded + - File listing is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -178,88 +144,54 @@ def sync( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: - r""" List Files in Staging Table - - List all files uploaded to a staging table with comprehensive metadata. - - **Purpose:** - Get a complete inventory of all files in a staging table, including upload status, - file sizes, row counts, and S3 locations. Essential for monitoring upload progress - and validating data before ingestion. - - **Use Cases:** - - Monitor file upload progress - - Verify files are ready for ingestion - - Check file formats and sizes - - Track storage usage per table - - Identify failed or incomplete uploads - - Pre-ingestion validation - - **What You Get:** - - File ID and name - - File format (parquet, csv, etc.) - - Size in bytes - - Row count (if available) - - Upload status and method - - Creation and upload timestamps - - S3 key for reference - - **Upload Status Values:** - - `created`: File record created, not yet uploaded - - `uploading`: Upload in progress - - `uploaded`: Successfully uploaded, ready for ingestion - - `failed`: Upload failed - - **Example Response:** - ```json - { - \"graph_id\": \"kg123\", - \"table_name\": \"Entity\", - \"files\": [ - { - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ], - \"total_files\": 1, - \"total_size_bytes\": 1048576 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files - ``` - - **Tips:** - - Only `uploaded` files are ingested - - Check `row_count` to estimate data volume - - Use `total_size_bytes` for storage monitoring - - Files with `failed` status should be deleted and re-uploaded - - **Note:** - File listing is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse] - """ + """List Files in Staging Table + + List all files uploaded to a staging table with comprehensive metadata. + + Get a complete inventory of all files in a staging table, including upload status, + file sizes, row counts, and S3 locations. Essential for monitoring upload progress + and validating data before ingestion. + + **Use Cases:** + - Monitor file upload progress + - Verify files are ready for ingestion + - Check file formats and sizes + - Track storage usage per table + - Identify failed or incomplete uploads + - Pre-ingestion validation + + **Returned Metadata:** + - File ID, name, and format (parquet, csv, json) + - Size in bytes and row count (if available) + - Upload status and method + - Creation and upload timestamps + - S3 key for reference + + **Upload Status Values:** + - `pending`: Upload URL generated, awaiting upload + - `uploaded`: Successfully uploaded, ready for ingestion + - `disabled`: Excluded from ingestion + - `archived`: Soft deleted + - `failed`: Upload failed + + **Important Notes:** + - Only `uploaded` files are ingested + - Check `row_count` to estimate data volume + - Use `total_size_bytes` for storage monitoring + - Files with `failed` status should be deleted and re-uploaded + - File listing is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse] + """ return sync_detailed( graph_id=graph_id, @@ -274,88 +206,54 @@ async def asyncio_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: - r""" List Files in Staging Table - - List all files uploaded to a staging table with comprehensive metadata. - - **Purpose:** - Get a complete inventory of all files in a staging table, including upload status, - file sizes, row counts, and S3 locations. Essential for monitoring upload progress - and validating data before ingestion. - - **Use Cases:** - - Monitor file upload progress - - Verify files are ready for ingestion - - Check file formats and sizes - - Track storage usage per table - - Identify failed or incomplete uploads - - Pre-ingestion validation - - **What You Get:** - - File ID and name - - File format (parquet, csv, etc.) - - Size in bytes - - Row count (if available) - - Upload status and method - - Creation and upload timestamps - - S3 key for reference - - **Upload Status Values:** - - `created`: File record created, not yet uploaded - - `uploading`: Upload in progress - - `uploaded`: Successfully uploaded, ready for ingestion - - `failed`: Upload failed - - **Example Response:** - ```json - { - \"graph_id\": \"kg123\", - \"table_name\": \"Entity\", - \"files\": [ - { - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ], - \"total_files\": 1, - \"total_size_bytes\": 1048576 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files - ``` - - **Tips:** - - Only `uploaded` files are ingested - - Check `row_count` to estimate data volume - - Use `total_size_bytes` for storage monitoring - - Files with `failed` status should be deleted and re-uploaded - - **Note:** - File listing is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]] - """ + """List Files in Staging Table + + List all files uploaded to a staging table with comprehensive metadata. + + Get a complete inventory of all files in a staging table, including upload status, + file sizes, row counts, and S3 locations. Essential for monitoring upload progress + and validating data before ingestion. + + **Use Cases:** + - Monitor file upload progress + - Verify files are ready for ingestion + - Check file formats and sizes + - Track storage usage per table + - Identify failed or incomplete uploads + - Pre-ingestion validation + + **Returned Metadata:** + - File ID, name, and format (parquet, csv, json) + - Size in bytes and row count (if available) + - Upload status and method + - Creation and upload timestamps + - S3 key for reference + + **Upload Status Values:** + - `pending`: Upload URL generated, awaiting upload + - `uploaded`: Successfully uploaded, ready for ingestion + - `disabled`: Excluded from ingestion + - `archived`: Soft deleted + - `failed`: Upload failed + + **Important Notes:** + - Only `uploaded` files are ingested + - Check `row_count` to estimate data volume + - Use `total_size_bytes` for storage monitoring + - Files with `failed` status should be deleted and re-uploaded + - File listing is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -373,88 +271,54 @@ async def asyncio( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]: - r""" List Files in Staging Table - - List all files uploaded to a staging table with comprehensive metadata. - - **Purpose:** - Get a complete inventory of all files in a staging table, including upload status, - file sizes, row counts, and S3 locations. Essential for monitoring upload progress - and validating data before ingestion. - - **Use Cases:** - - Monitor file upload progress - - Verify files are ready for ingestion - - Check file formats and sizes - - Track storage usage per table - - Identify failed or incomplete uploads - - Pre-ingestion validation - - **What You Get:** - - File ID and name - - File format (parquet, csv, etc.) - - Size in bytes - - Row count (if available) - - Upload status and method - - Creation and upload timestamps - - S3 key for reference - - **Upload Status Values:** - - `created`: File record created, not yet uploaded - - `uploading`: Upload in progress - - `uploaded`: Successfully uploaded, ready for ingestion - - `failed`: Upload failed - - **Example Response:** - ```json - { - \"graph_id\": \"kg123\", - \"table_name\": \"Entity\", - \"files\": [ - { - \"file_id\": \"f123\", - \"file_name\": \"entities_batch1.parquet\", - \"file_format\": \"parquet\", - \"size_bytes\": 1048576, - \"row_count\": 5000, - \"upload_status\": \"uploaded\", - \"upload_method\": \"presigned_url\", - \"created_at\": \"2025-10-28T10:00:00Z\", - \"uploaded_at\": \"2025-10-28T10:01:30Z\", - \"s3_key\": \"user-staging/user123/kg123/Entity/entities_batch1.parquet\" - } - ], - \"total_files\": 1, - \"total_size_bytes\": 1048576 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables/Entity/files - ``` - - **Tips:** - - Only `uploaded` files are ingested - - Check `row_count` to estimate data volume - - Use `total_size_bytes` for storage monitoring - - Files with `failed` status should be deleted and re-uploaded - - **Note:** - File listing is included - no credit consumption. - - Args: - graph_id (str): - table_name (str): Table name - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse] - """ + """List Files in Staging Table + + List all files uploaded to a staging table with comprehensive metadata. + + Get a complete inventory of all files in a staging table, including upload status, + file sizes, row counts, and S3 locations. Essential for monitoring upload progress + and validating data before ingestion. + + **Use Cases:** + - Monitor file upload progress + - Verify files are ready for ingestion + - Check file formats and sizes + - Track storage usage per table + - Identify failed or incomplete uploads + - Pre-ingestion validation + + **Returned Metadata:** + - File ID, name, and format (parquet, csv, json) + - Size in bytes and row count (if available) + - Upload status and method + - Creation and upload timestamps + - S3 key for reference + + **Upload Status Values:** + - `pending`: Upload URL generated, awaiting upload + - `uploaded`: Successfully uploaded, ready for ingestion + - `disabled`: Excluded from ingestion + - `archived`: Soft deleted + - `failed`: Upload failed + + **Important Notes:** + - Only `uploaded` files are ingested + - Check `row_count` to estimate data volume + - Use `total_size_bytes` for storage monitoring + - Files with `failed` status should be deleted and re-uploaded + - File listing is included - no credit consumption + + Args: + graph_id (str): + table_name (str): Table name + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/list_tables.py b/robosystems_client/api/tables/list_tables.py index 72e90c8..0aeb634 100644 --- a/robosystems_client/api/tables/list_tables.py +++ b/robosystems_client/api/tables/list_tables.py @@ -75,85 +75,53 @@ def sync_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: - r""" List Staging Tables - - List all DuckDB staging tables with comprehensive metrics and status. - - **Purpose:** - Get a complete inventory of all staging tables for a graph, including - file counts, storage sizes, and row estimates. Essential for monitoring - the data pipeline and determining which tables are ready for ingestion. - - **What You Get:** - - Table name and type (node/relationship) - - File count per table - - Total storage size in bytes - - Estimated row count - - S3 location pattern - - Ready-for-ingestion status - - **Use Cases:** - - Monitor data upload progress - - Check which tables have files ready - - Track storage consumption - - Validate pipeline before ingestion - - Capacity planning - - **Workflow:** - 1. List tables to see current state - 2. Upload files to empty tables - 3. Re-list to verify uploads - 4. Check file counts and sizes - 5. Ingest when ready - - **Example Response:** - ```json - { - \"tables\": [ - { - \"table_name\": \"Entity\", - \"row_count\": 5000, - \"file_count\": 3, - \"total_size_bytes\": 2457600, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\" - }, - { - \"table_name\": \"Transaction\", - \"row_count\": 15000, - \"file_count\": 5, - \"total_size_bytes\": 8192000, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\" - } - ], - \"total_count\": 2 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables - ``` - - **Tips:** - - Tables with `file_count > 0` have data ready - - Check `total_size_bytes` for storage monitoring - - Use `s3_location` to verify upload paths - - Empty tables (file_count=0) are skipped during ingestion - - **Note:** - Table queries are included - no credit consumption. - - Args: - graph_id (str): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]] - """ + """List Staging Tables + + List all DuckDB staging tables with comprehensive metrics and status. + + Get a complete inventory of all staging tables for a graph, including + file counts, storage sizes, and row estimates. Essential for monitoring + the data pipeline and determining which tables are ready for ingestion. + + **Returned Metrics:** + - Table name and type (node/relationship) + - File count per table + - Total storage size in bytes + - Estimated row count + - S3 location pattern + - Ready-for-ingestion status + + **Use Cases:** + - Monitor data upload progress + - Check which tables have files ready + - Track storage consumption + - Validate pipeline before ingestion + - Capacity planning + + **Workflow:** + 1. List tables to see current state + 2. Upload files to empty tables + 3. Re-list to verify uploads + 4. Check file counts and sizes + 5. Ingest when ready + + **Important Notes:** + - Tables with `file_count > 0` have data ready + - Check `total_size_bytes` for storage monitoring + - Use `s3_location` to verify upload paths + - Empty tables (file_count=0) are skipped during ingestion + - Table queries are included - no credit consumption + + Args: + graph_id (str): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -171,85 +139,53 @@ def sync( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: - r""" List Staging Tables - - List all DuckDB staging tables with comprehensive metrics and status. - - **Purpose:** - Get a complete inventory of all staging tables for a graph, including - file counts, storage sizes, and row estimates. Essential for monitoring - the data pipeline and determining which tables are ready for ingestion. - - **What You Get:** - - Table name and type (node/relationship) - - File count per table - - Total storage size in bytes - - Estimated row count - - S3 location pattern - - Ready-for-ingestion status - - **Use Cases:** - - Monitor data upload progress - - Check which tables have files ready - - Track storage consumption - - Validate pipeline before ingestion - - Capacity planning - - **Workflow:** - 1. List tables to see current state - 2. Upload files to empty tables - 3. Re-list to verify uploads - 4. Check file counts and sizes - 5. Ingest when ready - - **Example Response:** - ```json - { - \"tables\": [ - { - \"table_name\": \"Entity\", - \"row_count\": 5000, - \"file_count\": 3, - \"total_size_bytes\": 2457600, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\" - }, - { - \"table_name\": \"Transaction\", - \"row_count\": 15000, - \"file_count\": 5, - \"total_size_bytes\": 8192000, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\" - } - ], - \"total_count\": 2 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables - ``` - - **Tips:** - - Tables with `file_count > 0` have data ready - - Check `total_size_bytes` for storage monitoring - - Use `s3_location` to verify upload paths - - Empty tables (file_count=0) are skipped during ingestion - - **Note:** - Table queries are included - no credit consumption. - - Args: - graph_id (str): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, TableListResponse] - """ + """List Staging Tables + + List all DuckDB staging tables with comprehensive metrics and status. + + Get a complete inventory of all staging tables for a graph, including + file counts, storage sizes, and row estimates. Essential for monitoring + the data pipeline and determining which tables are ready for ingestion. + + **Returned Metrics:** + - Table name and type (node/relationship) + - File count per table + - Total storage size in bytes + - Estimated row count + - S3 location pattern + - Ready-for-ingestion status + + **Use Cases:** + - Monitor data upload progress + - Check which tables have files ready + - Track storage consumption + - Validate pipeline before ingestion + - Capacity planning + + **Workflow:** + 1. List tables to see current state + 2. Upload files to empty tables + 3. Re-list to verify uploads + 4. Check file counts and sizes + 5. Ingest when ready + + **Important Notes:** + - Tables with `file_count > 0` have data ready + - Check `total_size_bytes` for storage monitoring + - Use `s3_location` to verify upload paths + - Empty tables (file_count=0) are skipped during ingestion + - Table queries are included - no credit consumption + + Args: + graph_id (str): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, TableListResponse] + """ return sync_detailed( graph_id=graph_id, @@ -262,85 +198,53 @@ async def asyncio_detailed( *, client: AuthenticatedClient, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: - r""" List Staging Tables - - List all DuckDB staging tables with comprehensive metrics and status. - - **Purpose:** - Get a complete inventory of all staging tables for a graph, including - file counts, storage sizes, and row estimates. Essential for monitoring - the data pipeline and determining which tables are ready for ingestion. - - **What You Get:** - - Table name and type (node/relationship) - - File count per table - - Total storage size in bytes - - Estimated row count - - S3 location pattern - - Ready-for-ingestion status - - **Use Cases:** - - Monitor data upload progress - - Check which tables have files ready - - Track storage consumption - - Validate pipeline before ingestion - - Capacity planning - - **Workflow:** - 1. List tables to see current state - 2. Upload files to empty tables - 3. Re-list to verify uploads - 4. Check file counts and sizes - 5. Ingest when ready - - **Example Response:** - ```json - { - \"tables\": [ - { - \"table_name\": \"Entity\", - \"row_count\": 5000, - \"file_count\": 3, - \"total_size_bytes\": 2457600, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\" - }, - { - \"table_name\": \"Transaction\", - \"row_count\": 15000, - \"file_count\": 5, - \"total_size_bytes\": 8192000, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\" - } - ], - \"total_count\": 2 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables - ``` - - **Tips:** - - Tables with `file_count > 0` have data ready - - Check `total_size_bytes` for storage monitoring - - Use `s3_location` to verify upload paths - - Empty tables (file_count=0) are skipped during ingestion - - **Note:** - Table queries are included - no credit consumption. - - Args: - graph_id (str): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]] - """ + """List Staging Tables + + List all DuckDB staging tables with comprehensive metrics and status. + + Get a complete inventory of all staging tables for a graph, including + file counts, storage sizes, and row estimates. Essential for monitoring + the data pipeline and determining which tables are ready for ingestion. + + **Returned Metrics:** + - Table name and type (node/relationship) + - File count per table + - Total storage size in bytes + - Estimated row count + - S3 location pattern + - Ready-for-ingestion status + + **Use Cases:** + - Monitor data upload progress + - Check which tables have files ready + - Track storage consumption + - Validate pipeline before ingestion + - Capacity planning + + **Workflow:** + 1. List tables to see current state + 2. Upload files to empty tables + 3. Re-list to verify uploads + 4. Check file counts and sizes + 5. Ingest when ready + + **Important Notes:** + - Tables with `file_count > 0` have data ready + - Check `total_size_bytes` for storage monitoring + - Use `s3_location` to verify upload paths + - Empty tables (file_count=0) are skipped during ingestion + - Table queries are included - no credit consumption + + Args: + graph_id (str): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -356,85 +260,53 @@ async def asyncio( *, client: AuthenticatedClient, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]: - r""" List Staging Tables - - List all DuckDB staging tables with comprehensive metrics and status. - - **Purpose:** - Get a complete inventory of all staging tables for a graph, including - file counts, storage sizes, and row estimates. Essential for monitoring - the data pipeline and determining which tables are ready for ingestion. - - **What You Get:** - - Table name and type (node/relationship) - - File count per table - - Total storage size in bytes - - Estimated row count - - S3 location pattern - - Ready-for-ingestion status - - **Use Cases:** - - Monitor data upload progress - - Check which tables have files ready - - Track storage consumption - - Validate pipeline before ingestion - - Capacity planning - - **Workflow:** - 1. List tables to see current state - 2. Upload files to empty tables - 3. Re-list to verify uploads - 4. Check file counts and sizes - 5. Ingest when ready - - **Example Response:** - ```json - { - \"tables\": [ - { - \"table_name\": \"Entity\", - \"row_count\": 5000, - \"file_count\": 3, - \"total_size_bytes\": 2457600, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\" - }, - { - \"table_name\": \"Transaction\", - \"row_count\": 15000, - \"file_count\": 5, - \"total_size_bytes\": 8192000, - \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\" - } - ], - \"total_count\": 2 - } - ``` - - **Example Usage:** - ```bash - curl -H \"Authorization: Bearer YOUR_TOKEN\" \ - https://api.robosystems.ai/v1/graphs/kg123/tables - ``` - - **Tips:** - - Tables with `file_count > 0` have data ready - - Check `total_size_bytes` for storage monitoring - - Use `s3_location` to verify upload paths - - Empty tables (file_count=0) are skipped during ingestion - - **Note:** - Table queries are included - no credit consumption. - - Args: - graph_id (str): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, TableListResponse] - """ + """List Staging Tables + + List all DuckDB staging tables with comprehensive metrics and status. + + Get a complete inventory of all staging tables for a graph, including + file counts, storage sizes, and row estimates. Essential for monitoring + the data pipeline and determining which tables are ready for ingestion. + + **Returned Metrics:** + - Table name and type (node/relationship) + - File count per table + - Total storage size in bytes + - Estimated row count + - S3 location pattern + - Ready-for-ingestion status + + **Use Cases:** + - Monitor data upload progress + - Check which tables have files ready + - Track storage consumption + - Validate pipeline before ingestion + - Capacity planning + + **Workflow:** + 1. List tables to see current state + 2. Upload files to empty tables + 3. Re-list to verify uploads + 4. Check file counts and sizes + 5. Ingest when ready + + **Important Notes:** + - Tables with `file_count > 0` have data ready + - Check `total_size_bytes` for storage monitoring + - Use `s3_location` to verify upload paths + - Empty tables (file_count=0) are skipped during ingestion + - Table queries are included - no credit consumption + + Args: + graph_id (str): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, TableListResponse] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/api/tables/query_tables.py b/robosystems_client/api/tables/query_tables.py index 1a351ef..d68151a 100644 --- a/robosystems_client/api/tables/query_tables.py +++ b/robosystems_client/api/tables/query_tables.py @@ -99,7 +99,6 @@ def sync_detailed( Execute SQL queries on DuckDB staging tables for data inspection and validation. - **Purpose:** Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. @@ -122,27 +121,13 @@ def sync_detailed( - Aggregations, window functions, CTEs - Multiple table joins across staging area - **Example Queries:** - ```sql - -- Count rows in staging table - SELECT COUNT(*) FROM Entity; - - -- Check for nulls - SELECT * FROM Entity WHERE name IS NULL LIMIT 10; - - -- Find duplicates - SELECT identifier, COUNT(*) as cnt - FROM Entity - GROUP BY identifier - HAVING COUNT(*) > 1; - - -- Join across tables - SELECT e.name, COUNT(t.id) as transaction_count - FROM Entity e - LEFT JOIN Transaction t ON e.identifier = t.entity_id - GROUP BY e.name - ORDER BY transaction_count DESC; - ``` + **Common Operations:** + - Count rows: `SELECT COUNT(*) FROM Entity` + - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` + - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING + COUNT(*) > 1` + - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = + t.entity_id GROUP BY e.name` **Limits:** - Query timeout: 30 seconds @@ -155,7 +140,7 @@ def sync_detailed( Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query` **Note:** - Staging table queries are included - no credit consumption. + Staging table queries are included - no credit consumption Args: graph_id (str): @@ -191,7 +176,6 @@ def sync( Execute SQL queries on DuckDB staging tables for data inspection and validation. - **Purpose:** Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. @@ -214,27 +198,13 @@ def sync( - Aggregations, window functions, CTEs - Multiple table joins across staging area - **Example Queries:** - ```sql - -- Count rows in staging table - SELECT COUNT(*) FROM Entity; - - -- Check for nulls - SELECT * FROM Entity WHERE name IS NULL LIMIT 10; - - -- Find duplicates - SELECT identifier, COUNT(*) as cnt - FROM Entity - GROUP BY identifier - HAVING COUNT(*) > 1; - - -- Join across tables - SELECT e.name, COUNT(t.id) as transaction_count - FROM Entity e - LEFT JOIN Transaction t ON e.identifier = t.entity_id - GROUP BY e.name - ORDER BY transaction_count DESC; - ``` + **Common Operations:** + - Count rows: `SELECT COUNT(*) FROM Entity` + - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` + - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING + COUNT(*) > 1` + - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = + t.entity_id GROUP BY e.name` **Limits:** - Query timeout: 30 seconds @@ -247,7 +217,7 @@ def sync( Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query` **Note:** - Staging table queries are included - no credit consumption. + Staging table queries are included - no credit consumption Args: graph_id (str): @@ -278,7 +248,6 @@ async def asyncio_detailed( Execute SQL queries on DuckDB staging tables for data inspection and validation. - **Purpose:** Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. @@ -301,27 +270,13 @@ async def asyncio_detailed( - Aggregations, window functions, CTEs - Multiple table joins across staging area - **Example Queries:** - ```sql - -- Count rows in staging table - SELECT COUNT(*) FROM Entity; - - -- Check for nulls - SELECT * FROM Entity WHERE name IS NULL LIMIT 10; - - -- Find duplicates - SELECT identifier, COUNT(*) as cnt - FROM Entity - GROUP BY identifier - HAVING COUNT(*) > 1; - - -- Join across tables - SELECT e.name, COUNT(t.id) as transaction_count - FROM Entity e - LEFT JOIN Transaction t ON e.identifier = t.entity_id - GROUP BY e.name - ORDER BY transaction_count DESC; - ``` + **Common Operations:** + - Count rows: `SELECT COUNT(*) FROM Entity` + - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` + - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING + COUNT(*) > 1` + - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = + t.entity_id GROUP BY e.name` **Limits:** - Query timeout: 30 seconds @@ -334,7 +289,7 @@ async def asyncio_detailed( Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query` **Note:** - Staging table queries are included - no credit consumption. + Staging table queries are included - no credit consumption Args: graph_id (str): @@ -368,7 +323,6 @@ async def asyncio( Execute SQL queries on DuckDB staging tables for data inspection and validation. - **Purpose:** Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. @@ -391,27 +345,13 @@ async def asyncio( - Aggregations, window functions, CTEs - Multiple table joins across staging area - **Example Queries:** - ```sql - -- Count rows in staging table - SELECT COUNT(*) FROM Entity; - - -- Check for nulls - SELECT * FROM Entity WHERE name IS NULL LIMIT 10; - - -- Find duplicates - SELECT identifier, COUNT(*) as cnt - FROM Entity - GROUP BY identifier - HAVING COUNT(*) > 1; - - -- Join across tables - SELECT e.name, COUNT(t.id) as transaction_count - FROM Entity e - LEFT JOIN Transaction t ON e.identifier = t.entity_id - GROUP BY e.name - ORDER BY transaction_count DESC; - ``` + **Common Operations:** + - Count rows: `SELECT COUNT(*) FROM Entity` + - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` + - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING + COUNT(*) > 1` + - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = + t.entity_id GROUP BY e.name` **Limits:** - Query timeout: 30 seconds @@ -424,7 +364,7 @@ async def asyncio( Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query` **Note:** - Staging table queries are included - no credit consumption. + Staging table queries are included - no credit consumption Args: graph_id (str): diff --git a/robosystems_client/api/tables/update_file_status.py b/robosystems_client/api/tables/update_file_status.py index 7d4fa4f..ef41761 100644 --- a/robosystems_client/api/tables/update_file_status.py +++ b/robosystems_client/api/tables/update_file_status.py @@ -112,83 +112,57 @@ def sync_detailed( Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus ] ]: - r""" Update File Upload Status - - Update file status after upload completes. - - **Purpose:** - Mark files as uploaded after successful S3 upload. The backend validates - the file, calculates size and row count, enforces storage limits, and - registers the DuckDB table for queries. - - **Status Values:** - - `uploaded`: File successfully uploaded to S3 (triggers validation) - - `disabled`: Exclude file from ingestion - - `archived`: Soft delete file - - **What Happens on 'uploaded' Status:** - 1. Verify file exists in S3 - 2. Calculate actual file size - 3. Enforce tier storage limits - 4. Calculate or estimate row count - 5. Update table statistics - 6. Register DuckDB external table - 7. File ready for ingestion - - **Row Count Calculation:** - - **Parquet**: Exact count from file metadata - - **CSV**: Count rows (minus header) - - **JSON**: Count array elements - - **Fallback**: Estimate from file size if reading fails - - **Storage Limits:** - Enforced per subscription tier: - - Prevents uploads exceeding tier limit - - Returns HTTP 413 if limit exceeded - - Check current usage before large uploads - - **Example Response:** - ```json - { - \"status\": \"success\", - \"file_id\": \"f123\", - \"upload_status\": \"uploaded\", - \"file_size_bytes\": 1048576, - \"row_count\": 5000, - \"message\": \"File validated and ready for ingestion\" - } - ``` - - **Example Usage:** - ```bash - # After uploading file to S3 presigned URL - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Always call this after S3 upload completes - - Check response for actual row count - - Storage limit errors (413) mean tier upgrade needed - - DuckDB registration failures are non-fatal (retried later) - - **Note:** - Status updates are included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File identifier - body (FileStatusUpdate): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]] - """ + """Update File Upload Status + + Update file status after upload completes. + + Marks files as uploaded after successful S3 upload. The backend validates + the file, calculates size and row count, enforces storage limits, and + registers the DuckDB table for queries. + + **Status Values:** + - `uploaded`: File successfully uploaded to S3 (triggers validation) + - `disabled`: Exclude file from ingestion + - `archived`: Soft delete file + + **What Happens on 'uploaded' Status:** + 1. Verify file exists in S3 + 2. Calculate actual file size + 3. Enforce tier storage limits + 4. Calculate or estimate row count + 5. Update table statistics + 6. Register DuckDB external table + 7. File ready for ingestion + + **Row Count Calculation:** + - **Parquet**: Exact count from file metadata + - **CSV**: Count rows (minus header) + - **JSON**: Count array elements + - **Fallback**: Estimate from file size if reading fails + + **Storage Limits:** + Enforced per subscription tier. Returns HTTP 413 if limit exceeded. + Check current usage before large uploads. + + **Important Notes:** + - Always call this after S3 upload completes + - Check response for actual row count + - Storage limit errors (413) mean tier upgrade needed + - DuckDB registration failures are non-fatal (retried later) + - Status updates are included - no credit consumption + + Args: + graph_id (str): + file_id (str): File identifier + body (FileStatusUpdate): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -214,83 +188,57 @@ def sync( Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus ] ]: - r""" Update File Upload Status - - Update file status after upload completes. - - **Purpose:** - Mark files as uploaded after successful S3 upload. The backend validates - the file, calculates size and row count, enforces storage limits, and - registers the DuckDB table for queries. - - **Status Values:** - - `uploaded`: File successfully uploaded to S3 (triggers validation) - - `disabled`: Exclude file from ingestion - - `archived`: Soft delete file - - **What Happens on 'uploaded' Status:** - 1. Verify file exists in S3 - 2. Calculate actual file size - 3. Enforce tier storage limits - 4. Calculate or estimate row count - 5. Update table statistics - 6. Register DuckDB external table - 7. File ready for ingestion - - **Row Count Calculation:** - - **Parquet**: Exact count from file metadata - - **CSV**: Count rows (minus header) - - **JSON**: Count array elements - - **Fallback**: Estimate from file size if reading fails - - **Storage Limits:** - Enforced per subscription tier: - - Prevents uploads exceeding tier limit - - Returns HTTP 413 if limit exceeded - - Check current usage before large uploads - - **Example Response:** - ```json - { - \"status\": \"success\", - \"file_id\": \"f123\", - \"upload_status\": \"uploaded\", - \"file_size_bytes\": 1048576, - \"row_count\": 5000, - \"message\": \"File validated and ready for ingestion\" - } - ``` - - **Example Usage:** - ```bash - # After uploading file to S3 presigned URL - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Always call this after S3 upload completes - - Check response for actual row count - - Storage limit errors (413) mean tier upgrade needed - - DuckDB registration failures are non-fatal (retried later) - - **Note:** - Status updates are included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File identifier - body (FileStatusUpdate): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus] - """ + """Update File Upload Status + + Update file status after upload completes. + + Marks files as uploaded after successful S3 upload. The backend validates + the file, calculates size and row count, enforces storage limits, and + registers the DuckDB table for queries. + + **Status Values:** + - `uploaded`: File successfully uploaded to S3 (triggers validation) + - `disabled`: Exclude file from ingestion + - `archived`: Soft delete file + + **What Happens on 'uploaded' Status:** + 1. Verify file exists in S3 + 2. Calculate actual file size + 3. Enforce tier storage limits + 4. Calculate or estimate row count + 5. Update table statistics + 6. Register DuckDB external table + 7. File ready for ingestion + + **Row Count Calculation:** + - **Parquet**: Exact count from file metadata + - **CSV**: Count rows (minus header) + - **JSON**: Count array elements + - **Fallback**: Estimate from file size if reading fails + + **Storage Limits:** + Enforced per subscription tier. Returns HTTP 413 if limit exceeded. + Check current usage before large uploads. + + **Important Notes:** + - Always call this after S3 upload completes + - Check response for actual row count + - Storage limit errors (413) mean tier upgrade needed + - DuckDB registration failures are non-fatal (retried later) + - Status updates are included - no credit consumption + + Args: + graph_id (str): + file_id (str): File identifier + body (FileStatusUpdate): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus] + """ return sync_detailed( graph_id=graph_id, @@ -311,83 +259,57 @@ async def asyncio_detailed( Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus ] ]: - r""" Update File Upload Status - - Update file status after upload completes. - - **Purpose:** - Mark files as uploaded after successful S3 upload. The backend validates - the file, calculates size and row count, enforces storage limits, and - registers the DuckDB table for queries. - - **Status Values:** - - `uploaded`: File successfully uploaded to S3 (triggers validation) - - `disabled`: Exclude file from ingestion - - `archived`: Soft delete file - - **What Happens on 'uploaded' Status:** - 1. Verify file exists in S3 - 2. Calculate actual file size - 3. Enforce tier storage limits - 4. Calculate or estimate row count - 5. Update table statistics - 6. Register DuckDB external table - 7. File ready for ingestion - - **Row Count Calculation:** - - **Parquet**: Exact count from file metadata - - **CSV**: Count rows (minus header) - - **JSON**: Count array elements - - **Fallback**: Estimate from file size if reading fails - - **Storage Limits:** - Enforced per subscription tier: - - Prevents uploads exceeding tier limit - - Returns HTTP 413 if limit exceeded - - Check current usage before large uploads - - **Example Response:** - ```json - { - \"status\": \"success\", - \"file_id\": \"f123\", - \"upload_status\": \"uploaded\", - \"file_size_bytes\": 1048576, - \"row_count\": 5000, - \"message\": \"File validated and ready for ingestion\" - } - ``` - - **Example Usage:** - ```bash - # After uploading file to S3 presigned URL - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Always call this after S3 upload completes - - Check response for actual row count - - Storage limit errors (413) mean tier upgrade needed - - DuckDB registration failures are non-fatal (retried later) - - **Note:** - Status updates are included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File identifier - body (FileStatusUpdate): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]] - """ + """Update File Upload Status + + Update file status after upload completes. + + Marks files as uploaded after successful S3 upload. The backend validates + the file, calculates size and row count, enforces storage limits, and + registers the DuckDB table for queries. + + **Status Values:** + - `uploaded`: File successfully uploaded to S3 (triggers validation) + - `disabled`: Exclude file from ingestion + - `archived`: Soft delete file + + **What Happens on 'uploaded' Status:** + 1. Verify file exists in S3 + 2. Calculate actual file size + 3. Enforce tier storage limits + 4. Calculate or estimate row count + 5. Update table statistics + 6. Register DuckDB external table + 7. File ready for ingestion + + **Row Count Calculation:** + - **Parquet**: Exact count from file metadata + - **CSV**: Count rows (minus header) + - **JSON**: Count array elements + - **Fallback**: Estimate from file size if reading fails + + **Storage Limits:** + Enforced per subscription tier. Returns HTTP 413 if limit exceeded. + Check current usage before large uploads. + + **Important Notes:** + - Always call this after S3 upload completes + - Check response for actual row count + - Storage limit errors (413) mean tier upgrade needed + - DuckDB registration failures are non-fatal (retried later) + - Status updates are included - no credit consumption + + Args: + graph_id (str): + file_id (str): File identifier + body (FileStatusUpdate): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]] + """ kwargs = _get_kwargs( graph_id=graph_id, @@ -411,83 +333,57 @@ async def asyncio( Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus ] ]: - r""" Update File Upload Status - - Update file status after upload completes. - - **Purpose:** - Mark files as uploaded after successful S3 upload. The backend validates - the file, calculates size and row count, enforces storage limits, and - registers the DuckDB table for queries. - - **Status Values:** - - `uploaded`: File successfully uploaded to S3 (triggers validation) - - `disabled`: Exclude file from ingestion - - `archived`: Soft delete file - - **What Happens on 'uploaded' Status:** - 1. Verify file exists in S3 - 2. Calculate actual file size - 3. Enforce tier storage limits - 4. Calculate or estimate row count - 5. Update table statistics - 6. Register DuckDB external table - 7. File ready for ingestion - - **Row Count Calculation:** - - **Parquet**: Exact count from file metadata - - **CSV**: Count rows (minus header) - - **JSON**: Count array elements - - **Fallback**: Estimate from file size if reading fails - - **Storage Limits:** - Enforced per subscription tier: - - Prevents uploads exceeding tier limit - - Returns HTTP 413 if limit exceeded - - Check current usage before large uploads - - **Example Response:** - ```json - { - \"status\": \"success\", - \"file_id\": \"f123\", - \"upload_status\": \"uploaded\", - \"file_size_bytes\": 1048576, - \"row_count\": 5000, - \"message\": \"File validated and ready for ingestion\" - } - ``` - - **Example Usage:** - ```bash - # After uploading file to S3 presigned URL - curl -X PATCH \"https://api.robosystems.ai/v1/graphs/kg123/tables/files/f123\" \ - -H \"Authorization: Bearer YOUR_TOKEN\" \ - -H \"Content-Type: application/json\" \ - -d '{\"status\": \"uploaded\"}' - ``` - - **Tips:** - - Always call this after S3 upload completes - - Check response for actual row count - - Storage limit errors (413) mean tier upgrade needed - - DuckDB registration failures are non-fatal (retried later) - - **Note:** - Status updates are included - no credit consumption. - - Args: - graph_id (str): - file_id (str): File identifier - body (FileStatusUpdate): - - Raises: - errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. - httpx.TimeoutException: If the request takes longer than Client.timeout. - - Returns: - Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus] - """ + """Update File Upload Status + + Update file status after upload completes. + + Marks files as uploaded after successful S3 upload. The backend validates + the file, calculates size and row count, enforces storage limits, and + registers the DuckDB table for queries. + + **Status Values:** + - `uploaded`: File successfully uploaded to S3 (triggers validation) + - `disabled`: Exclude file from ingestion + - `archived`: Soft delete file + + **What Happens on 'uploaded' Status:** + 1. Verify file exists in S3 + 2. Calculate actual file size + 3. Enforce tier storage limits + 4. Calculate or estimate row count + 5. Update table statistics + 6. Register DuckDB external table + 7. File ready for ingestion + + **Row Count Calculation:** + - **Parquet**: Exact count from file metadata + - **CSV**: Count rows (minus header) + - **JSON**: Count array elements + - **Fallback**: Estimate from file size if reading fails + + **Storage Limits:** + Enforced per subscription tier. Returns HTTP 413 if limit exceeded. + Check current usage before large uploads. + + **Important Notes:** + - Always call this after S3 upload completes + - Check response for actual row count + - Storage limit errors (413) mean tier upgrade needed + - DuckDB registration failures are non-fatal (retried later) + - Status updates are included - no credit consumption + + Args: + graph_id (str): + file_id (str): File identifier + body (FileStatusUpdate): + + Raises: + errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. + httpx.TimeoutException: If the request takes longer than Client.timeout. + + Returns: + Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus] + """ return ( await asyncio_detailed( diff --git a/robosystems_client/models/create_graph_request.py b/robosystems_client/models/create_graph_request.py index 45d3b24..8530ca6 100644 --- a/robosystems_client/models/create_graph_request.py +++ b/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. diff --git a/robosystems_client/models/cypher_query_request.py b/robosystems_client/models/cypher_query_request.py index b65f86a..1d3f827 100644 --- a/robosystems_client/models/cypher_query_request.py +++ b/robosystems_client/models/cypher_query_request.py @@ -2,7 +2,6 @@ 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 @@ -28,7 +27,6 @@ class CypherQueryRequest: 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 +50,7 @@ def to_dict(self) -> dict[str, Any]: timeout = self.timeout field_dict: dict[str, Any] = {} - field_dict.update(self.additional_properties) + field_dict.update( { "query": query, @@ -108,21 +106,4 @@ def _parse_timeout(data: object) -> Union[None, Unset, int]: 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 From b1e93adb74e34dabdef669554fcb6ead213d24f8 Mon Sep 17 00:00:00 2001 From: Joey French Date: Wed, 29 Oct 2025 02:32:38 -0500 Subject: [PATCH 3/4] Enhance API documentation and improve security practices ## Summary This commit updates the API documentation across various endpoints and reinforces security best practices for query execution. ## Key Changes - Added detailed documentation on using parameterized queries to prevent injection attacks in `execute_cypher_query`, `query_tables`, and `table_query_request`. - Revised the `export_graph_schema` and `get_graph_schema` endpoints to clarify the distinction between declared and runtime schemas, including examples of usage. - Removed the deprecated `GetGraphSchemaResponseGetgraphschema` model to streamline the codebase. ## Testing Notes - Verify that the updated documentation accurately reflects the API behavior and security practices. - Test all affected endpoints to ensure they function correctly with the new documentation. ## Infrastructure Considerations These enhancements improve developer experience and security, reducing the risk of injection vulnerabilities during query execution. --- .../api/query/execute_cypher_query.py | 48 +++- .../api/schema/export_graph_schema.py | 242 ++++++++++++++++-- .../api/schema/get_graph_schema.py | 162 +++++++++--- robosystems_client/api/tables/query_tables.py | 68 ++++- robosystems_client/models/__init__.py | 8 +- .../models/cypher_query_request.py | 6 +- .../models/schema_export_response.py | 6 +- .../models/schema_info_response.py | 77 ++++++ ...hema.py => schema_info_response_schema.py} | 12 +- .../models/schema_validation_response.py | 13 +- .../models/table_query_request.py | 39 ++- 11 files changed, 590 insertions(+), 91 deletions(-) create mode 100644 robosystems_client/models/schema_info_response.py rename robosystems_client/models/{get_graph_schema_response_getgraphschema.py => schema_info_response_schema.py} (76%) diff --git a/robosystems_client/api/query/execute_cypher_query.py b/robosystems_client/api/query/execute_cypher_query.py index d9b0661..084057b 100644 --- a/robosystems_client/api/query/execute_cypher_query.py +++ b/robosystems_client/api/query/execute_cypher_query.py @@ -123,7 +123,7 @@ def sync_detailed( chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Response[Union[Any, HTTPValidationError]]: - """Execute Cypher Query (Read-Only) + r"""Execute Cypher Query (Read-Only) Execute a read-only Cypher query with intelligent response optimization. @@ -133,6 +133,16 @@ def sync_detailed( 1. Create file upload: `POST /v1/graphs/{graph_id}/tables/{table_name}/files` 2. Ingest to graph: `POST /v1/graphs/{graph_id}/tables/ingest` + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string interpolation to prevent injection attacks: + - āœ… SAFE: `MATCH (n:Entity {type: $entity_type}) RETURN n` with `parameters: {\"entity_type\": + \"Company\"}` + - āŒ UNSAFE: `MATCH (n:Entity {type: \"Company\"}) RETURN n` with user input concatenated into query + string + + Query parameters provide automatic escaping and type safety. All examples in this API use + parameterized queries. + This endpoint automatically selects the best execution strategy based on: - Query characteristics (size, complexity) - Client capabilities (SSE, NDJSON, JSON) @@ -218,7 +228,7 @@ def sync( chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Optional[Union[Any, HTTPValidationError]]: - """Execute Cypher Query (Read-Only) + r"""Execute Cypher Query (Read-Only) Execute a read-only Cypher query with intelligent response optimization. @@ -228,6 +238,16 @@ def sync( 1. Create file upload: `POST /v1/graphs/{graph_id}/tables/{table_name}/files` 2. Ingest to graph: `POST /v1/graphs/{graph_id}/tables/ingest` + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string interpolation to prevent injection attacks: + - āœ… SAFE: `MATCH (n:Entity {type: $entity_type}) RETURN n` with `parameters: {\"entity_type\": + \"Company\"}` + - āŒ UNSAFE: `MATCH (n:Entity {type: \"Company\"}) RETURN n` with user input concatenated into query + string + + Query parameters provide automatic escaping and type safety. All examples in this API use + parameterized queries. + This endpoint automatically selects the best execution strategy based on: - Query characteristics (size, complexity) - Client capabilities (SSE, NDJSON, JSON) @@ -308,7 +328,7 @@ async def asyncio_detailed( chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Response[Union[Any, HTTPValidationError]]: - """Execute Cypher Query (Read-Only) + r"""Execute Cypher Query (Read-Only) Execute a read-only Cypher query with intelligent response optimization. @@ -318,6 +338,16 @@ async def asyncio_detailed( 1. Create file upload: `POST /v1/graphs/{graph_id}/tables/{table_name}/files` 2. Ingest to graph: `POST /v1/graphs/{graph_id}/tables/ingest` + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string interpolation to prevent injection attacks: + - āœ… SAFE: `MATCH (n:Entity {type: $entity_type}) RETURN n` with `parameters: {\"entity_type\": + \"Company\"}` + - āŒ UNSAFE: `MATCH (n:Entity {type: \"Company\"}) RETURN n` with user input concatenated into query + string + + Query parameters provide automatic escaping and type safety. All examples in this API use + parameterized queries. + This endpoint automatically selects the best execution strategy based on: - Query characteristics (size, complexity) - Client capabilities (SSE, NDJSON, JSON) @@ -401,7 +431,7 @@ async def asyncio( chunk_size: Union[None, Unset, int] = UNSET, test_mode: Union[Unset, bool] = False, ) -> Optional[Union[Any, HTTPValidationError]]: - """Execute Cypher Query (Read-Only) + r"""Execute Cypher Query (Read-Only) Execute a read-only Cypher query with intelligent response optimization. @@ -411,6 +441,16 @@ async def asyncio( 1. Create file upload: `POST /v1/graphs/{graph_id}/tables/{table_name}/files` 2. Ingest to graph: `POST /v1/graphs/{graph_id}/tables/ingest` + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string interpolation to prevent injection attacks: + - āœ… SAFE: `MATCH (n:Entity {type: $entity_type}) RETURN n` with `parameters: {\"entity_type\": + \"Company\"}` + - āŒ UNSAFE: `MATCH (n:Entity {type: \"Company\"}) RETURN n` with user input concatenated into query + string + + Query parameters provide automatic escaping and type safety. All examples in this API use + parameterized queries. + This endpoint automatically selects the best execution strategy based on: - Query characteristics (size, complexity) - Client capabilities (SSE, NDJSON, JSON) diff --git a/robosystems_client/api/schema/export_graph_schema.py b/robosystems_client/api/schema/export_graph_schema.py index c1b9fdb..a2332e6 100644 --- a/robosystems_client/api/schema/export_graph_schema.py +++ b/robosystems_client/api/schema/export_graph_schema.py @@ -1,5 +1,5 @@ from http import HTTPStatus -from typing import Any, Optional, Union +from typing import Any, Optional, Union, cast import httpx @@ -35,17 +35,29 @@ def _get_kwargs( def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]: +) -> Optional[Union[Any, HTTPValidationError, SchemaExportResponse]]: if response.status_code == 200: response_200 = SchemaExportResponse.from_dict(response.json()) return response_200 + if response.status_code == 403: + response_403 = cast(Any, None) + return response_403 + + 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: @@ -54,7 +66,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[HTTPValidationError, SchemaExportResponse]]: +) -> Response[Union[Any, HTTPValidationError, SchemaExportResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,23 +81,69 @@ def sync_detailed( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, -) -> Response[Union[HTTPValidationError, SchemaExportResponse]]: - """Export Graph Schema +) -> Response[Union[Any, HTTPValidationError, SchemaExportResponse]]: + """Export Declared Graph Schema + + Export the declared schema definition of an existing graph. + + ## What This Returns + + This endpoint returns the **original schema definition** that was used to create the graph: + - The schema as it was **declared** during graph creation + - Complete node and relationship definitions + - Property types and constraints + - Schema metadata (name, version, type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema/export`) when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + **Use `/schema` instead** when you need: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes (discovered from data) + - Current runtime database structure for querying + + ## Export Formats + + ### JSON Format (`format=json`) + Returns structured JSON with nodes, relationships, and properties. + Best for programmatic access and API integration. + + ### YAML Format (`format=yaml`) + Returns human-readable YAML with comments. + Best for documentation and configuration management. + + ### Cypher DDL Format (`format=cypher`) + Returns Cypher CREATE statements for recreating the schema. + Best for database migration and replication. + + ## Data Statistics - Export the schema of an existing graph in JSON, YAML, or Cypher format + Set `include_data_stats=true` to include: + - Node counts by label + - Relationship counts by type + - Total nodes and relationships + + This combines declared schema with runtime statistics. + + This operation is included - no credit consumption required. Args: graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph - Default: False. + (node counts, relationship counts) Default: False. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, SchemaExportResponse]] + Response[Union[Any, HTTPValidationError, SchemaExportResponse]] """ kwargs = _get_kwargs( @@ -107,23 +165,69 @@ def sync( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, -) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]: - """Export Graph Schema +) -> Optional[Union[Any, HTTPValidationError, SchemaExportResponse]]: + """Export Declared Graph Schema + + Export the declared schema definition of an existing graph. + + ## What This Returns + + This endpoint returns the **original schema definition** that was used to create the graph: + - The schema as it was **declared** during graph creation + - Complete node and relationship definitions + - Property types and constraints + - Schema metadata (name, version, type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema/export`) when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + **Use `/schema` instead** when you need: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes (discovered from data) + - Current runtime database structure for querying + + ## Export Formats + + ### JSON Format (`format=json`) + Returns structured JSON with nodes, relationships, and properties. + Best for programmatic access and API integration. + + ### YAML Format (`format=yaml`) + Returns human-readable YAML with comments. + Best for documentation and configuration management. + + ### Cypher DDL Format (`format=cypher`) + Returns Cypher CREATE statements for recreating the schema. + Best for database migration and replication. + + ## Data Statistics + + Set `include_data_stats=true` to include: + - Node counts by label + - Relationship counts by type + - Total nodes and relationships - Export the schema of an existing graph in JSON, YAML, or Cypher format + This combines declared schema with runtime statistics. + + This operation is included - no credit consumption required. Args: graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph - Default: False. + (node counts, relationship counts) Default: False. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, SchemaExportResponse] + Union[Any, HTTPValidationError, SchemaExportResponse] """ return sync_detailed( @@ -140,23 +244,69 @@ async def asyncio_detailed( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, -) -> Response[Union[HTTPValidationError, SchemaExportResponse]]: - """Export Graph Schema +) -> Response[Union[Any, HTTPValidationError, SchemaExportResponse]]: + """Export Declared Graph Schema + + Export the declared schema definition of an existing graph. + + ## What This Returns + + This endpoint returns the **original schema definition** that was used to create the graph: + - The schema as it was **declared** during graph creation + - Complete node and relationship definitions + - Property types and constraints + - Schema metadata (name, version, type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema/export`) when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + **Use `/schema` instead** when you need: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes (discovered from data) + - Current runtime database structure for querying + + ## Export Formats + + ### JSON Format (`format=json`) + Returns structured JSON with nodes, relationships, and properties. + Best for programmatic access and API integration. + + ### YAML Format (`format=yaml`) + Returns human-readable YAML with comments. + Best for documentation and configuration management. + + ### Cypher DDL Format (`format=cypher`) + Returns Cypher CREATE statements for recreating the schema. + Best for database migration and replication. + + ## Data Statistics + + Set `include_data_stats=true` to include: + - Node counts by label + - Relationship counts by type + - Total nodes and relationships + + This combines declared schema with runtime statistics. - Export the schema of an existing graph in JSON, YAML, or Cypher format + This operation is included - no credit consumption required. Args: graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph - Default: False. + (node counts, relationship counts) Default: False. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[HTTPValidationError, SchemaExportResponse]] + Response[Union[Any, HTTPValidationError, SchemaExportResponse]] """ kwargs = _get_kwargs( @@ -176,23 +326,69 @@ async def asyncio( client: AuthenticatedClient, format_: Union[Unset, str] = "json", include_data_stats: Union[Unset, bool] = False, -) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]: - """Export Graph Schema +) -> Optional[Union[Any, HTTPValidationError, SchemaExportResponse]]: + """Export Declared Graph Schema + + Export the declared schema definition of an existing graph. + + ## What This Returns + + This endpoint returns the **original schema definition** that was used to create the graph: + - The schema as it was **declared** during graph creation + - Complete node and relationship definitions + - Property types and constraints + - Schema metadata (name, version, type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema/export`) when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + **Use `/schema` instead** when you need: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes (discovered from data) + - Current runtime database structure for querying + + ## Export Formats + + ### JSON Format (`format=json`) + Returns structured JSON with nodes, relationships, and properties. + Best for programmatic access and API integration. + + ### YAML Format (`format=yaml`) + Returns human-readable YAML with comments. + Best for documentation and configuration management. + + ### Cypher DDL Format (`format=cypher`) + Returns Cypher CREATE statements for recreating the schema. + Best for database migration and replication. + + ## Data Statistics + + Set `include_data_stats=true` to include: + - Node counts by label + - Relationship counts by type + - Total nodes and relationships + + This combines declared schema with runtime statistics. - Export the schema of an existing graph in JSON, YAML, or Cypher format + This operation is included - no credit consumption required. Args: graph_id (str): format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'. include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph - Default: False. + (node counts, relationship counts) Default: False. Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[HTTPValidationError, SchemaExportResponse] + Union[Any, HTTPValidationError, SchemaExportResponse] """ return ( diff --git a/robosystems_client/api/schema/get_graph_schema.py b/robosystems_client/api/schema/get_graph_schema.py index c68f5d6..bef70d4 100644 --- a/robosystems_client/api/schema/get_graph_schema.py +++ b/robosystems_client/api/schema/get_graph_schema.py @@ -5,10 +5,8 @@ from ... import errors from ...client import AuthenticatedClient, Client -from ...models.get_graph_schema_response_getgraphschema import ( - GetGraphSchemaResponseGetgraphschema, -) from ...models.http_validation_error import HTTPValidationError +from ...models.schema_info_response import SchemaInfoResponse from ...types import Response @@ -25,9 +23,9 @@ def _get_kwargs( def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Optional[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Optional[Union[Any, HTTPValidationError, SchemaInfoResponse]]: if response.status_code == 200: - response_200 = GetGraphSchemaResponseGetgraphschema.from_dict(response.json()) + response_200 = SchemaInfoResponse.from_dict(response.json()) return response_200 @@ -44,6 +42,10 @@ def _parse_response( response_500 = cast(Any, None) return response_500 + if response.status_code == 504: + response_504 = cast(Any, None) + return response_504 + if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) else: @@ -52,7 +54,7 @@ def _parse_response( def _build_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response -) -> Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Response[Union[Any, HTTPValidationError, SchemaInfoResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -65,18 +67,43 @@ def sync_detailed( graph_id: str, *, client: AuthenticatedClient, -) -> Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Response[Union[Any, HTTPValidationError, SchemaInfoResponse]]: """Get Runtime Graph Schema Get runtime schema information for the specified graph database. - This endpoint inspects the actual graph database structure and returns: + ## What This Returns + + This endpoint inspects the **actual current state** of the graph database and returns: - **Node Labels**: All node types currently in the database - **Relationship Types**: All relationship types currently in the database - - **Node Properties**: Properties for each node type (limited to first 10 for performance) + - **Node Properties**: Properties discovered from actual data (up to 10 properties per node type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema`) when you need to know: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes + - What relationships have been created + - Current database structure for querying + + **Use `/schema/export` instead** when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + ## Example Use Cases + + - **Building queries**: See what node labels and properties exist to write accurate Cypher + - **Data exploration**: Discover what's in an unfamiliar graph + - **Schema drift detection**: Compare runtime vs declared schema + - **API integration**: Dynamically adapt to current graph structure + + ## Performance Note - This shows what actually exists in the database right now - the runtime state. - For the declared schema definition, use GET /schema/export instead. + Property discovery is limited to 10 properties per node type for performance. + For complete schema definitions, use `/schema/export`. This operation is included - no credit consumption required. @@ -88,7 +115,7 @@ def sync_detailed( httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]] + Response[Union[Any, HTTPValidationError, SchemaInfoResponse]] """ kwargs = _get_kwargs( @@ -106,18 +133,43 @@ def sync( graph_id: str, *, client: AuthenticatedClient, -) -> Optional[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Optional[Union[Any, HTTPValidationError, SchemaInfoResponse]]: """Get Runtime Graph Schema Get runtime schema information for the specified graph database. - This endpoint inspects the actual graph database structure and returns: + ## What This Returns + + This endpoint inspects the **actual current state** of the graph database and returns: - **Node Labels**: All node types currently in the database - **Relationship Types**: All relationship types currently in the database - - **Node Properties**: Properties for each node type (limited to first 10 for performance) + - **Node Properties**: Properties discovered from actual data (up to 10 properties per node type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema`) when you need to know: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes + - What relationships have been created + - Current database structure for querying + + **Use `/schema/export` instead** when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + ## Example Use Cases + + - **Building queries**: See what node labels and properties exist to write accurate Cypher + - **Data exploration**: Discover what's in an unfamiliar graph + - **Schema drift detection**: Compare runtime vs declared schema + - **API integration**: Dynamically adapt to current graph structure + + ## Performance Note - This shows what actually exists in the database right now - the runtime state. - For the declared schema definition, use GET /schema/export instead. + Property discovery is limited to 10 properties per node type for performance. + For complete schema definitions, use `/schema/export`. This operation is included - no credit consumption required. @@ -129,7 +181,7 @@ def sync( httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError] + Union[Any, HTTPValidationError, SchemaInfoResponse] """ return sync_detailed( @@ -142,18 +194,43 @@ async def asyncio_detailed( graph_id: str, *, client: AuthenticatedClient, -) -> Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Response[Union[Any, HTTPValidationError, SchemaInfoResponse]]: """Get Runtime Graph Schema Get runtime schema information for the specified graph database. - This endpoint inspects the actual graph database structure and returns: + ## What This Returns + + This endpoint inspects the **actual current state** of the graph database and returns: - **Node Labels**: All node types currently in the database - **Relationship Types**: All relationship types currently in the database - - **Node Properties**: Properties for each node type (limited to first 10 for performance) + - **Node Properties**: Properties discovered from actual data (up to 10 properties per node type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema`) when you need to know: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes + - What relationships have been created + - Current database structure for querying + + **Use `/schema/export` instead** when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + ## Example Use Cases + + - **Building queries**: See what node labels and properties exist to write accurate Cypher + - **Data exploration**: Discover what's in an unfamiliar graph + - **Schema drift detection**: Compare runtime vs declared schema + - **API integration**: Dynamically adapt to current graph structure + + ## Performance Note - This shows what actually exists in the database right now - the runtime state. - For the declared schema definition, use GET /schema/export instead. + Property discovery is limited to 10 properties per node type for performance. + For complete schema definitions, use `/schema/export`. This operation is included - no credit consumption required. @@ -165,7 +242,7 @@ async def asyncio_detailed( httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Response[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]] + Response[Union[Any, HTTPValidationError, SchemaInfoResponse]] """ kwargs = _get_kwargs( @@ -181,18 +258,43 @@ async def asyncio( graph_id: str, *, client: AuthenticatedClient, -) -> Optional[Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError]]: +) -> Optional[Union[Any, HTTPValidationError, SchemaInfoResponse]]: """Get Runtime Graph Schema Get runtime schema information for the specified graph database. - This endpoint inspects the actual graph database structure and returns: + ## What This Returns + + This endpoint inspects the **actual current state** of the graph database and returns: - **Node Labels**: All node types currently in the database - **Relationship Types**: All relationship types currently in the database - - **Node Properties**: Properties for each node type (limited to first 10 for performance) + - **Node Properties**: Properties discovered from actual data (up to 10 properties per node type) + + ## Runtime vs Declared Schema + + **Use this endpoint** (`/schema`) when you need to know: + - What data is ACTUALLY in the database right now + - What properties exist on real nodes + - What relationships have been created + - Current database structure for querying + + **Use `/schema/export` instead** when you need: + - The original schema definition used to create the graph + - Schema in a specific format (JSON, YAML, Cypher DDL) + - Schema for documentation or version control + - Schema to replicate in another graph + + ## Example Use Cases + + - **Building queries**: See what node labels and properties exist to write accurate Cypher + - **Data exploration**: Discover what's in an unfamiliar graph + - **Schema drift detection**: Compare runtime vs declared schema + - **API integration**: Dynamically adapt to current graph structure + + ## Performance Note - This shows what actually exists in the database right now - the runtime state. - For the declared schema definition, use GET /schema/export instead. + Property discovery is limited to 10 properties per node type for performance. + For complete schema definitions, use `/schema/export`. This operation is included - no credit consumption required. @@ -204,7 +306,7 @@ async def asyncio( httpx.TimeoutException: If the request takes longer than Client.timeout. Returns: - Union[Any, GetGraphSchemaResponseGetgraphschema, HTTPValidationError] + Union[Any, HTTPValidationError, SchemaInfoResponse] """ return ( diff --git a/robosystems_client/api/tables/query_tables.py b/robosystems_client/api/tables/query_tables.py index d68151a..0999e17 100644 --- a/robosystems_client/api/tables/query_tables.py +++ b/robosystems_client/api/tables/query_tables.py @@ -95,13 +95,22 @@ def sync_detailed( client: AuthenticatedClient, body: TableQueryRequest, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: - """Query Staging Tables with SQL + r"""Query Staging Tables with SQL Execute SQL queries on DuckDB staging tables for data inspection and validation. Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string concatenation to prevent SQL injection: + - āœ… SAFE: `SELECT * FROM Entity WHERE type = ? LIMIT ?` with `parameters: [\"Company\", 100]` + - āŒ UNSAFE: `SELECT * FROM Entity WHERE type = 'Company' LIMIT 100` with user input concatenated + into SQL string + + Query parameters provide automatic escaping and type safety. Use `?` placeholders with parameters + array. + **Use Cases:** - Validate data quality before graph ingestion - Inspect row-level data for debugging @@ -123,11 +132,13 @@ def sync_detailed( **Common Operations:** - Count rows: `SELECT COUNT(*) FROM Entity` + - Filter by type: `SELECT * FROM Entity WHERE entity_type = ? LIMIT ?` with `parameters: + [\"Company\", 100]` - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING COUNT(*) > 1` - - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = - t.entity_id GROUP BY e.name` + - Filter amounts: `SELECT * FROM Transaction WHERE amount > ? AND date >= ?` with `parameters: + [1000, \"2024-01-01\"]` **Limits:** - Query timeout: 30 seconds @@ -172,13 +183,22 @@ def sync( client: AuthenticatedClient, body: TableQueryRequest, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: - """Query Staging Tables with SQL + r"""Query Staging Tables with SQL Execute SQL queries on DuckDB staging tables for data inspection and validation. Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string concatenation to prevent SQL injection: + - āœ… SAFE: `SELECT * FROM Entity WHERE type = ? LIMIT ?` with `parameters: [\"Company\", 100]` + - āŒ UNSAFE: `SELECT * FROM Entity WHERE type = 'Company' LIMIT 100` with user input concatenated + into SQL string + + Query parameters provide automatic escaping and type safety. Use `?` placeholders with parameters + array. + **Use Cases:** - Validate data quality before graph ingestion - Inspect row-level data for debugging @@ -200,11 +220,13 @@ def sync( **Common Operations:** - Count rows: `SELECT COUNT(*) FROM Entity` + - Filter by type: `SELECT * FROM Entity WHERE entity_type = ? LIMIT ?` with `parameters: + [\"Company\", 100]` - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING COUNT(*) > 1` - - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = - t.entity_id GROUP BY e.name` + - Filter amounts: `SELECT * FROM Transaction WHERE amount > ? AND date >= ?` with `parameters: + [1000, \"2024-01-01\"]` **Limits:** - Query timeout: 30 seconds @@ -244,13 +266,22 @@ async def asyncio_detailed( client: AuthenticatedClient, body: TableQueryRequest, ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: - """Query Staging Tables with SQL + r"""Query Staging Tables with SQL Execute SQL queries on DuckDB staging tables for data inspection and validation. Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string concatenation to prevent SQL injection: + - āœ… SAFE: `SELECT * FROM Entity WHERE type = ? LIMIT ?` with `parameters: [\"Company\", 100]` + - āŒ UNSAFE: `SELECT * FROM Entity WHERE type = 'Company' LIMIT 100` with user input concatenated + into SQL string + + Query parameters provide automatic escaping and type safety. Use `?` placeholders with parameters + array. + **Use Cases:** - Validate data quality before graph ingestion - Inspect row-level data for debugging @@ -272,11 +303,13 @@ async def asyncio_detailed( **Common Operations:** - Count rows: `SELECT COUNT(*) FROM Entity` + - Filter by type: `SELECT * FROM Entity WHERE entity_type = ? LIMIT ?` with `parameters: + [\"Company\", 100]` - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING COUNT(*) > 1` - - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = - t.entity_id GROUP BY e.name` + - Filter amounts: `SELECT * FROM Transaction WHERE amount > ? AND date >= ?` with `parameters: + [1000, \"2024-01-01\"]` **Limits:** - Query timeout: 30 seconds @@ -319,13 +352,22 @@ async def asyncio( client: AuthenticatedClient, body: TableQueryRequest, ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]: - """Query Staging Tables with SQL + r"""Query Staging Tables with SQL Execute SQL queries on DuckDB staging tables for data inspection and validation. Query raw staging data directly with SQL before ingestion into the graph database. Useful for data quality checks, validation, and exploratory analysis. + **Security Best Practice - Use Parameterized Queries:** + ALWAYS use query parameters instead of string concatenation to prevent SQL injection: + - āœ… SAFE: `SELECT * FROM Entity WHERE type = ? LIMIT ?` with `parameters: [\"Company\", 100]` + - āŒ UNSAFE: `SELECT * FROM Entity WHERE type = 'Company' LIMIT 100` with user input concatenated + into SQL string + + Query parameters provide automatic escaping and type safety. Use `?` placeholders with parameters + array. + **Use Cases:** - Validate data quality before graph ingestion - Inspect row-level data for debugging @@ -347,11 +389,13 @@ async def asyncio( **Common Operations:** - Count rows: `SELECT COUNT(*) FROM Entity` + - Filter by type: `SELECT * FROM Entity WHERE entity_type = ? LIMIT ?` with `parameters: + [\"Company\", 100]` - Check for nulls: `SELECT * FROM Entity WHERE name IS NULL LIMIT 10` - Find duplicates: `SELECT identifier, COUNT(*) as cnt FROM Entity GROUP BY identifier HAVING COUNT(*) > 1` - - Join tables: `SELECT e.name, COUNT(t.id) FROM Entity e LEFT JOIN Transaction t ON e.identifier = - t.entity_id GROUP BY e.name` + - Filter amounts: `SELECT * FROM Transaction WHERE amount > ? AND date >= ?` with `parameters: + [1000, \"2024-01-01\"]` **Limits:** - Query timeout: 30 seconds diff --git a/robosystems_client/models/__init__.py b/robosystems_client/models/__init__.py index 4c40b7a..b2e0021 100644 --- a/robosystems_client/models/__init__.py +++ b/robosystems_client/models/__init__.py @@ -124,9 +124,6 @@ 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_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 @@ "GetGraphBillingHistoryResponseGetgraphbillinghistory", "GetGraphLimitsResponseGetgraphlimits", "GetGraphMonthlyBillResponseGetgraphmonthlybill", - "GetGraphSchemaResponseGetgraphschema", "GetGraphUsageDetailsResponseGetgraphusagedetails", "GetOperationStatusResponseGetoperationstatus", "GetSharedRepositoryLimitsResponseGetsharedrepositorylimits", @@ -417,6 +415,8 @@ "SchemaExportResponse", "SchemaExportResponseDataStatsType0", "SchemaExportResponseSchemaDefinitionType0", + "SchemaInfoResponse", + "SchemaInfoResponseSchema", "SchemaValidationRequest", "SchemaValidationRequestSchemaDefinitionType0", "SchemaValidationResponse", diff --git a/robosystems_client/models/cypher_query_request.py b/robosystems_client/models/cypher_query_request.py index 1d3f827..982ca05 100644 --- a/robosystems_client/models/cypher_query_request.py +++ b/robosystems_client/models/cypher_query_request.py @@ -19,8 +19,10 @@ 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. """ diff --git a/robosystems_client/models/schema_export_response.py b/robosystems_client/models/schema_export_response.py index dffd5fc..da700c2 100644 --- a/robosystems_client/models/schema_export_response.py +++ b/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 diff --git a/robosystems_client/models/schema_info_response.py b/robosystems_client/models/schema_info_response.py new file mode 100644 index 0000000..527217c --- /dev/null +++ b/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 diff --git a/robosystems_client/models/get_graph_schema_response_getgraphschema.py b/robosystems_client/models/schema_info_response_schema.py similarity index 76% rename from robosystems_client/models/get_graph_schema_response_getgraphschema.py rename to robosystems_client/models/schema_info_response_schema.py index 036899e..bf5551a 100644 --- a/robosystems_client/models/get_graph_schema_response_getgraphschema.py +++ b/robosystems_client/models/schema_info_response_schema.py @@ -4,12 +4,12 @@ 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 @@ def to_dict(self) -> dict[str, Any]: @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]: diff --git a/robosystems_client/models/schema_validation_response.py b/robosystems_client/models/schema_validation_response.py index 2a9238d..9dee137 100644 --- a/robosystems_client/models/schema_validation_response.py +++ b/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 diff --git a/robosystems_client/models/table_query_request.py b/robosystems_client/models/table_query_request.py index 7207df4..a239c91 100644 --- a/robosystems_client/models/table_query_request.py +++ b/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 @@ 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 @@ def to_dict(self) -> dict[str, Any]: "sql": sql, } ) + if parameters is not UNSET: + field_dict["parameters"] = parameters return field_dict @@ -33,8 +50,26 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: 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 From 24d64cbbc431e07f995aff2eb701b04a179175d1 Mon Sep 17 00:00:00 2001 From: Joey French Date: Wed, 29 Oct 2025 02:58:48 -0500 Subject: [PATCH 4/4] Enhance NDJSON response handling in execute_cypher_query This commit updates the `_parse_response` function to handle NDJSON responses more effectively. If the response's content type indicates NDJSON, the function will now return `None` instead of attempting to parse the response as JSON. This change improves the robustness of the query execution process when dealing with different response formats. Testing Notes: - Verify that the function correctly identifies and handles NDJSON responses without errors. - Ensure that other response types continue to be processed as expected. --- robosystems_client/api/query/execute_cypher_query.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/robosystems_client/api/query/execute_cypher_query.py b/robosystems_client/api/query/execute_cypher_query.py index 084057b..c0d311e 100644 --- a/robosystems_client/api/query/execute_cypher_query.py +++ b/robosystems_client/api/query/execute_cypher_query.py @@ -61,6 +61,12 @@ def _parse_response( *, client: Union[AuthenticatedClient, Client], response: httpx.Response ) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: + content_type = response.headers.get("content-type", "") + if ( + "application/x-ndjson" in content_type + or response.headers.get("x-stream-format") == "ndjson" + ): + return None response_200 = response.json() return response_200