Skip to content

revert openai agents sdk temp fix#367

Merged
bbqiu merged 2 commits intomainfrom
bbqiu/revert-temp-fix
Mar 5, 2026
Merged

revert openai agents sdk temp fix#367
bbqiu merged 2 commits intomainfrom
bbqiu/revert-temp-fix

Conversation

@bbqiu
Copy link
Copy Markdown
Collaborator

@bbqiu bbqiu commented Mar 5, 2026

reverting #342 now that mlflow/mlflow#21038 is released

manually tested

running the following file generates the results below in the two diff cases:

import databricks_openai
import mlflow
import os

os.environ["DATABRICKS_CONFIG_PROFILE"] = "dogfood"
mlflow.set_tracking_uri("databricks")
mlflow.set_experiment(experiment_id="1981078730603410")

# mlflow.openai.autolog(disable_openai_agent_tracer=False)
mlflow.openai.autolog()

client = databricks_openai.DatabricksOpenAI()

response = client.chat.completions.create(
    model="databricks-gpt-5",
    messages=[{"role": "user", "content": "What is the capital of France?"}],
)

print(response)

Results confirm everything:
  ┌───────────────────┬─────────────────────┬──────────────────────────────────────────────────────────────────────┐
  │ databricks-openai │       mlflow        │                                Result                                │
  ├───────────────────┼─────────────────────┼──────────────────────────────────────────────────────────────────────┤
  │ 0.12.0            │ 3.10.1 (with fix)   │ Works                                                                │
  ├───────────────────┼─────────────────────┼──────────────────────────────────────────────────────────────────────┤
  │ 0.12.0            │ 3.10.0 (without     │ AttributeError: 'NoneType' object has no attribute                   │
  │                   │ fix)                │ '_multi_processor'                                                   │
  └───────────────────┴─────────────────────┴──────────────────────────────────────────────────────────────────────┘```

Signed-off-by: Bryan Qiu <bryan.qiu@databricks.com>
@bbqiu bbqiu changed the title start revert openai agents sdk temp fix Mar 5, 2026
@bbqiu bbqiu requested a review from fanzeyi March 5, 2026 23:54
Signed-off-by: Bryan Qiu <bryan.qiu@databricks.com>
@bbqiu bbqiu enabled auto-merge (squash) March 5, 2026 23:56
@bbqiu bbqiu merged commit 779f3b6 into main Mar 5, 2026
46 checks passed
@bbqiu bbqiu deleted the bbqiu/revert-temp-fix branch March 5, 2026 23:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants