eaaca141f7
* feat: add ID format validation to identity schemas Add ID format validation to IdentityCreate, IdentityUpsert, and IdentityUpdate schemas using existing validator types from letta.validators. Changes: - agent_ids: Optional[List[str]] → Optional[List[AgentId]] - block_ids: Optional[List[str]] → Optional[List[BlockId]] This ensures malformed IDs are rejected with 422 validation errors instead of causing 500 database errors. 🤖 Generated with [Letta Code](https://letta.com) Co-Authored-By: Letta <noreply@letta.com> * chore: regenerate API spec and SDK --------- Co-authored-by: Letta <noreply@letta.com>
91 lines
4.5 KiB
Python
91 lines
4.5 KiB
Python
from enum import Enum
|
|
from typing import List, Optional, Union
|
|
|
|
from pydantic import Field
|
|
|
|
from letta.schemas.enums import PrimitiveType
|
|
from letta.schemas.letta_base import LettaBase
|
|
from letta.validators import AgentId, BlockId
|
|
|
|
|
|
class IdentityType(str, Enum):
|
|
"""
|
|
Enum to represent the type of the identity.
|
|
"""
|
|
|
|
org = "org"
|
|
user = "user"
|
|
other = "other"
|
|
|
|
|
|
class IdentityPropertyType(str, Enum):
|
|
"""
|
|
Enum to represent the type of the identity property.
|
|
"""
|
|
|
|
string = "string"
|
|
number = "number"
|
|
boolean = "boolean"
|
|
json = "json"
|
|
|
|
|
|
class IdentityBase(LettaBase):
|
|
__id_prefix__ = PrimitiveType.IDENTITY.value
|
|
|
|
|
|
class IdentityProperty(LettaBase):
|
|
"""A property of an identity"""
|
|
|
|
key: str = Field(..., description="The key of the property")
|
|
value: Union[str, int, float, bool, dict] = Field(..., description="The value of the property")
|
|
type: IdentityPropertyType = Field(..., description="The type of the property")
|
|
|
|
|
|
class Identity(IdentityBase):
|
|
id: str = IdentityBase.generate_id_field()
|
|
identifier_key: str = Field(..., description="External, user-generated identifier key of the identity.")
|
|
name: str = Field(..., description="The name of the identity.")
|
|
identity_type: IdentityType = Field(..., description="The type of the identity.")
|
|
project_id: Optional[str] = Field(None, description="The project id of the identity, if applicable.")
|
|
agent_ids: List[str] = Field(..., description="The IDs of the agents associated with the identity.", deprecated=True)
|
|
block_ids: List[str] = Field(..., description="The IDs of the blocks associated with the identity.", deprecated=True)
|
|
organization_id: Optional[str] = Field(None, description="The organization id of the user")
|
|
properties: List[IdentityProperty] = Field(default_factory=list, description="List of properties associated with the identity")
|
|
|
|
|
|
class IdentityCreate(LettaBase):
|
|
identifier_key: str = Field(..., description="External, user-generated identifier key of the identity.")
|
|
name: str = Field(..., description="The name of the identity.")
|
|
identity_type: IdentityType = Field(..., description="The type of the identity.")
|
|
project_id: Optional[str] = Field(None, description="The project id of the identity, if applicable.")
|
|
agent_ids: Optional[List[AgentId]] = Field(None, description="The agent ids that are associated with the identity.", deprecated=True)
|
|
block_ids: Optional[List[BlockId]] = Field(None, description="The IDs of the blocks associated with the identity.", deprecated=True)
|
|
properties: Optional[List[IdentityProperty]] = Field(None, description="List of properties associated with the identity.")
|
|
|
|
|
|
class IdentityUpsert(LettaBase):
|
|
identifier_key: str = Field(..., description="External, user-generated identifier key of the identity.")
|
|
name: str = Field(..., description="The name of the identity.")
|
|
identity_type: IdentityType = Field(..., description="The type of the identity.")
|
|
project_id: Optional[str] = Field(None, description="The project id of the identity, if applicable.")
|
|
agent_ids: Optional[List[AgentId]] = Field(None, description="The agent ids that are associated with the identity.", deprecated=True)
|
|
block_ids: Optional[List[BlockId]] = Field(None, description="The IDs of the blocks associated with the identity.", deprecated=True)
|
|
properties: Optional[List[IdentityProperty]] = Field(None, description="List of properties associated with the identity.")
|
|
|
|
|
|
class IdentityUpdate(LettaBase):
|
|
identifier_key: Optional[str] = Field(None, description="External, user-generated identifier key of the identity.")
|
|
name: Optional[str] = Field(None, description="The name of the identity.")
|
|
identity_type: Optional[IdentityType] = Field(None, description="The type of the identity.")
|
|
agent_ids: Optional[List[AgentId]] = Field(None, description="The agent ids that are associated with the identity.", deprecated=True)
|
|
block_ids: Optional[List[BlockId]] = Field(None, description="The IDs of the blocks associated with the identity.", deprecated=True)
|
|
properties: Optional[List[IdentityProperty]] = Field(None, description="List of properties associated with the identity.")
|
|
|
|
|
|
class PaginatedIdentities(LettaBase):
|
|
"""Paginated response for identities"""
|
|
|
|
data: List[Identity] = Field(..., description="List of identities")
|
|
next_cursor: Optional[str] = Field(None, description="Cursor for fetching the next page")
|
|
has_more: bool = Field(..., description="Whether more results exist after this page")
|