Add optional JWT token parameter to all API methods

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