@alpic80/rivet-core 1.19.1-aidon.3 → 1.24.0-aidon.3

package/dist/types/plugins/anthropic/anthropic.d.ts

@@ -63,7 +63,23 @@ export declare const anthropicModels: {
         };
         displayName: string;
     };
-    'claude-3-5-sonnet-20240620': {
+    'claude-3-5-sonnet-latest': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'claude-3-5-haiku-latest': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'claude-3-7-sonnet-latest': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -84,6 +100,7 @@ export type Claude3ChatMessage = {
 export type Claude3ChatMessageTextContentPart = {
     type: 'text';
     text: string;
+    cache_control: CacheControl;
 };
 export type Claude3ChatMessageImageContentPart = {
     type: 'image';
@@ -92,6 +109,21 @@ export type Claude3ChatMessageImageContentPart = {
         media_type: string;
         data: string;
     };
+    cache_control: CacheControl;
+};
+export type Claude3ChatMessageDocumentContentPart = {
+    type: 'document';
+    source: {
+        type: 'base64';
+        media_type: string;
+        data: string;
+    };
+    title: string | undefined;
+    context: string | undefined;
+    citations: undefined | {
+        enabled: true;
+    };
+    cache_control: CacheControl;
 };
 export type Claude3ChatMessageToolResultContentPart = {
     type: 'tool_result';
@@ -100,19 +132,22 @@ export type Claude3ChatMessageToolResultContentPart = {
         type: 'text';
         text: string;
     }[];
+    cache_control: CacheControl;
 };
 export type Claude3ChatMessageToolUseContentPart = {
     type: 'tool_use';
     id: string;
     name: string;
     input: object;
+    cache_control: CacheControl;
 };
-export type Claude3ChatMessageContentPart = Claude3ChatMessageTextContentPart | Claude3ChatMessageImageContentPart | Claude3ChatMessageToolResultContentPart | Claude3ChatMessageToolUseContentPart;
+export type Claude3ChatMessageContentPart = Claude3ChatMessageTextContentPart | Claude3ChatMessageImageContentPart | Claude3ChatMessageToolResultContentPart | Claude3ChatMessageToolUseContentPart | Claude3ChatMessageDocumentContentPart;
 export type ChatMessageOptions = {
+    apiEndpoint: string;
     apiKey: string;
     model: AnthropicModels;
     messages: Claude3ChatMessage[];
-    system?: string;
+    system?: SystemPrompt;
     max_tokens: number;
     stop_sequences?: string[];
     temperature?: number;
@@ -125,8 +160,10 @@ export type ChatMessageOptions = {
         description: string;
         input_schema: object;
     }[];
+    beta?: string;
 };
 export type ChatCompletionOptions = {
+    apiEndpoint: string;
     apiKey: string;
     model: AnthropicModels;
     prompt: string;
@@ -143,6 +180,15 @@ export type ChatCompletionChunk = {
     stop_reason: 'stop_sequence' | null;
     model: string;
 };
+export type CacheControl = null | {
+    type: 'ephemeral';
+};
+export type SystemPrompt = string | SystemPromptMessage[];
+export type SystemPromptMessage = {
+    cache_control: CacheControl;
+    type: 'text';
+    text: string;
+};
 export type ChatMessageChunk = {
     type: 'message_start';
     message: {
@@ -167,6 +213,11 @@ export type ChatMessageChunk = {
     content_block: {
         type: 'text';
         text: string;
+    } | {
+        type: 'tool_use';
+        id: string;
+        name: string;
+        input?: object;
     };
 } | {
     type: 'ping';
@@ -176,6 +227,12 @@ export type ChatMessageChunk = {
     delta: {
         type: 'text_delta';
         text: string;
+    } | {
+        type: 'citations_delta';
+        citation: ChatMessageCitation;
+    } | {
+        type: 'input_json_delta';
+        partial_json: string;
     };
 } | {
     type: 'message_delta';
@@ -188,16 +245,13 @@ export type ChatMessageChunk = {
     };
 } | {
     type: 'message_stop';
+} | {
+    type: 'content_block_stop';
+    index: number;
 };
 export type ChatMessageResponse = {
     id: string;
-    content: ({
-        text: string;
-    } | {
-        id: string;
-        name: string;
-        input: object;
-    })[];
+    content: ChatMessageContentItem[];
     model: string;
     stop_reason: 'end_turn';
     stop_sequence: string;
@@ -206,9 +260,36 @@ export type ChatMessageResponse = {
         output_tokens: number;
     };
 };
-export declare function streamChatCompletions({ apiKey, signal, ...rest }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;
-export declare function callMessageApi({ apiKey, signal, tools, ...rest }: ChatMessageOptions): Promise<ChatMessageResponse>;
-export declare function streamMessageApi({ apiKey, signal, ...rest }: ChatMessageOptions): AsyncGenerator<ChatMessageChunk>;
+export type ChatMessageTextContentItem = {
+    type: 'text';
+    text: string;
+    citations?: ChatMessageCitation[];
+};
+export type ChatMessageToolUseContentItem = {
+    type: 'tool_use';
+    id: string;
+    name: string;
+    input: object;
+};
+export type ChatMessageContentItem = ChatMessageTextContentItem | ChatMessageToolUseContentItem;
+export type ChatMessageCitation = {
+    type: 'char_location';
+    cited_text: string;
+    document_index: number;
+    document_title: string | null;
+    start_char_index: number;
+    end_chat_index: number;
+} | {
+    type: 'page_location';
+    cited_text: string;
+    document_index: number;
+    document_title: string | null;
+    page_number: number;
+    end_page_number: number;
+};
+export declare function streamChatCompletions({ apiEndpoint, apiKey, signal, ...rest }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;
+export declare function callMessageApi({ apiEndpoint, apiKey, signal, tools, beta, ...rest }: ChatMessageOptions): Promise<ChatMessageResponse>;
+export declare function streamMessageApi({ apiEndpoint, apiKey, signal, beta, ...rest }: ChatMessageOptions): AsyncGenerator<ChatMessageChunk>;
 export declare class AnthropicError extends Error {
     readonly response: Response;
     readonly responseJson: unknown;

package/dist/types/plugins/anthropic/nodes/ChatAnthropicNode.d.ts

@@ -1,5 +1,5 @@
 import { type ChartNode, type ChatMessage, type Inputs, type PluginNodeImpl } from '../../../index.js';
-import { type AnthropicModels, type Claude3ChatMessage } from '../anthropic.js';
+import { type AnthropicModels, type Claude3ChatMessage, type SystemPrompt } from '../anthropic.js';
 export type ChatAnthropicNode = ChartNode<'chatAnthropic', ChatAnthropicNodeData>;
 export type ChatAnthropicNodeConfigData = {
     model: AnthropicModels;
@@ -10,6 +10,9 @@ export type ChatAnthropicNodeConfigData = {
     maxTokens: number;
     stop?: string;
     enableToolUse?: boolean;
+    endpoint?: string;
+    overrideModel?: string;
+    enableCitations?: boolean;
 };
 export type ChatAnthropicNodeData = ChatAnthropicNodeConfigData & {
     useModelInput: boolean;
@@ -20,11 +23,13 @@ export type ChatAnthropicNodeData = ChatAnthropicNodeConfigData & {
     useMaxTokensInput: boolean;
     useStop: boolean;
     useStopInput: boolean;
+    useEndpointInput: boolean;
+    useOverrideModelInput: boolean;
     /** Given the same set of inputs, return the same output without hitting GPT */
     cache: boolean;
     useAsGraphPartialOutput?: boolean;
 };
 export declare const ChatAnthropicNodeImpl: PluginNodeImpl<ChatAnthropicNode>;
 export declare const chatAnthropicNode: import("../../../index.js").PluginNodeDefinition<ChatAnthropicNode>;
-export declare function getSystemPrompt(inputs: Inputs): string | undefined;
+export declare function getSystemPrompt(inputs: Inputs): SystemPrompt | undefined;
 export declare function chatMessagesToClaude3ChatMessages(chatMessages: ChatMessage[]): Promise<Claude3ChatMessage[]>;

package/dist/types/plugins/google/google.d.ts

@@ -1,4 +1,5 @@
-export type GoogleModel = {
+import { type Content, type FunctionCall, type Tool } from '@google/generative-ai';
+export type GoogleModelDeprecated = {
     maxTokens: number;
     cost: {
         prompt: number;
@@ -6,7 +7,7 @@ export type GoogleModel = {
     };
     displayName: string;
 };
-export declare const googleModels: {
+export declare const googleModelsDeprecated: {
     'gemini-pro': {
         maxTokens: number;
         cost: {
@@ -24,28 +25,96 @@ export declare const googleModels: {
         displayName: string;
     };
 };
-export type GoogleModels = keyof typeof googleModels;
-export declare const googleModelOptions: {
+export type GoogleModelsDeprecated = keyof typeof googleModelsDeprecated;
+export declare const generativeAiGoogleModels: {
+    'gemini-2.0-flash-001': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-2.0-pro-exp-02-05': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-2.5-pro-exp-03-25': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-2.0-flash-lite-preview-02-05': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-2.0-flash-thinking-exp-01-21': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-1.5-flash': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-1.5-pro': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-1.0-pro': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'gemini-1.0-pro-vision': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+};
+export type GenerativeAiGoogleModel = keyof typeof generativeAiGoogleModels;
+export declare const googleModelOptionsDeprecated: {
+    value: string;
+    label: string;
+}[];
+export declare const generativeAiOptions: {
     value: string;
     label: string;
 }[];
-export interface GoogleChatMessage {
-    role: 'user' | 'assistant';
-    parts: ({
-        text: string;
-    } | {
-        inline_data: {
-            mime_type: string;
-            data: string;
-        };
-    })[];
-}
 export type ChatCompletionOptions = {
     project: string;
     location: string;
     applicationCredentials: string;
-    model: GoogleModels;
-    prompt: GoogleChatMessage[];
+    model: GoogleModelsDeprecated;
+    prompt: Content[];
     max_output_tokens: number;
     temperature?: number;
     top_p?: number;
@@ -53,8 +122,22 @@ export type ChatCompletionOptions = {
     signal?: AbortSignal;
 };
 export type ChatCompletionChunk = {
-    completion: string;
+    completion?: string;
+    function_calls?: FunctionCall[];
     finish_reason: 'FINISH_REASON_UNSPECIFIED' | 'FINISH_REASON_STOP' | 'FINISH_REASON_MAX_TOKENS' | 'FINISH_REASON_SAFETY' | 'FINISH_REASON_RECITATION' | 'FINISH_REASON_OTHER' | undefined;
     model: string;
 };
+export type StreamGenerativeAiOptions = {
+    apiKey: string;
+    model: GenerativeAiGoogleModel;
+    systemPrompt: string | undefined;
+    prompt: Content[];
+    maxOutputTokens: number;
+    temperature: number | undefined;
+    topP: number | undefined;
+    topK: number | undefined;
+    signal?: AbortSignal;
+    tools: Tool[] | undefined;
+};
+export declare function streamGenerativeAi({ apiKey, model, systemPrompt, prompt, maxOutputTokens, temperature, topP, topK, signal, tools, }: StreamGenerativeAiOptions): AsyncGenerator<ChatCompletionChunk>;
 export declare function streamChatCompletions({ project, location, applicationCredentials, model, signal, max_output_tokens, temperature, top_p, top_k, prompt, }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;

package/dist/types/plugins/google/nodes/ChatGoogleNode.d.ts

@@ -1,8 +1,8 @@
 import { type ChartNode, type ChatMessage, type Inputs, type PluginNodeImpl } from '../../../index.js';
-import { type GoogleModels } from '../google.js';
+import { type GenerativeAiGoogleModel } from '../google.js';
 export type ChatGoogleNode = ChartNode<'chatGoogle', ChatGoogleNodeData>;
 export type ChatGoogleNodeConfigData = {
-    model: GoogleModels;
+    model: GenerativeAiGoogleModel;
     temperature: number;
     useTopP: boolean;
     top_p?: number;
@@ -16,6 +16,7 @@ export type ChatGoogleNodeData = ChatGoogleNodeConfigData & {
     useTopKInput: boolean;
     useUseTopPInput: boolean;
     useMaxTokensInput: boolean;
+    useToolCalling: boolean;
     /** Given the same set of inputs, return the same output without hitting GPT */
     cache: boolean;
     useAsGraphPartialOutput?: boolean;

package/dist/types/recording/RecordedEvents.d.ts

@@ -58,9 +58,11 @@ export type RecordedEventsMap = OverrideProperties<ProcessEvents, {
     /** Called when a user input node requires user input. Call the callback when finished, or call userInput() on the GraphProcessor with the results. */
     userInput: {
         nodeId: NodeId;
+        inputStrings: string[];
         inputs: Inputs;
         callback: (values: StringArrayDataValue) => void;
         processId: ProcessId;
+        renderingType: 'text' | 'markdown';
     };
     /** Called when a node has partially processed, with the current partial output values for the node. */
     partialOutput: {
@@ -97,4 +99,5 @@ export type Recording = {
 export type SerializedRecording = {
     version: number;
     recording: Recording;
+    assets: Record<string, string>;
 };

package/dist/types/utils/base64.d.ts

@@ -1,2 +1,3 @@
 export declare function uint8ArrayToBase64(uint8Array: Uint8Array): Promise<string | undefined>;
-export declare function base64ToUint8Array(base64: string): Uint8Array;
+export declare function uint8ArrayToBase64Sync(uint8Array: Uint8Array): string;
+export declare function base64ToUint8Array(base64: string): Uint8Array<ArrayBuffer>;

package/dist/types/utils/chatMessageToOpenAIChatCompletionMessage.d.ts

@@ -1,3 +1,5 @@
 import type { ChatMessage } from '../index.js';
 import type { ChatCompletionRequestMessage } from './openai.js';
-export declare function chatMessageToOpenAIChatCompletionMessage(message: ChatMessage): Promise<ChatCompletionRequestMessage>;
+export declare function chatMessageToOpenAIChatCompletionMessage(message: ChatMessage, options: {
+    isReasoningModel: boolean;
+}): Promise<ChatCompletionRequestMessage>;

package/dist/types/utils/interpolation.d.ts

@@ -1 +1,4 @@
+export declare const TOKEN_MATCH_REGEX: RegExp;
+export declare const ESCAPED_TOKEN_REGEX: RegExp;
 export declare function interpolate(baseString: string, values: Record<string, string>): string;
+export declare function extractInterpolationVariables(template: string): string[];