robosystems-client 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

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/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/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/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