modify function name for clarity

Agenta-AI · Dec 15, 2023 · b2d0546 · b2d0546
1 parent ed33322
commit b2d0546
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agenta-backend/agenta_backend/routers/observability_router.py b/agenta-backend/agenta_backend/routers/observability_router.py
@@ -8,7 +8,7 @@
     get_variant_traces,
     create_app_trace,
     create_trace_span,
-    get_single_trace_object,
+    get_trace_single,
     trace_status_update,
     get_trace_spans,
     add_feedback_to_trace,
@@ -74,7 +74,7 @@ async def get_single_trace(
 ):
     # Get user and org id
     kwargs: dict = await get_user_and_org_id(request.state.user_id)
-    trace = await get_single_trace_object(trace_id, **kwargs)
+    trace = await get_trace_single(trace_id, **kwargs)
     return trace
 
 

diff --git a/agenta-backend/agenta_backend/services/event_db_manager.py b/agenta-backend/agenta_backend/services/event_db_manager.py
@@ -85,7 +85,7 @@ async def create_app_trace(payload: CreateTrace, **kwargs: dict) -> str:
     return trace_db_to_pydantic(trace)["trace_id"]
 
 
-async def get_single_trace_object(trace_id: str, **kwargs: dict) -> Trace:
+async def get_trace_single(trace_id: str, **kwargs: dict) -> Trace:
     """Get a single trace.
 
     Args: