phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix import locs

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-23 17:07:21 -07:00
parent 6c0e1bd38b
commit e6254b84ec
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_pass_through_endpoints.py
View file

@ -14,7 +14,7 @@ import asyncio
from unittest.mock import Mock from unittest.mock import Mock
import httpx import httpx
import litellm_proxy
from litellm_proxy.proxy_server import initialize_pass_through_endpoints from litellm_proxy.proxy_server import initialize_pass_through_endpoints

2
tests/logging_callback_tests/test_opentelemetry_unit_tests.py
View file

@ -26,7 +26,7 @@ class TestOpentelemetryUnitTests(BaseLoggingCallbackTest):
def test_parallel_tool_calls(self, mock_response_obj: ModelResponse): def test_parallel_tool_calls(self, mock_response_obj: ModelResponse):
tool_calls = mock_response_obj.choices[0].message.tool_calls tool_calls = mock_response_obj.choices[0].message.tool_calls
from litellm.integrations.opentelemetry import OpenTelemetry from litellm.integrations.opentelemetry import OpenTelemetry
from litellm_proxy._types import SpanAttributes from litellm.types.integrations.opentelemetry import SpanAttributes
kv_pair_dict = OpenTelemetry._tool_calls_kv_pair(tool_calls) kv_pair_dict = OpenTelemetry._tool_calls_kv_pair(tool_calls)

2
tests/logging_callback_tests/test_otel_logging.py
View file

@ -15,7 +15,7 @@ import asyncio
import logging import logging
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
from litellm._logging import verbose_logger from litellm._logging import verbose_logger
from litellm_proxy._types import SpanAttributes from litellm.types.integrations.opentelemetry import SpanAttributes
verbose_logger.setLevel(logging.DEBUG) verbose_logger.setLevel(logging.DEBUG)