22 lines
653 B
Python
22 lines
653 B
Python
import uuid
|
|
from typing import Optional
|
|
|
|
from letta.orm.sandbox_config import AgentEnvironmentVariable
|
|
from letta.serialize_schemas.marshmallow_base import BaseSchema
|
|
|
|
|
|
class SerializedAgentEnvironmentVariableSchema(BaseSchema):
|
|
"""
|
|
Marshmallow schema for serializing/deserializing AgentEnvironmentVariable objects.
|
|
"""
|
|
|
|
__pydantic_model__ = None
|
|
|
|
def generate_id(self) -> Optional[str]:
|
|
# TODO: This is brittle and duplicated in orm/sandbox_config.py
|
|
return f"agent-env-{uuid.uuid4()}"
|
|
|
|
class Meta(BaseSchema.Meta):
|
|
model = AgentEnvironmentVariable
|
|
exclude = BaseSchema.Meta.exclude + ("agent",)
|