mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
f3a24d22d5
* Feat: Add Langtrace integration * add langtrace service name * fix timestamps for traces * add tests * Discard Callback + use existing otel logger * cleanup * remove print statments * remove callback * add docs * docs * add logging docs * format logging * remove emoji and add litellm proxy example * format logging * format `logging.md` * add langtrace docs to logging.md * sync conflict
33 lines
805 B
Python
33 lines
805 B
Python
import os
|
|
import sys
|
|
import time
|
|
|
|
import pytest
|
|
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
|
|
from langtrace_python_sdk import langtrace
|
|
|
|
import litellm
|
|
|
|
sys.path.insert(0, os.path.abspath("../.."))
|
|
|
|
|
|
@pytest.fixture()
|
|
def exporter():
|
|
exporter = InMemorySpanExporter()
|
|
langtrace.init(batch=False, custom_remote_exporter=exporter)
|
|
litellm.success_callback = ["langtrace"]
|
|
litellm.set_verbose = True
|
|
|
|
return exporter
|
|
|
|
|
|
@pytest.mark.parametrize("model", ["claude-2.1", "gpt-3.5-turbo"])
|
|
def test_langtrace_logging(exporter, model):
|
|
litellm.completion(
|
|
model=model,
|
|
messages=[{"role": "user", "content": "This is a test"}],
|
|
max_tokens=1000,
|
|
temperature=0.7,
|
|
timeout=5,
|
|
mock_response="hi",
|
|
)
|