This commit is contained in:
Martin McKeaveney 2024-09-18 14:21:10 +01:00
parent 9cbaa02f5f
commit 69b2bf9bdb
2 changed files with 3 additions and 4 deletions

View File

@ -262,4 +262,3 @@ export async function getSCIMConfig(): Promise<SCIMInnerConfig | undefined> {
export async function getAIConfig(): Promise<AIConfig | undefined> { export async function getAIConfig(): Promise<AIConfig | undefined> {
return getConfig<AIConfig>(ConfigType.AI) return getConfig<AIConfig>(ConfigType.AI)
} }

View File

@ -35,7 +35,7 @@ jest.mock("@budibase/pro", () => ({
features: { features: {
isAICustomConfigsEnabled: jest.fn(), isAICustomConfigsEnabled: jest.fn(),
isBudibaseAIEnabled: jest.fn(), isBudibaseAIEnabled: jest.fn(),
} },
})) }))
const mockedOpenAI = OpenAI as jest.MockedClass<typeof OpenAI> const mockedOpenAI = OpenAI as jest.MockedClass<typeof OpenAI>
@ -102,7 +102,7 @@ describe("test the openai action", () => {
) )
const res = await runStep("OPENAI", { const res = await runStep("OPENAI", {
prompt: OPENAI_PROMPT,jj prompt: OPENAI_PROMPT,
}) })
expect(res.response).toEqual( expect(res.response).toEqual(
@ -118,7 +118,7 @@ describe("test the openai action", () => {
const prompt = "What is the meaning of life?" const prompt = "What is the meaning of life?"
await runStep("OPENAI", { await runStep("OPENAI", {
model: "gpt-4o-mini", model: "gpt-4o-mini",
prompt prompt,
}) })
expect(pro.ai.LargeLanguageModel).toHaveBeenCalledWith("gpt-4o-mini") expect(pro.ai.LargeLanguageModel).toHaveBeenCalledWith("gpt-4o-mini")