update tests

2024-12-04 10:35:04 +00:00 · 2024-12-04 10:35:04 +00:00 · 46bd790b3a
parent e119e310ef
commit 46bd790b3a
2 changed files with 4 additions and 2 deletions
--- a/packages/server/src/api/routes/tests/row.spec.ts
+++ b/packages/server/src/api/routes/tests/row.spec.ts
@ -48,7 +48,7 @@ jest.mock("@budibase/pro", () => ({
  ai: {
    LargeLanguageModel: {
      forCurrentTenant: async () => ({
-        initialised: true,
+        llm: {},
        run: jest.fn(() => `Mock LLM Response`),
        buildPromptFromAIOperation: jest.fn(),
      }),
--- a/packages/server/src/automations/steps/openai.ts
+++ b/packages/server/src/automations/steps/openai.ts
@ -108,7 +108,9 @@ export async function run({

    let llmWrapper
    if (budibaseAIEnabled || customConfigsEnabled) {
-      llmWrapper = await pro.ai.LargeLanguageModel.forCurrentTenant(inputs.model)
+      llmWrapper = await pro.ai.LargeLanguageModel.forCurrentTenant(
+        inputs.model
+      )
    }

    response = llmWrapper?.llm