feat: remove identities from .af (#1388)

This commit is contained in:
Sarah Wooders
2025-03-24 21:57:52 -07:00
committed by GitHub
parent d3dcd043d3
commit bd4a23a1f9
5 changed files with 10 additions and 9 deletions

View File

@@ -105,4 +105,7 @@ class MarshmallowAgentSchema(BaseSchema):
"sources",
"source_passages",
"agent_passages",
"identities",
"is_deleted",
"groups",
)

View File

@@ -12,4 +12,4 @@ class SerializedBlockSchema(BaseSchema):
class Meta(BaseSchema.Meta):
model = Block
exclude = BaseSchema.Meta.exclude + ("agents",)
exclude = BaseSchema.Meta.exclude + ("agents", "identities", "is_deleted")

View File

@@ -12,4 +12,4 @@ class SerializedToolSchema(BaseSchema):
class Meta(BaseSchema.Meta):
model = Tool
exclude = BaseSchema.Meta.exclude
exclude = BaseSchema.Meta.exclude + ("is_deleted",)

View File

@@ -10,8 +10,6 @@ from letta.schemas.llm_config import LLMConfig
class CoreMemoryBlockSchema(BaseModel):
created_at: str
description: Optional[str]
identities: List[Any]
is_deleted: bool
is_template: bool
label: str
limit: int
@@ -42,7 +40,6 @@ class TagSchema(BaseModel):
class ToolEnvVarSchema(BaseModel):
created_at: str
description: Optional[str]
is_deleted: bool
key: str
updated_at: str
value: str
@@ -76,7 +73,6 @@ class ToolSchema(BaseModel):
args_json_schema: Optional[Any]
created_at: str
description: str
is_deleted: bool
json_schema: ToolJSONSchema
name: str
return_char_limit: int
@@ -94,9 +90,6 @@ class AgentSchema(BaseModel):
created_at: str
description: Optional[str]
embedding_config: EmbeddingConfig
groups: List[Any]
identities: List[Any]
is_deleted: bool
llm_config: LLMConfig
message_buffer_autoclear: bool
messages: List[MessageSchema]

View File

@@ -351,6 +351,11 @@ def test_append_copy_suffix_simple(local_client, server, serialize_test_agent, d
"""Test deserializing JSON into an Agent instance."""
result = server.agent_manager.serialize(agent_id=serialize_test_agent.id, actor=default_user)
# write file
with open("test_agent_serialization.json", "w") as f:
# write json
f.write(json.dumps(result.model_dump(), indent=4))
# Deserialize the agent
agent_copy = server.agent_manager.deserialize(serialized_agent=result, actor=other_user, append_copy_suffix=append_copy_suffix)