diff --git a/lib/ai-gateway.sh b/lib/ai-gateway.sh
index 1d89ff4..e12f82f 100644
--- a/lib/ai-gateway.sh
+++ b/lib/ai-gateway.sh
@@ -182,7 +182,10 @@ api_model_id = sys.argv[4]
 data = json.loads(path.read_text(encoding="utf-8"))
 provider = data.setdefault("provider", {}).setdefault(provider_id, {})
 provider.setdefault("name", "WP AI Gateway")
-provider.setdefault("npm", "@ai-sdk/openai-compatible")
+if provider_id == "openai":
+    provider["npm"] = "@ai-sdk/openai"
+else:
+    provider.setdefault("npm", "@ai-sdk/openai-compatible")
 provider.setdefault("env", ["OPENAI_API_KEY"])
 provider.setdefault("options", {})
 provider["options"].setdefault("baseURL", "{env:OPENAI_BASE_URL}")
diff --git a/tests/ai-gateway.sh b/tests/ai-gateway.sh
index 2c2f8b7..cae8ab9 100755
--- a/tests/ai-gateway.sh
+++ b/tests/ai-gateway.sh
@@ -181,6 +181,7 @@ with open(sys.argv[1], encoding="utf-8") as handle:
     data = json.load(handle)
 
 model = data["provider"]["openai"]["models"]["gpt-5.6-sol"]
+assert data["provider"]["openai"]["npm"] == "@ai-sdk/openai"
 assert data["model"] == "openai/gpt-5.6-sol"
 assert model["id"] == "site-default"
 PY