Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions langfuse/openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ class OpenAiDefinition:
type: str
sync: bool
min_version: Optional[str] = None
max_version: Optional[str] = None


OPENAI_METHODS_V0 = [
Expand Down Expand Up @@ -116,6 +117,7 @@ class OpenAiDefinition:
type="chat",
sync=True,
min_version="1.50.0",
max_version="1.92.0",
),
OpenAiDefinition(
module="openai.resources.beta.chat.completions",
Expand All @@ -124,6 +126,7 @@ class OpenAiDefinition:
type="chat",
sync=False,
min_version="1.50.0",
max_version="1.92.0",
),
OpenAiDefinition(
module="openai.resources.responses",
Expand Down Expand Up @@ -799,6 +802,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}",
Expand Down
Loading
Loading