scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 206

test(integration): use rc 108 database read helper

Browse source
This commit is contained in:
Yuneng Jiang 2026-11-03 14:27:15 -08:00
parent b5049a43ce
commit c86300491f
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/observability/test_presidio_masking_endpoints.py
View file

@ -11,6 +11,7 @@ from typing import Final
import pytest
import yaml
from integration._support.client import Gateway, eventually, gateway_from_environment, list_value, object_value
from integration._support.database import read_rows
from integration._support.process import owned_proxy_process
from integration._support.wire import Reply, Request, Wire, wire_server
from pydantic import BaseModel, ConfigDict, JsonValue, TypeAdapter
@ -479,7 +480,7 @@ def _ask(rig: Rig, model: str, surface: Surface, guardrail: str, prompt: str) ->
def _spend_row(rig: Rig, model: str) -> dict[str, JsonValue]:
rows: Final = eventually(
lambda: rig.gateway.rows('SELECT response, metadata FROM "LiteLLM_SpendLogs" WHERE model_group = %s', (model,)),
lambda: read_rows('SELECT response, metadata FROM "LiteLLM_SpendLogs" WHERE model_group = %s', (model,)),
lambda found: len(found) == 1,
seconds=70,
)