From 87d0fcf99099ad423e7cfb1d0f24c092d68cd959 Mon Sep 17 00:00:00 2001
From: Abraham Leal <45460452+abraham-leal@users.noreply.github.com>
Date: Mon, 16 Dec 2024 17:47:10 -0600
Subject: [PATCH] fix more typing stuff

---
 weave/integrations/langchain/langchain_nvidia_ai_endpoints.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/weave/integrations/langchain/langchain_nvidia_ai_endpoints.py b/weave/integrations/langchain/langchain_nvidia_ai_endpoints.py
index 03f0d1953e6..d9cfd65941d 100644
--- 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