phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

use inmemory exporter for testing

Browse source
This commit is contained in:
yujonglee 2024-06-04 09:04:19 +09:00
parent 61c8e9d2ec
commit 0d8a7d5cf0
2 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_proxy_server.py
View file

@ -604,11 +604,12 @@ def test_load_router_config(mock_cache, fake_env_vars):
# test_load_router_config()
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
@mock_patch_acompletion()
def test_otel_with_proxy_server(mock_acompletion, client_no_auth, capsys):
litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter="console"))]
def test_otel_with_proxy_server(mock_acompletion, client_no_auth):
exporter = InMemorySpanExporter()
litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))]
data = {"model": "gpt-3.5-turbo", "messages": [{"role": "user", "content": "hi"}]}
@ -626,4 +627,5 @@ def test_otel_with_proxy_server(mock_acompletion, client_no_auth, capsys):
assert response.status_code == 200
assert response.json() == example_completion_result
print(capsys.readouterr())
spans = exporter.get_finished_spans()
assert len(spans) == 0