fix

Author: andrewtruong

weave/trace_server/reference/generate.py

@@ -122,7 +122,7 @@ def op_create_v2(
         entity: str,
         project: str,
         body: tsi.OpCreateV2Body,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.OpCreateV2Res:
         """Create an op object."""
         project_id = f"{entity}/{project}"
@@ -138,7 +138,7 @@ def op_read_v2(
         project: str,
         object_id: str,
         digest: str,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.OpReadV2Res:
         """Get an op object."""
         project_id = f"{entity}/{project}"
@@ -168,7 +168,7 @@ def op_list_v2(
         project: str,
         limit: int | None = None,
         offset: int | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> StreamingResponse:
         """List op objects."""
         project_id = f"{entity}/{project}"
@@ -187,7 +187,7 @@ def op_delete_v2(
         project: str,
         object_id: str,
         digests: list[str] | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.OpDeleteV2Res:
         """Delete an op object. If digests are provided, only those versions are deleted. Otherwise, all versions are deleted."""
         project_id = f"{entity}/{project}"
@@ -204,7 +204,7 @@ def dataset_create_v2(
         entity: str,
         project: str,
         body: tsi.DatasetCreateV2Body,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.DatasetCreateV2Res:
         """Create a dataset object."""
         project_id = f"{entity}/{project}"
@@ -220,7 +220,7 @@ def dataset_read_v2(
         project: str,
         object_id: str,
         digest: str,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.DatasetReadV2Res:
         """Get a dataset object."""
         project_id = f"{entity}/{project}"
@@ -252,7 +252,7 @@ def dataset_list_v2(
         project: str,
         limit: int | None = None,
         offset: int | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> StreamingResponse:
         """List dataset objects."""
         project_id = f"{entity}/{project}"
@@ -271,7 +271,7 @@ def dataset_delete_v2(
         project: str,
         object_id: str,
         digests: list[str] | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.DatasetDeleteV2Res:
         """Delete a dataset object."""
         project_id = f"{entity}/{project}"
@@ -288,7 +288,7 @@ def scorer_create_v2(
         entity: str,
         project: str,
         body: tsi.ScorerCreateV2Body,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.ScorerCreateV2Res:
         """Create a scorer object."""
         project_id = f"{entity}/{project}"
@@ -304,7 +304,7 @@ def scorer_read_v2(
         project: str,
         object_id: str,
         digest: str,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.ScorerReadV2Res:
         """Get a scorer object."""
         project_id = f"{entity}/{project}"
@@ -336,7 +336,7 @@ def scorer_list_v2(
         project: str,
         limit: int | None = None,
         offset: int | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> StreamingResponse:
         """List scorer objects."""
         project_id = f"{entity}/{project}"
@@ -355,7 +355,7 @@ def scorer_delete_v2(
         project: str,
         object_id: str,
         digests: list[str] | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.ScorerDeleteV2Res:
         """Delete a scorer object."""
         project_id = f"{entity}/{project}"
@@ -370,7 +370,7 @@ def scorer_delete_v2(
     )
     def evaluation_create_v2(
         req: tsi.EvaluationCreateV2Req,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.EvaluationCreateV2Res:
         """Create an evaluation object."""
         return service.trace_server_interface.evaluation_create_v2(req)
@@ -384,7 +384,7 @@ def evaluation_read_v2(
         project: str,
         object_id: str,
         digest: str,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.EvaluationReadV2Res:
         """Get an evaluation object."""
         project_id = f"{entity}/{project}"
@@ -416,7 +416,7 @@ def evaluation_list_v2(
         project: str,
         limit: int | None = None,
         offset: int | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> StreamingResponse:
         """List evaluation objects."""
         project_id = f"{entity}/{project}"
@@ -434,7 +434,7 @@ def evaluation_delete_v2(
         project: str,
         object_id: str,
         digests: list[str] | None = None,
-        service: weave.trace_server.trace_service.TraceService = Depends(get_service),  # noqa: B008
+        service: TraceService = Depends(get_service),  # noqa: B008
     ) -> tsi.EvaluationDeleteV2Res:
         """Delete an evaluation object."""
         project_id = f"{entity}/{project}"