diff --git a/weave/trace/vals.py b/weave/trace/vals.py index 818ac1ea133..4d2c9b58d4d 100644 --- a/weave/trace/vals.py +++ b/weave/trace/vals.py @@ -31,8 +31,8 @@ from weave.trace.table import Table from weave.trace_server.errors import ObjectDeletedError from weave.trace_server.trace_server_interface import ( + ObjDeleteReq, ObjReadReq, - ObjVersionDeleteReq, TableQueryReq, TableRowFilter, TraceServerInterface, @@ -136,8 +136,8 @@ def delete(self) -> None: if not isinstance(self.ref, (ObjectRef, OpRef)): raise TypeError("Deleting non-object refs is not yet supported.") ref: ObjectRef = typing.cast(ObjectRef, self.ref) - self.server.obj_version_delete( - ObjVersionDeleteReq( + self.server.obj_delete( + ObjDeleteReq( project_id=f"{ref.entity}/{ref.project}", object_id=ref.name, digests=[ref.digest],