use optional

wandb · Dec 14, 2024 · c09b1f7 · c09b1f7
1 parent fc5e18c
commit c09b1f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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":