🐛 (types.py): add missing artifact_type to ChatOutputResponse to ensure correct artifact type is set

This commit is contained in:
cristhianzl 2024-06-07 12:16:31 -03:00
commit 2757ae4069

View file

@ -111,7 +111,7 @@ class InterfaceVertex(Vertex):
message = self._built_object
artifact_type = ArtifactType.STREAM if stream_url is not None else ArtifactType.OBJECT
artifacts = ChatOutputResponse(
message=message, sender=sender, sender_name=sender_name, stream_url=stream_url, files=files
message=message, sender=sender, sender_name=sender_name, stream_url=stream_url, files=files, type=artifact_type,
)
self.will_stream = stream_url is not None