From 737f8aa61b3541f274f37e712460dee651dcceff Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 7 Jun 2024 08:25:26 -0700
Subject: [PATCH] feat - add parent_otel_span to UserAPIKeyAuth

---
 litellm/proxy/_types.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 8a95f4e1d8..e089c0429d 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -6,6 +6,8 @@ from datetime import datetime
 import uuid, json, sys, os
 from litellm.types.router import UpdateRouterConfig
 from litellm.types.utils import ProviderField
+from typing_extensions import Annotated
+from opentelemetry.trace import Span
 
 
 class LitellmUserRoles(str, enum.Enum):
@@ -1194,6 +1196,7 @@ class UserAPIKeyAuth(
         ]
     ] = None
     allowed_model_region: Optional[Literal["eu"]] = None
+    parent_otel_span: Optional[Span] = None
 
     @model_validator(mode="before")
     @classmethod
@@ -1206,6 +1209,9 @@ class UserAPIKeyAuth(
                 values.update({"api_key": hash_token(values.get("api_key"))})
         return values
 
+    class Config:
+        arbitrary_types_allowed = True
+
 
 class LiteLLM_Config(LiteLLMBase):
     param_name: str