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

fix open_telemetry_logger

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-06 22:30:49 -07:00
parent 37e7a7b2d5
commit 78a6604295
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -400,7 +400,7 @@ disable_spend_logs = False
jwt_handler = JWTHandler() jwt_handler = JWTHandler()
prompt_injection_detection_obj: Optional[_OPTIONAL_PromptInjectionDetection] = None prompt_injection_detection_obj: Optional[_OPTIONAL_PromptInjectionDetection] = None
store_model_in_db: bool = False store_model_in_db: bool = False
open_telemetry_logger: Optional[litellm.integrations.opentelemetry.OpenTelemetry] = None open_telemetry_logger = None
### INITIALIZE GLOBAL LOGGING OBJECT ### ### INITIALIZE GLOBAL LOGGING OBJECT ###
proxy_logging_obj = ProxyLogging(user_api_key_cache=user_api_key_cache) proxy_logging_obj = ProxyLogging(user_api_key_cache=user_api_key_cache)
### REDIS QUEUE ### ### REDIS QUEUE ###
@ -845,7 +845,10 @@ async def user_api_key_auth(
verbose_proxy_logger.debug("api key: %s", api_key) verbose_proxy_logger.debug("api key: %s", api_key)
if prisma_client is not None: if prisma_client is not None:
_valid_token: Optional[BaseModel] = await prisma_client.get_data( _valid_token: Optional[BaseModel] = await prisma_client.get_data(
token=api_key, table_name="combined_view" token=api_key,
table_name="combined_view",
parent_otel_span=parent_otel_span,
proxy_logging_obj=proxy_logging_obj,
) )
if _valid_token is not None: if _valid_token is not None:
valid_token = UserAPIKeyAuth( valid_token = UserAPIKeyAuth(