fix more typing stuff

wandb · Dec 16, 2024 · 87d0fcf · 87d0fcf
1 parent f6f56bc
commit 87d0fcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/weave/integrations/langchain/langchain_nvidia_ai_endpoints.py b/weave/integrations/langchain/langchain_nvidia_ai_endpoints.py
@@ -113,7 +113,7 @@ def post_process_to_openai_format(output: Any) -> dict:
 ## Need a separate stream variant as the passed objects don't provide an indication
 def process_inputs_to_openai_format_stream(
     func: Op, args: tuple, kwargs: dict
-) -> ProcessedInputs | None:
+) -> ProcessedInputs:
     from langchain_core.messages import convert_to_openai_messages
 
     original_args = args
@@ -146,7 +146,7 @@ def process_inputs_to_openai_format_stream(
 
 def process_inputs_to_openai_format(
     func: Op, args: tuple, kwargs: dict
-) -> ProcessedInputs | None:
+) -> ProcessedInputs:
     from langchain_core.messages import convert_to_openai_messages
 
     original_args = args