diff --git a/langfuse/openai.py b/langfuse/openai.py index 7d0f2a454..b3981eb40 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -60,6 +60,7 @@ class OpenAiDefinition: type: str sync: bool min_version: Optional[str] = None + max_version: Optional[str] = None OPENAI_METHODS_V0 = [ @@ -116,6 +117,7 @@ class OpenAiDefinition: type="chat", sync=True, min_version="1.50.0", + max_version="1.91.0", ), OpenAiDefinition( module="openai.resources.beta.chat.completions", @@ -124,6 +126,23 @@ class OpenAiDefinition: type="chat", sync=False, min_version="1.50.0", + max_version="1.91.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", @@ -799,6 +818,11 @@ def register_tracing(): ): 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}",