support o3-mini

This commit is contained in:
Anton Shpigunov
2025-01-31 23:09:05 +02:00
parent 2bcc7c6e53
commit 58fdb6f9ef
2 changed files with 22 additions and 0 deletions

View File

@@ -289,6 +289,20 @@ 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" }],
})
})
it("should handle API errors", async () => {
mockCreate.mockRejectedValueOnce(new Error("API Error"))
await expect(handler.completePrompt("Test prompt")).rejects.toThrow(