robosystems-client 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py

@@ -15,20 +15,14 @@ def _get_kwargs(
   *,
   body: TierUpgradeRequest,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
-
   _kwargs: dict[str, Any] = {
     "method": "put",
     "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}/upgrade",
-    "cookies": cookies,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -84,7 +78,6 @@ def sync_detailed(
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -93,7 +86,6 @@ def sync_detailed(
   Args:
       subscription_id (str):
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -108,7 +100,6 @@ def sync_detailed(
     subscription_id=subscription_id,
     body=body,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -124,7 +115,6 @@ def sync(
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -133,7 +123,6 @@ def sync(
   Args:
       subscription_id (str):
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -149,7 +138,6 @@ def sync(
     client=client,
     body=body,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -159,7 +147,6 @@ async def asyncio_detailed(
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -168,7 +155,6 @@ async def asyncio_detailed(
   Args:
       subscription_id (str):
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -183,7 +169,6 @@ async def asyncio_detailed(
     subscription_id=subscription_id,
     body=body,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -197,7 +182,6 @@ async def asyncio(
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -206,7 +190,6 @@ async def asyncio(
   Args:
       subscription_id (str):
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -223,6 +206,5 @@ async def asyncio(
       client=client,
       body=body,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

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
@@ -68,6 +85,7 @@ from .detailed_transactions_response_date_range import (
   DetailedTransactionsResponseDateRange,
 )
 from .detailed_transactions_response_summary import DetailedTransactionsResponseSummary
+from .email_verification_request import EmailVerificationRequest
 from .enhanced_credit_transaction_response import EnhancedCreditTransactionResponse
 from .enhanced_credit_transaction_response_metadata import (
   EnhancedCreditTransactionResponseMetadata,
@@ -75,6 +93,10 @@ from .enhanced_credit_transaction_response_metadata import (
 from .error_response import ErrorResponse
 from .exchange_token_request import ExchangeTokenRequest
 from .exchange_token_request_metadata_type_0 import ExchangeTokenRequestMetadataType0
+from .forgot_password_request import ForgotPasswordRequest
+from .forgot_password_response_forgotpassword import (
+  ForgotPasswordResponseForgotpassword,
+)
 from .get_all_credit_summaries_response_getallcreditsummaries import (
   GetAllCreditSummariesResponseGetallcreditsummaries,
 )
@@ -168,6 +190,11 @@ from .register_request import RegisterRequest
 from .repository_credits_response import RepositoryCreditsResponse
 from .repository_plan import RepositoryPlan
 from .repository_type import RepositoryType
+from .resend_verification_email_response_resendverificationemail import (
+  ResendVerificationEmailResponseResendverificationemail,
+)
+from .reset_password_request import ResetPasswordRequest
+from .reset_password_validate_response import ResetPasswordValidateResponse
 from .response_mode import ResponseMode
 from .s3_copy_request import S3CopyRequest
 from .s3_copy_request_file_format import S3CopyRequestFileFormat
@@ -187,6 +214,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 +271,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 +299,8 @@ __all__ = (
   "BackupRestoreRequest",
   "BackupStatsResponse",
   "BackupStatsResponseBackupFormats",
+  "BatchAgentRequest",
+  "BatchAgentResponse",
   "CancellationResponse",
   "CancelOperationResponseCanceloperation",
   "CheckCreditBalanceResponseCheckcreditbalance",
@@ -281,7 +322,7 @@ __all__ = (
   "CreateSubgraphRequest",
   "CreateSubgraphRequestMetadataType0",
   "CreditsSummaryResponse",
-  "CreditsSummaryResponseCreditsByAddonItem",
+  "CreditsSummaryResponseCreditsByAddonType0Item",
   "CreditSummary",
   "CreditSummaryResponse",
   "CustomSchemaDefinition",
@@ -299,11 +340,14 @@ __all__ = (
   "DetailedTransactionsResponse",
   "DetailedTransactionsResponseDateRange",
   "DetailedTransactionsResponseSummary",
+  "EmailVerificationRequest",
   "EnhancedCreditTransactionResponse",
   "EnhancedCreditTransactionResponseMetadata",
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
+  "ForgotPasswordRequest",
+  "ForgotPasswordResponseForgotpassword",
   "GetAllCreditSummariesResponseGetallcreditsummaries",
   "GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits",
   "GetBackupDownloadUrlResponseGetbackupdownloadurl",
@@ -361,6 +405,9 @@ __all__ = (
   "RepositoryCreditsResponse",
   "RepositoryPlan",
   "RepositoryType",
+  "ResendVerificationEmailResponseResendverificationemail",
+  "ResetPasswordRequest",
+  "ResetPasswordValidateResponse",
   "ResponseMode",
   "S3CopyRequest",
   "S3CopyRequestFileFormat",
@@ -374,6 +421,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)