Skip to content

Commit ccbc9f6

Browse files
chore: add oepnai provider
1 parent cb773fe commit ccbc9f6

File tree

3 files changed

+40
-2
lines changed

3 files changed

+40
-2
lines changed

package-lock.json

Lines changed: 19 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

package.json

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,14 @@
3030
"reformat": "prettier --write .",
3131
"generate": "./scripts/generate.sh",
3232
"test": "node --experimental-vm-modules node_modules/jest/bin/jest.js --coverage --testPathIgnorePatterns=/tests/accuracy/",
33-
"test:accuracy": "node --experimental-vm-modules node_modules/jest/bin/jest.js --testPathPattern tests/accuracy"
33+
"test:accuracy": "node --experimental-vm-modules node_modules/jest/bin/jest.js --testPathPattern tests/accuracy",
34+
"test:accuracy-file": "node --experimental-vm-modules node_modules/jest/bin/jest.js"
3435
},
3536
"license": "Apache-2.0",
3637
"devDependencies": {
38+
"@ai-sdk/anthropic": "^1.2.12",
3739
"@ai-sdk/azure": "^1.3.23",
40+
"@ai-sdk/openai": "^1.3.22",
3841
"@eslint/js": "^9.24.0",
3942
"@himanshusinghs/google": "^1.2.11",
4043
"@jest/globals": "^30.0.0",

tests/accuracy/sdk/models.ts

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import { LanguageModelV1 } from "ai";
22
import { createGoogleGenerativeAI } from "@himanshusinghs/google";
33
import { createAzure } from "@ai-sdk/azure";
4+
import { createOpenAI } from "@ai-sdk/openai";
45
import { ollama } from "ollama-ai-provider";
56

67
export interface Model<P extends LanguageModelV1 = LanguageModelV1> {
@@ -12,6 +13,20 @@ export interface Model<P extends LanguageModelV1 = LanguageModelV1> {
1213
export class OpenAIModel implements Model {
1314
constructor(readonly modelName: string) {}
1415

16+
isAvailable(): boolean {
17+
return !!process.env.MDB_OPEN_AI_API_KEY;
18+
}
19+
20+
getModel() {
21+
return createOpenAI({
22+
apiKey: process.env.MDB_OPEN_AI_API_KEY,
23+
})(this.modelName);
24+
}
25+
}
26+
27+
export class AzureOpenAIModel implements Model {
28+
constructor(readonly modelName: string) {}
29+
1530
isAvailable(): boolean {
1631
return !!process.env.MDB_AZURE_OPEN_AI_API_KEY && !!process.env.MDB_AZURE_OPEN_AI_API_URL;
1732
}
@@ -53,7 +68,8 @@ export class OllamaModel implements Model {
5368

5469
const ALL_TESTABLE_MODELS = [
5570
new GeminiModel("gemini-2.0-flash"),
56-
new OpenAIModel("gpt-4o"),
71+
// new OpenAIModel("gpt-4o"),
72+
// new AzureOpenAIModel("gpt-4o"),
5773
// new OllamaModel("qwen3:1.7b"),
5874
];
5975

0 commit comments

Comments
 (0)