robosystems-client 0.2.12__tar.gz → 0.2.13__tar.gz

{robosystems_client-0.2.12 → robosystems_client-0.2.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.12
+Version: 0.2.13
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT

{robosystems_client-0.2.12 → robosystems_client-0.2.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.12"
+version = "0.2.13"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"

{robosystems_client-0.2.12 → robosystems_client-0.2.13}/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-0.2.12 → robosystems_client-0.2.13}/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-0.2.12 → robosystems_client-0.2.13}/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-0.2.12 → robosystems_client-0.2.13}/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-0.2.12 → robosystems_client-0.2.13}/robosystems_client/api/credits_/list_credit_transactions.py

@@ -141,7 +141,7 @@ def sync_detailed(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -206,7 +206,7 @@ def sync(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -266,7 +266,7 @@ async def asyncio_detailed(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -329,7 +329,7 @@ async def asyncio(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,

{robosystems_client-0.2.12 → robosystems_client-0.2.13}/robosystems_client/extensions/__init__.py

@@ -14,6 +14,14 @@ from .query_client import (
   QueryOptions,
   QueuedQueryError,
 )
+from .agent_client import (
+  AgentClient,
+  AgentResult,
+  QueuedAgentResponse,
+  AgentQueryRequest,
+  AgentOptions,
+  QueuedAgentError,
+)
 from .operation_client import (
   OperationClient,
   OperationStatus,
@@ -118,6 +126,13 @@ __all__ = [
   "QueryRequest",
   "QueryOptions",
   "QueuedQueryError",
+  # Agent Client
+  "AgentClient",
+  "AgentResult",
+  "QueuedAgentResponse",
+  "AgentQueryRequest",
+  "AgentOptions",
+  "QueuedAgentError",
   # Operation Client
   "OperationClient",
   "OperationStatus",
@@ -189,6 +204,16 @@ def stream_query(graph_id: str, query: str, parameters=None, chunk_size=None):
   return extensions.query.stream_query(graph_id, query, parameters, chunk_size)
 
 
+def agent_query(graph_id: str, message: str, context=None):
+  """Execute an agent query using the default extensions instance"""
+  return extensions.agent.query(graph_id, message, context)
+
+
+def analyze_financials(graph_id: str, message: str, on_progress=None):
+  """Execute financial agent using the default extensions instance"""
+  return extensions.agent.analyze_financials(graph_id, message, on_progress)
+
+
 # DataFrame convenience functions (if pandas is available)
 if (
   HAS_PANDAS