From c09b1f7365c95392952dcc5131187e33a3f5882f Mon Sep 17 00:00:00 2001
From: jwlee64 <josiah.lee@wandb.com>
Date: Fri, 13 Dec 2024 16:37:08 -0800
Subject: [PATCH] use optional

---
 weave/trace_server/llm_completion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/weave/trace_server/llm_completion.py b/weave/trace_server/llm_completion.py
index 7b1c6006dc6..bec4361c856 100644
--- a/weave/trace_server/llm_completion.py
+++ b/weave/trace_server/llm_completion.py
@@ -13,7 +13,7 @@
 def lite_llm_completion(
     api_key: str,
     inputs: tsi.CompletionsCreateRequestInputs,
-    provider: str | None = None,
+    provider: Optional[str] = None,
 ) -> tsi.CompletionsCreateRes:
     aws_access_key_id, aws_secret_access_key, aws_region_name = None, None, None
     if provider == "bedrock" or provider == "bedrock_converse":