fix test

Signed-off-by: kerthcet <kerthcet@gmail.com>

Author: kerthcet

tests/integration/server/test_graphql_query.py

@@ -35,7 +35,7 @@ def test_query_single_team(execute_graphql, test_org_id, test_user_id, test_team
             updatedAt
             totalExperiments
             totalRuns
-            aggregatedTokens {{
+            aggregatedUsage {{
                 totalTokens
                 inputTokens
                 outputTokens
@@ -352,7 +352,7 @@ async def test_query_single_run(
             meta
             status
             createdAt
-            aggregatedTokens {{
+            aggregatedUsage {{
                 totalTokens
                 inputTokens
                 outputTokens
@@ -378,9 +378,9 @@ async def test_query_single_run(
     assert response.data["run"]["status"] == "COMPLETED"
     assert len(response.data["run"]["spans"]) > 0
     # Verify tokens are fetched from ClickHouse via GraphQL fields
-    assert response.data["run"]["aggregatedTokens"]["totalTokens"] is not None
-    assert response.data["run"]["aggregatedTokens"]["inputTokens"] is not None
-    assert response.data["run"]["aggregatedTokens"]["outputTokens"] is not None
+    assert response.data["run"]["aggregatedUsage"]["totalTokens"] is not None
+    assert response.data["run"]["aggregatedUsage"]["inputTokens"] is not None
+    assert response.data["run"]["aggregatedUsage"]["outputTokens"] is not None
 
     # Verify tokens are NOT cached in meta anymore
     metadb = runtime.storage_runtime().metadb