diff --git a/python/langsmith/schemas.py b/python/langsmith/schemas.py index 4e3a63320..3958ac94b 100644 --- a/python/langsmith/schemas.py +++ b/python/langsmith/schemas.py @@ -372,7 +372,7 @@ class RunBase(BaseModel): tags: Optional[List[str]] = None """Tags for categorizing or annotating the run.""" - attachments: Attachments | Dict[str, AttachmentInfo] = Field(default_factory=dict) + attachments: Union[Attachments, Dict[str, AttachmentInfo]] = Field(default_factory=dict) """Attachments associated with the run. Each entry is a tuple of (mime_type, bytes)."""