robosystems-client 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

robosystems_client/models/__init__.py

@@ -2,11 +2,26 @@
 
 from .account_info import AccountInfo
 from .add_on_credit_info import AddOnCreditInfo
+from .agent_list_response import AgentListResponse
+from .agent_list_response_agents import AgentListResponseAgents
+from .agent_list_response_agents_additional_property import (
+  AgentListResponseAgentsAdditionalProperty,
+)
 from .agent_message import AgentMessage
+from .agent_metadata_response import AgentMetadataResponse
+from .agent_mode import AgentMode
+from .agent_recommendation import AgentRecommendation
+from .agent_recommendation_request import AgentRecommendationRequest
+from .agent_recommendation_request_context_type_0 import (
+  AgentRecommendationRequestContextType0,
+)
+from .agent_recommendation_response import AgentRecommendationResponse
 from .agent_request import AgentRequest
 from .agent_request_context_type_0 import AgentRequestContextType0
 from .agent_response import AgentResponse
+from .agent_response_error_details_type_0 import AgentResponseErrorDetailsType0
 from .agent_response_metadata_type_0 import AgentResponseMetadataType0
+from .agent_response_tokens_used_type_0 import AgentResponseTokensUsedType0
 from .api_key_info import APIKeyInfo
 from .api_keys_response import APIKeysResponse
 from .auth_response import AuthResponse
@@ -19,6 +34,8 @@ from .backup_response import BackupResponse
 from .backup_restore_request import BackupRestoreRequest
 from .backup_stats_response import BackupStatsResponse
 from .backup_stats_response_backup_formats import BackupStatsResponseBackupFormats
+from .batch_agent_request import BatchAgentRequest
+from .batch_agent_response import BatchAgentResponse
 from .cancel_operation_response_canceloperation import (
   CancelOperationResponseCanceloperation,
 )
@@ -46,8 +63,8 @@ from .create_subgraph_request_metadata_type_0 import CreateSubgraphRequestMetada
 from .credit_summary import CreditSummary
 from .credit_summary_response import CreditSummaryResponse
 from .credits_summary_response import CreditsSummaryResponse
