phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix - traceloop logging

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-01 19:33:41 -07:00
parent 373a41ca6d
commit 85daa467c5
1 changed files with 3 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/integrations/traceloop.py
View file

@ -9,6 +9,7 @@ class TraceloopLogger:
from traceloop.sdk.tracing.tracing import TracerWrapper from traceloop.sdk.tracing.tracing import TracerWrapper
from traceloop.sdk import Traceloop from traceloop.sdk import Traceloop
from traceloop.sdk.instruments import Instruments from traceloop.sdk.instruments import Instruments
from opentelemetry.sdk.trace.export import ConsoleSpanExporter
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
verbose_logger.error( verbose_logger.error(
f"Traceloop not installed, try running 'pip install traceloop-sdk' to fix this error: {e}\n{traceback.format_exc()}" f"Traceloop not installed, try running 'pip install traceloop-sdk' to fix this error: {e}\n{traceback.format_exc()}"
@ -17,13 +18,6 @@ class TraceloopLogger:
Traceloop.init( Traceloop.init(
app_name="Litellm-Server", app_name="Litellm-Server",
disable_batch=True, disable_batch=True,
instruments=[
Instruments.CHROMA,
Instruments.PINECONE,
Instruments.WEAVIATE,
Instruments.LLAMA_INDEX,
Instruments.LANGCHAIN,
],
) )
self.tracer_wrapper = TracerWrapper() self.tracer_wrapper = TracerWrapper()
@ -50,6 +44,8 @@ class TraceloopLogger:
tracer = self.tracer_wrapper.get_tracer() tracer = self.tracer_wrapper.get_tracer()
optional_params = kwargs.get("optional_params", {}) optional_params = kwargs.get("optional_params", {})
start_time = int(start_time.timestamp())
end_time = int(end_time.timestamp())
span = tracer.start_span( span = tracer.start_span(
"litellm.completion", kind=SpanKind.CLIENT, start_time=start_time "litellm.completion", kind=SpanKind.CLIENT, start_time=start_time
) )