robosystems-client 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

robosystems_client/api/user_limits/get_all_shared_repository_limits.py

@@ -0,0 +1,223 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_all_shared_repository_limits_response_getallsharedrepositorylimits import (
+  GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+)
+from ...models.http_validation_error import HTTPValidationError
+from ...types import UNSET, Response, Unset
+
+
+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
+
+  _kwargs: dict[str, Any] = {
+    "method": "get",
+    "url": "/v1/user/limits/shared-repositories/summary",
+    "cookies": cookies,
+  }
+
+  _kwargs["headers"] = headers
+  return _kwargs
+
+
+def _parse_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  if response.status_code == 200:
+    response_200 = (
+      GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits.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:
+    return None
+
+
+def _build_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  return Response(
+    status_code=HTTPStatus(response.status_code),
+    content=response.content,
+    headers=response.headers,
+    parsed=_parse_response(client=client, response=response),
+  )
+
+
+def sync_detailed(
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Response[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  """Get all shared repository limits
+
+   Get rate limit status for all shared repositories the user has access to.
+
+  Args:
+      authorization (Union[None, Unset, str]):
+      auth_token (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]]
+  """
+
+  kwargs = _get_kwargs(
+    authorization=authorization,
+    auth_token=auth_token,
+  )
+
+  response = client.get_httpx_client().request(
+    **kwargs,
+  )
+
+  return _build_response(client=client, response=response)
+
+
+def sync(
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Optional[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  """Get all shared repository limits
+
+   Get rate limit status for all shared repositories the user has access to.
+
+  Args:
+      authorization (Union[None, Unset, str]):
+      auth_token (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]
+  """
+
+  return sync_detailed(
+    client=client,
+    authorization=authorization,
+    auth_token=auth_token,
+  ).parsed
+
+
+async def asyncio_detailed(
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Response[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  """Get all shared repository limits
+
+   Get rate limit status for all shared repositories the user has access to.
+
+  Args:
+      authorization (Union[None, Unset, str]):
+      auth_token (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]]
+  """
+
+  kwargs = _get_kwargs(
+    authorization=authorization,
+    auth_token=auth_token,
+  )
+
+  response = await client.get_async_httpx_client().request(**kwargs)
+
+  return _build_response(client=client, response=response)
+
+
+async def asyncio(
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Optional[
+  Union[
+    GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
+    HTTPValidationError,
+  ]
+]:
+  """Get all shared repository limits
+
+   Get rate limit status for all shared repositories the user has access to.
+
+  Args:
+      authorization (Union[None, Unset, str]):
+      auth_token (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]
+  """
+
+  return (
+    await asyncio_detailed(
+      client=client,
+      authorization=authorization,
+      auth_token=auth_token,
+    )
+  ).parsed

robosystems_client/api/user_limits/get_shared_repository_limits.py

@@ -0,0 +1,248 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_shared_repository_limits_response_getsharedrepositorylimits import (
+  GetSharedRepositoryLimitsResponseGetsharedrepositorylimits,
+)
+from ...models.http_validation_error import HTTPValidationError
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+  repository: str,
+  *,
+  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
+
+  _kwargs: dict[str, Any] = {
+    "method": "get",
+    "url": f"/v1/user/limits/shared-repositories/{repository}",
+    "cookies": cookies,
+  }
+
+  _kwargs["headers"] = headers
+  return _kwargs
+
+
+def _parse_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  if response.status_code == 200:
+    response_200 = GetSharedRepositoryLimitsResponseGetsharedrepositorylimits.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:
+    return None
+
+
+def _build_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  return Response(
+    status_code=HTTPStatus(response.status_code),
+    content=response.content,
+    headers=response.headers,
+    parsed=_parse_response(client=client, response=response),
+  )
+
+
+def sync_detailed(
+  repository: str,
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Response[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  """Get shared repository rate limit status
+
+   Get current rate limit status and usage for a shared repository.
+
+      Returns:
+      - Current usage across different time windows
+      - Rate limits based on subscription tier
+      - Remaining quota
+      - Reset times
+
+      Note: All queries are FREE - this only shows rate limit status.
+
+  Args:
+      repository (str): Repository name (e.g., 'sec')
+      authorization (Union[None, Unset, str]):
+      auth_token (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[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]]
+  """
+
+  kwargs = _get_kwargs(
+    repository=repository,
+    authorization=authorization,
+    auth_token=auth_token,
+  )
+
+  response = client.get_httpx_client().request(
+    **kwargs,
+  )
+
+  return _build_response(client=client, response=response)
+
+
+def sync(
+  repository: str,
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Optional[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  """Get shared repository rate limit status
+
+   Get current rate limit status and usage for a shared repository.
+
+      Returns:
+      - Current usage across different time windows
+      - Rate limits based on subscription tier
+      - Remaining quota
+      - Reset times
+
+      Note: All queries are FREE - this only shows rate limit status.
+
+  Args:
+      repository (str): Repository name (e.g., 'sec')
+      authorization (Union[None, Unset, str]):
+      auth_token (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[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+  """
+
+  return sync_detailed(
+    repository=repository,
+    client=client,
+    authorization=authorization,
+    auth_token=auth_token,
+  ).parsed
+
+
+async def asyncio_detailed(
+  repository: str,
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Response[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  """Get shared repository rate limit status
+
+   Get current rate limit status and usage for a shared repository.
+
+      Returns:
+      - Current usage across different time windows
+      - Rate limits based on subscription tier
+      - Remaining quota
+      - Reset times
+
+      Note: All queries are FREE - this only shows rate limit status.
+
+  Args:
+      repository (str): Repository name (e.g., 'sec')
+      authorization (Union[None, Unset, str]):
+      auth_token (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[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]]
+  """
+
+  kwargs = _get_kwargs(
+    repository=repository,
+    authorization=authorization,
+    auth_token=auth_token,
+  )
+
+  response = await client.get_async_httpx_client().request(**kwargs)
+
+  return _build_response(client=client, response=response)
+
+
+async def asyncio(
+  repository: str,
+  *,
+  client: AuthenticatedClient,
+  authorization: Union[None, Unset, str] = UNSET,
+  auth_token: Union[None, Unset, str] = UNSET,
+) -> Optional[
+  Union[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+]:
+  """Get shared repository rate limit status
+
+   Get current rate limit status and usage for a shared repository.
+
+      Returns:
+      - Current usage across different time windows
+      - Rate limits based on subscription tier
+      - Remaining quota
+      - Reset times
+
+      Note: All queries are FREE - this only shows rate limit status.
+
+  Args:
+      repository (str): Repository name (e.g., 'sec')
+      authorization (Union[None, Unset, str]):
+      auth_token (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[GetSharedRepositoryLimitsResponseGetsharedrepositorylimits, HTTPValidationError]
+  """
+
+  return (
+    await asyncio_detailed(
+      repository=repository,
+      client=client,
+      authorization=authorization,
+      auth_token=auth_token,
+    )
+  ).parsed

robosystems_client/api/user_subscriptions/get_repository_credits.py

@@ -11,7 +11,7 @@ from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
-  repository_type: str,
+  repository: str,
   *,
   authorization: Union[None, Unset, str] = UNSET,
   auth_token: Union[None, Unset, str] = UNSET,
@@ -26,7 +26,7 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/user/subscriptions/shared-repositories/credits/{repository_type}",
+    "url": f"/v1/user/subscriptions/shared-repositories/credits/{repository}",
     "cookies": cookies,
   }
 
@@ -69,7 +69,7 @@ def _build_response(
 
 
 def sync_detailed(
-  repository_type: str,
+  repository: str,
   *,
   client: AuthenticatedClient,
   authorization: Union[None, Unset, str] = UNSET,
@@ -77,10 +77,10 @@ def sync_detailed(
 ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
 
-   Get credit balance for a specific shared repository type
+   Get credit balance for a specific shared repository
 
   Args:
-      repository_type (str):
+      repository (str):
       authorization (Union[None, Unset, str]):
       auth_token (Union[None, Unset, str]):
 
@@ -93,7 +93,7 @@ def sync_detailed(
   """
 
   kwargs = _get_kwargs(
-    repository_type=repository_type,
+    repository=repository,
     authorization=authorization,
     auth_token=auth_token,
   )
@@ -106,7 +106,7 @@ def sync_detailed(
 
 
 def sync(
-  repository_type: str,
+  repository: str,
   *,
   client: AuthenticatedClient,
   authorization: Union[None, Unset, str] = UNSET,
@@ -114,10 +114,10 @@ def sync(
 ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
 
-   Get credit balance for a specific shared repository type
+   Get credit balance for a specific shared repository
 
   Args:
-      repository_type (str):
+      repository (str):
       authorization (Union[None, Unset, str]):
       auth_token (Union[None, Unset, str]):
 
@@ -130,7 +130,7 @@ def sync(
   """
 
   return sync_detailed(
-    repository_type=repository_type,
+    repository=repository,
     client=client,
     authorization=authorization,
     auth_token=auth_token,
@@ -138,7 +138,7 @@ def sync(
 
 
 async def asyncio_detailed(
-  repository_type: str,
+  repository: str,
   *,
   client: AuthenticatedClient,
   authorization: Union[None, Unset, str] = UNSET,
@@ -146,10 +146,10 @@ async def asyncio_detailed(
 ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
 
-   Get credit balance for a specific shared repository type
+   Get credit balance for a specific shared repository
 
   Args:
-      repository_type (str):
+      repository (str):
       authorization (Union[None, Unset, str]):
       auth_token (Union[None, Unset, str]):
 
@@ -162,7 +162,7 @@ async def asyncio_detailed(
   """
 
   kwargs = _get_kwargs(
-    repository_type=repository_type,
+    repository=repository,
     authorization=authorization,
     auth_token=auth_token,
   )
@@ -173,7 +173,7 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-  repository_type: str,
+  repository: str,
   *,
   client: AuthenticatedClient,
   authorization: Union[None, Unset, str] = UNSET,
@@ -181,10 +181,10 @@ async def asyncio(
 ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
 
-   Get credit balance for a specific shared repository type
+   Get credit balance for a specific shared repository
 
   Args:
-      repository_type (str):
+      repository (str):
       authorization (Union[None, Unset, str]):
       auth_token (Union[None, Unset, str]):
 
@@ -198,7 +198,7 @@ async def asyncio(
 
   return (
     await asyncio_detailed(
-      repository_type=repository_type,
+      repository=repository,
       client=client,
       authorization=authorization,
       auth_token=auth_token,

robosystems_client/extensions/README.md

@@ -599,13 +599,9 @@ MIT License - see [LICENSE](LICENSE) file for details.
 
 ## 📞 Support
 
-- **Documentation**: [docs.robosystems.ai](https://docs.robosystems.ai)
-- **API Reference**: [api.robosystems.ai/docs](https://api.robosystems.ai/docs)
-- **Issues**: [GitHub Issues](https://github.com/robosystems/python-sdk/issues)
-- **Email**: support@robosystems.ai
+- **API Reference**: [api.robosystems.ai](https://api.robosystems.ai)
+- **Issues**: [GitHub Issues](https://github.com/RoboFinSystems/robosystems-python-client/issues)
 
 ---
 
 **RoboSystems Python Client Extensions** - Production-ready streaming, monitoring, and query capabilities for financial knowledge graphs.
-
-*Built with ❤️ by the RoboSystems team*

robosystems_client/extensions/__init__.py

@@ -1,4 +1,4 @@
-"""RoboSystems SDK Extensions for Python
+"""RoboSystems Client Extensions for Python
 
 Enhanced clients with SSE support for the RoboSystems API.
 Provides seamless integration with streaming operations, queue management,

robosystems_client/extensions/auth_integration.py

@@ -1,6 +1,6 @@
-"""Authentication Integration for RoboSystems SDK Extensions
+"""Authentication Integration for RoboSystems Client Extensions
 
-Provides proper integration with the generated SDK authentication system.
+Provides proper integration with the generated Client authentication system.
 """
 
 from typing import Dict, Any

robosystems_client/extensions/extensions.py

@@ -1,4 +1,4 @@
-"""RoboSystems SDK Extensions - Main entry point
+"""RoboSystems Client Extensions - Main entry point
 
 Enhanced clients with SSE support for the RoboSystems API.
 """

robosystems_client/extensions/tests/__init__.py

@@ -1 +1 @@
-"""Tests for RoboSystems SDK Extensions"""
+"""Tests for RoboSystems Client Extensions"""

robosystems_client/extensions/tests/test_integration.py

@@ -1,7 +1,7 @@
-"""Integration Tests for RoboSystems SDK Extensions
+"""Integration Tests for RoboSystems Client Extensions
 
 These tests demonstrate real usage patterns and verify the extensions work correctly
-with the generated SDK.
+with the generated Client.
 """
 
 import pytest
@@ -438,7 +438,7 @@ class TestUtilityIntegration:
 # Integration test runner
 def run_integration_tests():
   """Run all integration tests manually (for environments without pytest)"""
-  print("Running RoboSystems SDK Extensions Integration Tests...")
+  print("Running RoboSystems Client Extensions Integration Tests...")
 
   # Test classes to run
   test_classes = [

robosystems_client/extensions/tests/test_unit.py

@@ -1,4 +1,4 @@
-"""Unit Tests for RoboSystems SDK Extensions
+"""Unit Tests for RoboSystems Client Extensions
 
 Focused unit tests for individual components.
 """
@@ -14,8 +14,10 @@ import os
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../../.."))
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
-from sse_client import SSEClient, SSEConfig, SSEEvent
-from utils import (
+from robosystems_client.extensions import (
+  SSEClient,
+  SSEConfig,
+  SSEEvent,
   QueryBuilder,
   CacheManager,
   ProgressTracker,

robosystems_client/extensions/utils.py

@@ -1,4 +1,4 @@
-"""Utility functions for RoboSystems SDK Extensions
+"""Utility functions for RoboSystems Client Extensions
 
 Common helper functions for working with queries, operations, and data processing.
 """