-from .credits_summary_response_credits_by_addon_item import (
-  CreditsSummaryResponseCreditsByAddonItem,
+from .credits_summary_response_credits_by_addon_type_0_item import (
+  CreditsSummaryResponseCreditsByAddonType0Item,
 )
 from .custom_schema_definition import CustomSchemaDefinition
 from .custom_schema_definition_metadata import CustomSchemaDefinitionMetadata
@@ -187,6 +204,7 @@ from .schema_validation_response_compatibility_type_0 import (
 )
 from .schema_validation_response_stats_type_0 import SchemaValidationResponseStatsType0
 from .sec_connection_config import SECConnectionConfig
+from .selection_criteria import SelectionCriteria
 from .sso_complete_request import SSOCompleteRequest
 from .sso_exchange_request import SSOExchangeRequest
 from .sso_exchange_response import SSOExchangeResponse
@@ -243,11 +261,22 @@ from .validation_error import ValidationError
 __all__ = (
   "AccountInfo",
   "AddOnCreditInfo",
+  "AgentListResponse",
+  "AgentListResponseAgents",
+  "AgentListResponseAgentsAdditionalProperty",
   "AgentMessage",
+  "AgentMetadataResponse",
+  "AgentMode",
+  "AgentRecommendation",
+  "AgentRecommendationRequest",
+  "AgentRecommendationRequestContextType0",
+  "AgentRecommendationResponse",
   "AgentRequest",
   "AgentRequestContextType0",
   "AgentResponse",
+  "AgentResponseErrorDetailsType0",
   "AgentResponseMetadataType0",
+  "AgentResponseTokensUsedType0",
   "APIKeyInfo",
   "APIKeysResponse",
   "AuthResponse",
@@ -260,6 +289,8 @@ __all__ = (
   "BackupRestoreRequest",
   "BackupStatsResponse",
   "BackupStatsResponseBackupFormats",
+  "BatchAgentRequest",
+  "BatchAgentResponse",
   "CancellationResponse",
   "CancelOperationResponseCanceloperation",
   "CheckCreditBalanceResponseCheckcreditbalance",
@@ -281,7 +312,7 @@ __all__ = (
   "CreateSubgraphRequest",
   "CreateSubgraphRequestMetadataType0",
   "CreditsSummaryResponse",
-  "CreditsSummaryResponseCreditsByAddonItem",
+  "CreditsSummaryResponseCreditsByAddonType0Item",
   "CreditSummary",
   "CreditSummaryResponse",
   "CustomSchemaDefinition",
@@ -374,6 +405,7 @@ __all__ = (
   "SchemaValidationResponseCompatibilityType0",
   "SchemaValidationResponseStatsType0",
   "SECConnectionConfig",
+  "SelectionCriteria",
   "SSOCompleteRequest",
   "SSOExchangeRequest",
   "SSOExchangeResponse",

robosystems_client/models/agent_list_response.py

@@ -0,0 +1,74 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+  from ..models.agent_list_response_agents import AgentListResponseAgents
+
+
+T = TypeVar("T", bound="AgentListResponse")
+
+
+@_attrs_define
+class AgentListResponse:
+  """Response for listing available agents.
+
+  Attributes:
+      agents (AgentListResponseAgents): Dictionary of available agents with metadata
+      total (int): Total number of agents
+  """
+
+  agents: "AgentListResponseAgents"
+  total: int
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    agents = self.agents.to_dict()
+
+    total = self.total
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "agents": agents,
+        "total": total,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.agent_list_response_agents import AgentListResponseAgents
+
+    d = dict(src_dict)
+    agents = AgentListResponseAgents.from_dict(d.pop("agents"))
+
+    total = d.pop("total")
+
+    agent_list_response = cls(
+      agents=agents,
+      total=total,
+    )
+
+    agent_list_response.additional_properties = d
+    return agent_list_response
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/agent_list_response_agents.py

@@ -0,0 +1,67 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+  from ..models.agent_list_response_agents_additional_property import (
+    AgentListResponseAgentsAdditionalProperty,
+  )
+
+
+T = TypeVar("T", bound="AgentListResponseAgents")
+
+
+@_attrs_define
+class AgentListResponseAgents:
+  """Dictionary of available agents with metadata"""
+
+  additional_properties: dict[str, "AgentListResponseAgentsAdditionalProperty"] = (
+    _attrs_field(init=False, factory=dict)
+  )
+
+  def to_dict(self) -> dict[str, Any]:
+    field_dict: dict[str, Any] = {}
+    for prop_name, prop in self.additional_properties.items():
+      field_dict[prop_name] = prop.to_dict()
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.agent_list_response_agents_additional_property import (
+      AgentListResponseAgentsAdditionalProperty,
+    )
+
+    d = dict(src_dict)
+    agent_list_response_agents = cls()
+
+    additional_properties = {}
+    for prop_name, prop_dict in d.items():
+      additional_property = AgentListResponseAgentsAdditionalProperty.from_dict(
+        prop_dict
+      )
+
+      additional_properties[prop_name] = additional_property
+
+    agent_list_response_agents.additional_properties = additional_properties
+    return agent_list_response_agents
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> "AgentListResponseAgentsAdditionalProperty":
+    return self.additional_properties[key]
+
+  def __setitem__(
+    self, key: str, value: "AgentListResponseAgentsAdditionalProperty"
+  ) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/{credits_summary_response_credits_by_addon_item.py → agent_list_response_agents_additional_property.py}

@@ -4,11 +4,11 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="CreditsSummaryResponseCreditsByAddonItem")
+T = TypeVar("T", bound="AgentListResponseAgentsAdditionalProperty")
 
 
 @_attrs_define
-class CreditsSummaryResponseCreditsByAddonItem:
+class AgentListResponseAgentsAdditionalProperty:
   """ """
 
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -22,10 +22,10 @@ class CreditsSummaryResponseCreditsByAddonItem:
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    credits_summary_response_credits_by_addon_item = cls()
+    agent_list_response_agents_additional_property = cls()
 
-    credits_summary_response_credits_by_addon_item.additional_properties = d
-    return credits_summary_response_credits_by_addon_item
+    agent_list_response_agents_additional_property.additional_properties = d
+    return agent_list_response_agents_additional_property
 
   @property
   def additional_keys(self) -> list[str]:

robosystems_client/models/agent_message.py

@@ -1,8 +1,12 @@
+import datetime
 from collections.abc import Mapping
-from typing import Any, TypeVar
+from typing import Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..types import UNSET, Unset
 
 T = TypeVar("T", bound="AgentMessage")
 
@@ -14,10 +18,12 @@ class AgentMessage:
   Attributes:
       role (str): Message role (user/assistant)
       content (str): Message content
+      timestamp (Union[None, Unset, datetime.datetime]): Message timestamp
   """
 
   role: str
   content: str
+  timestamp: Union[None, Unset, datetime.datetime] = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -25,6 +31,14 @@ class AgentMessage:
 
     content = self.content
 
+    timestamp: Union[None, Unset, str]
+    if isinstance(self.timestamp, Unset):
+      timestamp = UNSET
+    elif isinstance(self.timestamp, datetime.datetime):
+      timestamp = self.timestamp.isoformat()
+    else:
+      timestamp = self.timestamp
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
     field_dict.update(
@@ -33,6 +47,8 @@ class AgentMessage:
         "content": content,
       }
     )
+    if timestamp is not UNSET:
+      field_dict["timestamp"] = timestamp
 
     return field_dict
 
@@ -43,9 +59,27 @@ class AgentMessage:
 
     content = d.pop("content")
 
+    def _parse_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      try:
+        if not isinstance(data, str):
+          raise TypeError()
+        timestamp_type_0 = isoparse(data)
+
+        return timestamp_type_0
+      except:  # noqa: E722
+        pass
+      return cast(Union[None, Unset, datetime.datetime], data)
+
+    timestamp = _parse_timestamp(d.pop("timestamp", UNSET))
+
     agent_message = cls(
       role=role,
       content=content,
+      timestamp=timestamp,
     )
 
     agent_message.additional_properties = d

robosystems_client/models/agent_metadata_response.py

@@ -0,0 +1,133 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="AgentMetadataResponse")
+
+
+@_attrs_define
+class AgentMetadataResponse:
+  """Response for agent metadata.
+
+  Attributes:
+      name (str): Agent name
+      description (str): Agent description
+      version (str): Agent version
+      capabilities (list[str]): Agent capabilities
+      supported_modes (list[str]): Supported execution modes
+      requires_credits (bool): Whether agent requires credits
+      author (Union[None, Unset, str]): Agent author
+      tags (Union[Unset, list[str]]): Agent tags
+  """
+
+  name: str
+  description: str
+  version: str
+  capabilities: list[str]
+  supported_modes: list[str]
+  requires_credits: bool
+  author: Union[None, Unset, str] = UNSET
+  tags: Union[Unset, list[str]] = UNSET
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    name = self.name
+
+    description = self.description
+
+    version = self.version
+
+    capabilities = self.capabilities
+
+    supported_modes = self.supported_modes
+
+    requires_credits = self.requires_credits
+
+    author: Union[None, Unset, str]
+    if isinstance(self.author, Unset):
+      author = UNSET
+    else:
+      author = self.author
+
+    tags: Union[Unset, list[str]] = UNSET
+    if not isinstance(self.tags, Unset):
+      tags = self.tags
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "name": name,
+        "description": description,
+        "version": version,
+        "capabilities": capabilities,
+        "supported_modes": supported_modes,
+        "requires_credits": requires_credits,
+      }
+    )
+    if author is not UNSET:
+      field_dict["author"] = author
+    if tags is not UNSET:
+      field_dict["tags"] = tags
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    name = d.pop("name")
+
+    description = d.pop("description")
+
+    version = d.pop("version")
+
+    capabilities = cast(list[str], d.pop("capabilities"))
+
+    supported_modes = cast(list[str], d.pop("supported_modes"))
+
+    requires_credits = d.pop("requires_credits")
+
+    def _parse_author(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    author = _parse_author(d.pop("author", UNSET))
+
+    tags = cast(list[str], d.pop("tags", UNSET))
+
+    agent_metadata_response = cls(
+      name=name,
+      description=description,
+      version=version,
+      capabilities=capabilities,
+      supported_modes=supported_modes,
+      requires_credits=requires_credits,
+      author=author,
+      tags=tags,
+    )
+
+    agent_metadata_response.additional_properties = d
+    return agent_metadata_response
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/agent_mode.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class AgentMode(str, Enum):
+  EXTENDED = "extended"
+  QUICK = "quick"
+  STANDARD = "standard"
+  STREAMING = "streaming"
+
+  def __str__(self) -> str:
+    return str(self.value)

robosystems_client/models/agent_recommendation.py

@@ -0,0 +1,106 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="AgentRecommendation")
+
+
+@_attrs_define
+class AgentRecommendation:
+  """Single agent recommendation.
+
+  Attributes:
+      agent_type (str): Agent type identifier
+      agent_name (str): Agent display name
+      confidence (float): Confidence score (0-1)
+      capabilities (list[str]): Agent capabilities
+      reason (Union[None, Unset, str]): Reason for recommendation
+  """
+
+  agent_type: str
+  agent_name: str
+  confidence: float
+  capabilities: list[str]
+  reason: Union[None, Unset, str] = UNSET
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    agent_type = self.agent_type
+
+    agent_name = self.agent_name
+
+    confidence = self.confidence
+
+    capabilities = self.capabilities
+
+    reason: Union[None, Unset, str]
+    if isinstance(self.reason, Unset):
+      reason = UNSET
+    else:
+      reason = self.reason
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "agent_type": agent_type,
+        "agent_name": agent_name,
+        "confidence": confidence,
+        "capabilities": capabilities,
+      }
+    )
+    if reason is not UNSET:
+      field_dict["reason"] = reason
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    agent_type = d.pop("agent_type")
+
+    agent_name = d.pop("agent_name")
+
+    confidence = d.pop("confidence")
+
+    capabilities = cast(list[str], d.pop("capabilities"))
+
+    def _parse_reason(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    reason = _parse_reason(d.pop("reason", UNSET))
+
+    agent_recommendation = cls(
+      agent_type=agent_type,
+      agent_name=agent_name,
+      confidence=confidence,
+      capabilities=capabilities,
+      reason=reason,
+    )
+
+    agent_recommendation.additional_properties = d
+    return agent_recommendation
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties