robosystems-client 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

robosystems_client/api/agent/auto_select_agent.py

@@ -9,19 +9,35 @@ 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 Response
+from ...models.response_mode import ResponseMode
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
   graph_id: str,
   *,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
 
+  params: dict[str, Any] = {}
+
+  json_mode: Union[None, Unset, str]
+  if isinstance(mode, Unset):
+    json_mode = UNSET
+  elif isinstance(mode, ResponseMode):
+    json_mode = mode.value
+  else:
+    json_mode = mode
+  params["mode"] = json_mode
+
+  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()
@@ -40,6 +56,10 @@ def _parse_response(
 
     return response_200
 
+  if response.status_code == 202:
+    response_202 = cast(Any, None)
+    return response_202
+
   if response.status_code == 400:
     response_400 = cast(Any, None)
     return response_400
@@ -84,10 +104,11 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   r"""Auto-select agent for query
 
-   Automatically select the best agent for your query.
+   Automatically select the best agent for your query with intelligent execution strategy.
 
   **Agent Selection Process:**
 
@@ -96,7 +117,8 @@ def sync_detailed(
   2. Enriching context with RAG if enabled
   3. Evaluating all available agents against selection criteria
   4. Selecting the best match based on confidence scores
-  5. Executing the query with the selected agent
+  5. Choosing execution strategy (sync/SSE/async) based on expected time
+  6. Executing the query with the selected agent
 
   **Available Agent Types:**
   - `financial`: Financial analysis, SEC filings, company metrics
@@ -109,6 +131,14 @@ def sync_detailed(
   - `extended`: Comprehensive analysis (~15-60s), deep research
   - `streaming`: Real-time response streaming
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   **Confidence Score Interpretation:**
   - `0.9-1.0`: High confidence, agent is ideal match
   - `0.7-0.9`: Good confidence, agent is suitable
@@ -132,6 +162,8 @@ def sync_detailed(
 
   Args:
       graph_id (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -145,6 +177,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    mode=mode,
   )
 
   response = client.get_httpx_client().request(
@@ -159,10 +192,11 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   r"""Auto-select agent for query
 
-   Automatically select the best agent for your query.
+   Automatically select the best agent for your query with intelligent execution strategy.
 
   **Agent Selection Process:**
 
@@ -171,7 +205,8 @@ def sync(
   2. Enriching context with RAG if enabled
   3. Evaluating all available agents against selection criteria
   4. Selecting the best match based on confidence scores
-  5. Executing the query with the selected agent
+  5. Choosing execution strategy (sync/SSE/async) based on expected time
+  6. Executing the query with the selected agent
 
   **Available Agent Types:**
   - `financial`: Financial analysis, SEC filings, company metrics
@@ -184,6 +219,14 @@ def sync(
   - `extended`: Comprehensive analysis (~15-60s), deep research
   - `streaming`: Real-time response streaming
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   **Confidence Score Interpretation:**
   - `0.9-1.0`: High confidence, agent is ideal match
   - `0.7-0.9`: Good confidence, agent is suitable
@@ -207,6 +250,8 @@ def sync(
 
   Args:
       graph_id (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -221,6 +266,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    mode=mode,
   ).parsed
 
 
@@ -229,10 +275,11 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   r"""Auto-select agent for query
 
-   Automatically select the best agent for your query.
+   Automatically select the best agent for your query with intelligent execution strategy.
 
   **Agent Selection Process:**
 
@@ -241,7 +288,8 @@ async def asyncio_detailed(
   2. Enriching context with RAG if enabled
   3. Evaluating all available agents against selection criteria
   4. Selecting the best match based on confidence scores
-  5. Executing the query with the selected agent
+  5. Choosing execution strategy (sync/SSE/async) based on expected time
+  6. Executing the query with the selected agent
 
   **Available Agent Types:**
   - `financial`: Financial analysis, SEC filings, company metrics
@@ -254,6 +302,14 @@ async def asyncio_detailed(
   - `extended`: Comprehensive analysis (~15-60s), deep research
   - `streaming`: Real-time response streaming
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   **Confidence Score Interpretation:**
   - `0.9-1.0`: High confidence, agent is ideal match
   - `0.7-0.9`: Good confidence, agent is suitable
@@ -277,6 +333,8 @@ async def asyncio_detailed(
 
   Args:
       graph_id (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -290,6 +348,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    mode=mode,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -302,10 +361,11 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   r"""Auto-select agent for query
 
-   Automatically select the best agent for your query.
+   Automatically select the best agent for your query with intelligent execution strategy.
 
   **Agent Selection Process:**
 
@@ -314,7 +374,8 @@ async def asyncio(
   2. Enriching context with RAG if enabled
   3. Evaluating all available agents against selection criteria
   4. Selecting the best match based on confidence scores
-  5. Executing the query with the selected agent
+  5. Choosing execution strategy (sync/SSE/async) based on expected time
+  6. Executing the query with the selected agent
 
   **Available Agent Types:**
   - `financial`: Financial analysis, SEC filings, company metrics
@@ -327,6 +388,14 @@ async def asyncio(
   - `extended`: Comprehensive analysis (~15-60s), deep research
   - `streaming`: Real-time response streaming
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   **Confidence Score Interpretation:**
   - `0.9-1.0`: High confidence, agent is ideal match
   - `0.7-0.9`: Good confidence, agent is suitable
@@ -350,6 +419,8 @@ async def asyncio(
 
   Args:
       graph_id (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -365,5 +436,6 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      mode=mode,
     )
   ).parsed

robosystems_client/api/agent/execute_specific_agent.py

@@ -9,7 +9,8 @@ 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 Response
+from ...models.response_mode import ResponseMode
+from ...types import UNSET, Response, Unset
 
 
 def _get_kwargs(
@@ -17,12 +18,27 @@ def _get_kwargs(
   agent_type: str,
   *,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
 
+  params: dict[str, Any] = {}
+
+  json_mode: Union[None, Unset, str]
+  if isinstance(mode, Unset):
+    json_mode = UNSET
+  elif isinstance(mode, ResponseMode):
+    json_mode = mode.value
+  else:
+    json_mode = mode
+  params["mode"] = json_mode
+
+  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()
@@ -41,6 +57,10 @@ def _parse_response(
 
     return response_200
 
+  if response.status_code == 202:
+    response_202 = cast(Any, None)
+    return response_202
+
   if response.status_code == 400:
     response_400 = cast(Any, None)
     return response_400
@@ -90,21 +110,32 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   """Execute specific agent
 
-   Execute a specific agent type directly.
+   Execute a specific agent type directly with intelligent execution strategy.
 
   Available agents:
   - **financial**: Financial analysis, SEC filings, accounting data
   - **research**: Deep research and comprehensive analysis
   - **rag**: Fast retrieval without AI (no credits required)
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   Use this endpoint when you know which agent you want to use.
 
   Args:
       graph_id (str):
       agent_type (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -119,6 +150,7 @@ def sync_detailed(
     graph_id=graph_id,
     agent_type=agent_type,
     body=body,
+    mode=mode,
   )
 
   response = client.get_httpx_client().request(
@@ -134,21 +166,32 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   """Execute specific agent
 
-   Execute a specific agent type directly.
+   Execute a specific agent type directly with intelligent execution strategy.
 
   Available agents:
   - **financial**: Financial analysis, SEC filings, accounting data
   - **research**: Deep research and comprehensive analysis
   - **rag**: Fast retrieval without AI (no credits required)
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   Use this endpoint when you know which agent you want to use.
 
   Args:
       graph_id (str):
       agent_type (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -164,6 +207,7 @@ def sync(
     agent_type=agent_type,
     client=client,
     body=body,
+    mode=mode,
   ).parsed
 
 
@@ -173,21 +217,32 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   """Execute specific agent
 
-   Execute a specific agent type directly.
+   Execute a specific agent type directly with intelligent execution strategy.
 
   Available agents:
   - **financial**: Financial analysis, SEC filings, accounting data
   - **research**: Deep research and comprehensive analysis
   - **rag**: Fast retrieval without AI (no credits required)
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   Use this endpoint when you know which agent you want to use.
 
   Args:
       graph_id (str):
       agent_type (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -202,6 +257,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     agent_type=agent_type,
     body=body,
+    mode=mode,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -215,21 +271,32 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: AgentRequest,
+  mode: Union[None, ResponseMode, Unset] = UNSET,
 ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
   """Execute specific agent
 
-   Execute a specific agent type directly.
+   Execute a specific agent type directly with intelligent execution strategy.
 
   Available agents:
   - **financial**: Financial analysis, SEC filings, accounting data
   - **research**: Deep research and comprehensive analysis
   - **rag**: Fast retrieval without AI (no credits required)
 
+  **Execution Strategies (automatic):**
+  - Fast operations (<5s): Immediate synchronous response
+  - Medium operations (5-30s): SSE streaming with progress updates
+  - Long operations (>30s): Async Celery worker with operation tracking
+
+  **Response Mode Override:**
+  Use query parameter `?mode=sync|async` to override automatic strategy selection.
+
   Use this endpoint when you know which agent you want to use.
 
   Args:
       graph_id (str):
       agent_type (str):
+      mode (Union[None, ResponseMode, Unset]): Override execution mode: sync, async, stream, or
+          auto
       body (AgentRequest): Request model for agent interactions.
 
   Raises:
@@ -246,5 +313,6 @@ async def asyncio(
       agent_type=agent_type,
       client=client,
       body=body,
+      mode=mode,
     )
   ).parsed

robosystems_client/api/agent/get_agent_metadata.py

@@ -16,7 +16,7 @@ def _get_kwargs(
 ) -> dict[str, Any]:
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/graphs/{graph_id}/agent/{agent_type}/metadata",
+    "url": f"/v1/graphs/{graph_id}/agent/{agent_type}",
   }
 
   return _kwargs

robosystems_client/api/agent/list_agents.py

@@ -28,7 +28,7 @@ def _get_kwargs(
 
   _kwargs: dict[str, Any] = {
     "method": "get",
-    "url": f"/v1/graphs/{graph_id}/agent/list",
+    "url": f"/v1/graphs/{graph_id}/agent",
     "params": params,
   }
 

robosystems_client/api/auth/register_user.py

@@ -83,8 +83,15 @@ def sync_detailed(
 ) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
   """Register New User
 
-   Register a new user account with email and password. Security controls vary by environment: CAPTCHA
-  and email verification are disabled in development for API testing, but required in production.
+   Register a new user account with email and password.
+
+  **Organization Creation**: RoboSystems is an org-centric platform. When you register, a personal
+  organization is automatically created for you. All resources (graphs, subscriptions, billing) belong
+  to organizations, not individual users. You can later upgrade your personal org to a team or
+  enterprise organization.
+
+  **Security Controls**: CAPTCHA and email verification are disabled in development for API testing,
+  but required in production.
 
   Args:
       body (RegisterRequest): Registration request model.
@@ -115,8 +122,15 @@ def sync(
 ) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
   """Register New User
 
-   Register a new user account with email and password. Security controls vary by environment: CAPTCHA
-  and email verification are disabled in development for API testing, but required in production.
+   Register a new user account with email and password.
+
+  **Organization Creation**: RoboSystems is an org-centric platform. When you register, a personal
+  organization is automatically created for you. All resources (graphs, subscriptions, billing) belong
+  to organizations, not individual users. You can later upgrade your personal org to a team or
+  enterprise organization.
+
+  **Security Controls**: CAPTCHA and email verification are disabled in development for API testing,
+  but required in production.
 
   Args:
       body (RegisterRequest): Registration request model.
@@ -142,8 +156,15 @@ async def asyncio_detailed(
 ) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
   """Register New User
 
-   Register a new user account with email and password. Security controls vary by environment: CAPTCHA
-  and email verification are disabled in development for API testing, but required in production.
+   Register a new user account with email and password.
+
+  **Organization Creation**: RoboSystems is an org-centric platform. When you register, a personal
+  organization is automatically created for you. All resources (graphs, subscriptions, billing) belong
+  to organizations, not individual users. You can later upgrade your personal org to a team or
+  enterprise organization.
+
+  **Security Controls**: CAPTCHA and email verification are disabled in development for API testing,
+  but required in production.
 
   Args:
       body (RegisterRequest): Registration request model.
@@ -172,8 +193,15 @@ async def asyncio(
 ) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
   """Register New User
 
-   Register a new user account with email and password. Security controls vary by environment: CAPTCHA
-  and email verification are disabled in development for API testing, but required in production.
+   Register a new user account with email and password.
+
+  **Organization Creation**: RoboSystems is an org-centric platform. When you register, a personal
+  organization is automatically created for you. All resources (graphs, subscriptions, billing) belong
+  to organizations, not individual users. You can later upgrade your personal org to a team or
+  enterprise organization.
+
+  **Security Controls**: CAPTCHA and email verification are disabled in development for API testing,
+  but required in production.
 
   Args:
       body (RegisterRequest): Registration request model.