diff --git a/langfuse/openai.py b/langfuse/openai.py index f27fedcff..d0c7a815b 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -62,6 +62,7 @@ class OpenAiDefinition: type: str sync: bool min_version: Optional[str] = None + max_version: Optional[str] = None OPENAI_METHODS_V0 = [ @@ -118,6 +119,7 @@ class OpenAiDefinition: type="chat", sync=True, min_version="1.50.0", + max_version="1.92.0", ), OpenAiDefinition( module="openai.resources.beta.chat.completions", @@ -126,6 +128,23 @@ class OpenAiDefinition: type="chat", sync=False, min_version="1.50.0", + max_version="1.92.0", + ), + OpenAiDefinition( + module="openai.resources.chat.completions", + object="Completions", + method="parse", + type="chat", + sync=True, + min_version="1.92.0", + ), + OpenAiDefinition( + module="openai.resources.chat.completions", + object="AsyncCompletions", + method="parse", + type="chat", + sync=False, + min_version="1.92.0", ), OpenAiDefinition( module="openai.resources.responses", @@ -876,6 +895,11 @@ def register_tracing(self): ) < Version(resource.min_version): continue + if resource.max_version is not None and Version( + openai.__version__ + ) >= Version(resource.max_version): + continue + wrap_function_wrapper( resource.module, f"{resource.object}.{resource.method}",