Merge pull request #686 from shpigunov/origin/feature/support-o3-mini

support o3-mini
This commit is contained in:
Matt Rubens
2025-01-31 16:26:34 -05:00
committed by GitHub
2 changed files with 23 additions and 0 deletions

View File

@@ -289,6 +289,21 @@ describe("OpenAiNativeHandler", () => {
}) })
}) })
it("should complete prompt successfully with o3-mini model", async () => {
handler = new OpenAiNativeHandler({
apiModelId: "o3-mini",
openAiNativeApiKey: "test-api-key",
})
const result = await handler.completePrompt("Test prompt")
expect(result).toBe("Test response")
expect(mockCreate).toHaveBeenCalledWith({
model: "o3-mini",
messages: [{ role: "user", content: "Test prompt" }],
temperature: 0,
})
})
it("should handle API errors", async () => { it("should handle API errors", async () => {
mockCreate.mockRejectedValueOnce(new Error("API Error")) mockCreate.mockRejectedValueOnce(new Error("API Error"))
await expect(handler.completePrompt("Test prompt")).rejects.toThrow( await expect(handler.completePrompt("Test prompt")).rejects.toThrow(

View File

@@ -510,6 +510,14 @@ export type OpenAiNativeModelId = keyof typeof openAiNativeModels
export const openAiNativeDefaultModelId: OpenAiNativeModelId = "gpt-4o" export const openAiNativeDefaultModelId: OpenAiNativeModelId = "gpt-4o"
export const openAiNativeModels = { export const openAiNativeModels = {
// don't support tool use yet // don't support tool use yet
"o3-mini": {
maxTokens: 100_000,
contextWindow: 200_000,
supportsImages: false,
supportsPromptCache: false,
inputPrice: 1.1,
outputPrice: 4.4,
},
o1: { o1: {
maxTokens: 100_000, maxTokens: 100_000,
contextWindow: 200_000, contextWindow: 200_000,