robosystems-client 0.2.3__tar.gz → 0.2.5__tar.gz

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/.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

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/PKG-INFO

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

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/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)

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/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":

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.3"
+version = "0.2.5"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -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"

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/robosystems_client/api/agent/auto_select_agent.py

@@ -9,35 +9,19 @@ from ...models.agent_request import AgentRequest
 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

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/robosystems_client/api/agent/batch_process_queries.py

@@ -8,35 +8,19 @@ from ...client import AuthenticatedClient, Client
 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

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/robosystems_client/api/agent/execute_specific_agent.py

@@ -9,7 +9,7 @@ from ...models.agent_request import AgentRequest
 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

{robosystems_client-0.2.3 → robosystems_client-0.2.5}/robosystems_client/api/agent/get_agent_metadata.py

@@ -7,38 +7,18 @@ from ... import errors
 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