Merge pull request #16158 from Budibase/BUDI-9238/fix-truncated-json

Fixing AI table generation
2025-05-15 17:03:30 +02:00 · 2025-05-15 17:03:30 +02:00 · 90aa62e643
parent 3e615249d7 a644505345
commit 90aa62e643
3 changed files with 3 additions and 2 deletions
--- a/packages/pro
+++ b/packages/pro
@ -1 +1 @@
-Subproject commit f2cbe5aff7645eb9b0b4e864924a1e1171ad85bf
+Subproject commit 271b8677c3ea814395cd67a10137900ec3a34dc7
--- a/packages/server/src/automations/steps/openai.ts
+++ b/packages/server/src/automations/steps/openai.ts
@ -41,7 +41,7 @@ export async function run({

  try {
    let response
-    const llm = await ai.getLLM(inputs.model)
+    const llm = await ai.getLLM({ model: inputs.model })
    response = llm
      ? (await llm.prompt(inputs.prompt)).message
      : await legacyOpenAIPrompt(inputs)
--- a/packages/types/src/sdk/ai.ts
+++ b/packages/types/src/sdk/ai.ts
@ -95,6 +95,7 @@ export type AIColumnSchema =
 export interface LLMConfigOptions {
  model: string
  apiKey?: string
+  maxTokens?: number
 }

 export interface LLMProviderConfig extends LLMConfigOptions {