robosystems-client 0.1.17__py3-none-any.whl → 0.1.18__py3-none-any.whl

robosystems_client/__init__.py

@@ -1,14 +1,25 @@
 """RoboSystems Python Client."""
 
-__version__ = "0.1.0"
-
 from .client import AuthenticatedClient, Client
 
+# Convenience alias for the main SDK
+RoboSystemsSDK = AuthenticatedClient
+
 __all__ = (
   "AuthenticatedClient",
   "Client",
   "RoboSystemsSDK",
 )
 
-# Convenience alias for the main SDK
-RoboSystemsSDK = AuthenticatedClient
+
+def _get_version() -> str:
+  """Get version from package metadata."""
+  try:
+    from importlib.metadata import version
+
+    return version("robosystems-client")
+  except Exception:
+    return "0.0.0+development"
+
+
+__version__ = _get_version()

robosystems_client/api/agent/auto_select_agent.py

@@ -16,15 +16,28 @@ 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/{graph_id}/agent",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -81,6 +94,7 @@ 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
@@ -97,6 +111,7 @@ 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.
 
@@ -111,6 +126,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -126,6 +142,7 @@ 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
@@ -142,6 +159,7 @@ 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.
 
@@ -157,6 +175,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -166,6 +185,7 @@ 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
@@ -182,6 +202,7 @@ 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.
 
@@ -196,6 +217,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -209,6 +231,7 @@ 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
@@ -225,6 +248,7 @@ 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.
 
@@ -241,6 +265,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/agent/batch_process_queries.py

@@ -15,15 +15,28 @@ 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/{graph_id}/agent/batch",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -76,6 +89,7 @@ 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
@@ -97,6 +111,7 @@ 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.
 
@@ -111,6 +126,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -126,6 +142,7 @@ 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
@@ -147,6 +164,7 @@ 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.
 
@@ -162,6 +180,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -171,6 +190,7 @@ 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
@@ -192,6 +212,7 @@ 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.
 
@@ -206,6 +227,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -219,6 +241,7 @@ 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
@@ -240,6 +263,7 @@ 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.
 
@@ -256,6 +280,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/agent/execute_specific_agent.py

@@ -17,15 +17,28 @@ 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/{graph_id}/agent/{agent_type}",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -86,6 +99,7 @@ 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
@@ -102,6 +116,7 @@ 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.
 
@@ -117,6 +132,7 @@ def sync_detailed(
     graph_id=graph_id,
     agent_type=agent_type,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -133,6 +149,7 @@ 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
@@ -149,6 +166,7 @@ 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.
 
@@ -165,6 +183,7 @@ def sync(
     agent_type=agent_type,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -175,6 +194,7 @@ 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
@@ -191,6 +211,7 @@ 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.
 
@@ -206,6 +227,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     agent_type=agent_type,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -220,6 +242,7 @@ 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
@@ -236,6 +259,7 @@ 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.
 
@@ -253,6 +277,7 @@ async def asyncio(
       agent_type=agent_type,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/agent/get_agent_metadata.py

@@ -14,15 +14,28 @@ 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/{graph_id}/agent/{agent_type}/metadata",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -65,6 +78,7 @@ 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
@@ -84,6 +98,7 @@ def sync_detailed(
   Args:
       graph_id (str): Graph database identifier
       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:
@@ -97,6 +112,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     agent_type=agent_type,
+    token=token,
     authorization=authorization,
   )
 
@@ -112,6 +128,7 @@ 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
@@ -131,6 +148,7 @@ def sync(
   Args:
       graph_id (str): Graph database identifier
       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:
@@ -145,6 +163,7 @@ def sync(
     graph_id=graph_id,
     agent_type=agent_type,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -154,6 +173,7 @@ 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
@@ -173,6 +193,7 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Graph database identifier
       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:
@@ -186,6 +207,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     agent_type=agent_type,
+    token=token,
     authorization=authorization,
   )
 
@@ -199,6 +221,7 @@ 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
@@ -218,6 +241,7 @@ async def asyncio(
   Args:
       graph_id (str): Graph database identifier
       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:
@@ -233,6 +257,7 @@ async def asyncio(
       graph_id=graph_id,
       agent_type=agent_type,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/agent/list_agents.py

@@ -14,6 +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] = {}
@@ -29,6 +30,13 @@ 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] = {
@@ -77,6 +85,7 @@ 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
@@ -95,6 +104,7 @@ def sync_detailed(
       graph_id (str): Graph database identifier
       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:
@@ -108,6 +118,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     capability=capability,
+    token=token,
     authorization=authorization,
   )
 
@@ -123,6 +134,7 @@ 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
@@ -141,6 +153,7 @@ def sync(
       graph_id (str): Graph database identifier
       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:
@@ -155,6 +168,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     capability=capability,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -164,6 +178,7 @@ 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
@@ -182,6 +197,7 @@ async def asyncio_detailed(
       graph_id (str): Graph database identifier
       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:
@@ -195,6 +211,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     capability=capability,
+    token=token,
     authorization=authorization,
   )
 
@@ -208,6 +225,7 @@ 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
@@ -226,6 +244,7 @@ async def asyncio(
       graph_id (str): Graph database identifier
       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:
@@ -241,6 +260,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       capability=capability,
+      token=token,
       authorization=authorization,
     )
   ).parsed