@budibase/pro 3.5.3 → 3.7.0

package/dist/ai/index.d.ts

@@ -1 +1 @@
-export { LargeLanguageModel } from "./LargeLanguageModel";
+export { getLLM } from "./llm";

package/dist/ai/llm.d.ts

@@ -0,0 +1,2 @@
+import { LLM } from "./models/base";
+export declare function getLLM(model?: string): Promise<LLM | undefined>;

package/dist/ai/models/anthropic.d.ts

@@ -1,15 +1,11 @@
-import { ILargeLanguageModel, LLMPromptResponse } from "../../types/ai";
+import AnthropicClient from "@anthropic-ai/sdk";
+import { LLMPromptResponse } from "../../types/ai";
+import { AIConfigOptions, LLM } from "./base";
+import { Prompt } from "../prompts";
 export type AnthropicModel = "claude-3-5-sonnet-20240620" | "claude-3-sonnet-20240229" | "claude-3-opus-20240229" | "claude-3-haiku-20240307";
-export interface AnthropicConfigOptions {
-    model: AnthropicModel;
-    apiKey: string;
-    measureUsage?: boolean;
-}
-export declare class Anthropic implements ILargeLanguageModel {
+export declare class Anthropic extends LLM {
     private client;
-    model: string;
-    constructor({ model, apiKey }: AnthropicConfigOptions);
-    chatCompletion(prompt: string): Promise<LLMPromptResponse>;
-    prompt(prompt: string): Promise<LLMPromptResponse>;
-    summarizeText(prompt: string): Promise<LLMPromptResponse>;
+    constructor(opts: AIConfigOptions);
+    firstTextBlock(message: AnthropicClient.Messages.Message): string | undefined;
+    protected chatCompletion(prompt: Prompt): Promise<LLMPromptResponse>;
 }

package/dist/ai/models/base.d.ts

@@ -0,0 +1,25 @@
+import { AIFieldMetadata, EnrichedBinding, Row, Snippet } from "@budibase/types";
+import { LLMPromptResponse } from "../../types/ai";
+import { Prompt } from "../prompts";
+export interface AIConfigOptions {
+    model: string;
+    apiKey: string;
+    measureUsage: boolean;
+}
+export declare abstract class LLM {
+    model: string;
+    apiKey: string;
+    measureUsage: boolean;
+    maxTokens: number;
+    constructor({ model, apiKey, measureUsage }: AIConfigOptions);
+    protected abstract chatCompletion(prompt: Prompt): Promise<LLMPromptResponse>;
+    prompt(prompt: string | Prompt): Promise<LLMPromptResponse>;
+    summarizeText(prompt: string): Promise<LLMPromptResponse>;
+    generateCronExpression(prompt: string): Promise<LLMPromptResponse>;
+    operation(schema: AIFieldMetadata, row: Row): Promise<LLMPromptResponse>;
+    private promptForOperation;
+    generateJs(prompt: string, opts?: {
+        bindings: EnrichedBinding[];
+        snippets: Snippet[];
+    }): Promise<LLMPromptResponse>;
+}

package/dist/ai/models/openai.d.ts

@@ -1,18 +1,11 @@
 import OpenAIClient from "openai";
-import { ILargeLanguageModel, LLMPromptResponse } from "../../types/ai";
+import { LLMPromptResponse } from "../../types/ai";
+import { AIConfigOptions, LLM } from "./base";
+import { Prompt } from "../prompts";
 export type OpenAIModel = "gpt-4o-mini" | "gpt-4o" | "gpt-4" | "gpt-4-turbo" | "gpt-3.5-turbo";
-export interface OpenAIConfigOptions {
-    model: OpenAIModel;
-    apiKey: string;
-    measureUsage: boolean;
-}
-export declare class OpenAI implements ILargeLanguageModel {
+export declare class OpenAI extends LLM {
     private client;
-    model: string;
-    measureUsage: boolean;
-    constructor({ model, apiKey, measureUsage }: OpenAIConfigOptions);
+    constructor({ model, apiKey, measureUsage }: AIConfigOptions);
     measureTokenUsage(usage?: OpenAIClient.CompletionUsage): number;
-    chatCompletion(prompt: string): Promise<LLMPromptResponse>;
-    prompt(prompt: string): Promise<LLMPromptResponse>;
-    summarizeText(prompt: string): Promise<LLMPromptResponse>;
+    protected chatCompletion(prompt: Prompt): Promise<LLMPromptResponse>;
 }

package/dist/ai/prompts/index.d.ts

@@ -1,9 +1,23 @@
-export declare function summarizeText(text: string): string;
-export declare function classifyText(text: string, categories: string[]): string;
-export declare function cleanData(text: string): string;
-export declare function generateSQL(prompt: string, tableSchema: string): string;
-export declare function generateCode(prompt: string): string;
-export declare function generateCronExpression(text: string): string;
-export declare function translate(text: string, language: string): string;
-export declare function sentimentAnalysis(text: string): string;
-export declare function searchWeb(text: string): string;
+import { EnrichedBinding, Snippet } from "@budibase/types";
+export interface Message {
+    role: "system" | "user";
+    content: string;
+}
+export declare class Prompt {
+    messages: Message[];
+    constructor(messages: Message[]);
+    user(content: string): this;
+    system(content: string): this;
+    static user(content: string): Prompt;
+    static system(content: string): Prompt;
+}
+export declare function summarizeText(text: string): Prompt;
+export declare function classifyText(text: string, categories: string[]): Prompt;
+export declare function cleanData(text: string): Prompt;
+export declare function generateSQL(prompt: string, tableSchema: string): Prompt;
+export declare function generateCode(prompt: string): Prompt;
+export declare function generateCronExpression(text: string): Prompt;
+export declare function translate(text: string, language: string): Prompt;
+export declare function sentimentAnalysis(text: string): Prompt;
+export declare function searchWeb(text: string): Prompt;
+export declare function generateJs(bindings: EnrichedBinding[], snippets: Snippet[]): Prompt;

package/dist/db/quotas/quotas.d.ts

@@ -16,6 +16,7 @@ interface SetAllUsageParams {
     opts?: {
         id?: string;
         appId?: string;
+        tenantId?: string;
     };
 }
 export declare const setAllUsage: (params: SetAllUsageParams | SetAllUsageParams[]) => Promise<QuotaUsage>;