@alpic80/rivet-core 1.19.1-aidon.2 → 1.24.0-aidon.1

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

@@ -3,9 +3,15 @@ export type OpenAIModel = {
     cost: {
         prompt: number;
         completion: number;
+        audioPrompt?: number;
+        audioCompletion?: number;
     };
     displayName: string;
+    supported?: {
+        parallelFunctionCalls: boolean;
+    };
 };
+export declare const defaultOpenaiSupported: NonNullable<OpenAIModel['supported']>;
 export declare const openaiModels: {
     'gpt-4': {
         maxTokens: number;
@@ -39,7 +45,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo': {
+    'gpt-4-0314': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -47,7 +53,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo-16k': {
+    'gpt-4-32k-0314': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -55,7 +61,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo-0613': {
+    'gpt-4-1106-preview': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -63,7 +69,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo-1106': {
+    'gpt-4-turbo': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -71,7 +77,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo-16k-0613': {
+    'gpt-4-vision-preview': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -79,7 +85,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-3.5-turbo-0301': {
+    'gpt-4o': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -87,7 +93,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-4-0314': {
+    'gpt-4o-mini': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -95,7 +101,7 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-4-32k-0314': {
+    'gpt-4o-mini-2024-07-18': {
         maxTokens: number;
         cost: {
             prompt: number;
@@ -103,53 +109,92 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
-    'gpt-4-1106-preview': {
+    o1: {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
     };
-    'gpt-4-turbo': {
+    'o1-preview': {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
     };
-    'gpt-4-vision-preview': {
+    'o1-preview-2024-09-12': {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
     };
-    'gpt-4o': {
+    'o1-mini': {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
     };
-    'gpt-4o-mini': {
+    'o1-mini-2024-09-12': {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
     };
-    'gpt-4o-mini-2024-07-18': {
+    'o3-mini': {
         maxTokens: number;
         cost: {
             prompt: number;
             completion: number;
         };
         displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
+    };
+    'o3-mini-2025-01-31': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+        supported: {
+            parallelFunctionCalls: false;
+        };
+    };
+    'gpt-4o-audio-preview': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+            audioPrompt: number;
+            audioCompletion: number;
+        };
+        displayName: string;
     };
     'local-model': {
         maxTokens: number;
@@ -170,11 +215,15 @@ export declare class OpenAIError extends Error {
     constructor(status: number, responseJson: any);
 }
 export type ChatCompletionRole = ChatCompletionRequestMessage['role'];
-export type ChatCompletionRequestMessage = ChatCompletionRequestSystemMessage | ChatCompletionRequestUserMessage | ChatCompletionRequestAssistantMessage | ChatCompletionRequestToolMessage;
+export type ChatCompletionRequestMessage = ChatCompletionRequestSystemMessage | ChatCompletionRequestUserMessage | ChatCompletionRequestAssistantMessage | ChatCompletionRequestToolMessage | ChatCompletionRequestDeveloperMessage;
 export type ChatCompletionRequestSystemMessage = {
     role: 'system';
     content: string | null;
 };
+export type ChatCompletionRequestDeveloperMessage = {
+    role: 'developer';
+    content: string | null;
+};
 export type ChatCompletionRequestUserMessage = {
     role: 'user';
     content: string | ChatCompletionRequestUserMessageContent[];
@@ -225,6 +274,8 @@ export type ChatCompletionOptions = {
     temperature?: number;
     top_p?: number;
     max_tokens?: number;
+    /** Only for o1 series of models. Otherwise max_tokens. */
+    max_completion_tokens?: number;
     n?: number;
     stop?: string | string[];
     presence_penalty?: number;
@@ -241,6 +292,7 @@ export type ChatCompletionOptions = {
             name: string;
         };
     };
+    parallel_tool_calls?: boolean;
     /** An object specifying the format that the model must output. Used to enable JSON mode. */
     response_format?: {
         type: 'text';
@@ -259,6 +311,19 @@ export type ChatCompletionOptions = {
             schema: object;
         };
     };
+    prediction?: {
+        type: 'content';
+        content: string | {
+            type: string;
+            text: string;
+        }[];
+    };
+    modalities: ('text' | 'audio')[] | undefined;
+    audio?: {
+        voice: string;
+        format: 'wav' | 'mp3' | 'flac' | 'opus' | 'pcm16';
+    };
+    reasoning_effort?: 'low' | 'medium' | 'high';
 };
 export type ChatCompletionResponse = {
     /** A unique identifier for the chat completion. */
@@ -282,9 +347,32 @@ export type ChatCompletionResponse = {
         prompt_tokens: number;
         /** Total number of tokens used in the request (prompt + completion). */
         total_tokens: number;
+        prompt_tokens_details: {
+            /** Number of tokens used from the cache. */
+            cached_tokens: number;
+            /** Number of tokens used for audio. */
+            audio_tokens: number;
+            text_tokens: number;
+            image_tokens: number;
+        };
+        completion_tokens_details: {
+            /** Number of tokens used for reasoning. */
+            reasoning_tokens: number;
+            /** Number of tokens used for audio. */
+            audio_tokens: number;
+            /** Number of tokens used for accepted predictions. */
+            accepted_prediction_tokens: number;
+            /** Number of tokens used for rejected predictions. */
+            rejected_prediction_tokens: number;
+            text_tokens: number;
+        };
     };
     /** A list of chat completion choices. Can be more than one if n is greater than 1. */
     choices: ChatCompletionResponseChoice[];
+} | {
+    error: {
+        message: string;
+    };
 };
 export type ChatCompletionResponseChoice = {
     /** The index of the choice in the list of choices. */
@@ -305,13 +393,35 @@ export type ChatCompletionResponseMessage = {
     content: string | null;
     /** The tool calls generated by the model, such as function calls. */
     tool_calls: OpenAIFunctionToolCall[];
+    refusal: string | null;
+    audio?: {
+        id: string;
+        data: string;
+        expires_at: number;
+        transcript: string;
+    };
 };
 export type ChatCompletionChunk = {
     object: 'chat.completion.chunk';
     created: number;
     model: string;
     choices?: ChatCompletionChunkChoice[];
-    usage?: CompletionUsage;
+    usage?: ChatCompletionChunkUsage;
+};
+export type ChatCompletionChunkUsage = {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+    prompt_tokens_details: {
+        cached_tokens: number;
+        audio_tokens: number;
+    };
+    completion_tokens_details: {
+        reasoning_tokens: number;
+        audio_tokens: number;
+        accepted_prediction_tokens: number;
+        rejected_prediction_tokens: number;
+    };
 };
 export type GptFunctionCall = {
     name: string;
@@ -321,11 +431,6 @@ export type GptFunctionCallDelta = {
     name?: string;
     arguments?: string;
 };
-export interface CompletionUsage {
-    completion_tokens: number;
-    prompt_tokens: number;
-    total_tokens: number;
-}
 export type ChatCompletionChunkChoice = {
     index: number;
     message_index: number;
@@ -353,6 +458,7 @@ export type ChatCompletionFunction = {
     parameters: object;
     strict: boolean;
 };
+export declare function chatCompletions({ endpoint, auth, signal, headers, timeout, ...rest }: ChatCompletionOptions): Promise<ChatCompletionResponse>;
 export declare function streamChatCompletions({ endpoint, auth, signal, headers, timeout, ...rest }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;
 export type OpenAIAssistant = {
     /** The identifier, which can be referenced in API endpoints. */

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

@@ -0,0 +1,8 @@
+interface TreeNode {
+    path: string;
+    name: string;
+    isDirectory: boolean;
+    children: TreeNode[];
+}
+export declare function createTreeFromPaths(paths: string[], rootPath: string): TreeNode;
+export {};

package/package.json

@@ -2,7 +2,7 @@
   "name": "@alpic80/rivet-core",
   "license": "MIT",
   "repository": "https://github.com/castortech/rivet",
-  "version": "1.19.1-aidon.2",
+  "version": "1.24.0-aidon.1",
   "packageManager": "yarn@3.5.0",
   "main": "dist/cjs/bundle.cjs",
   "module": "dist/esm/index.js",
@@ -38,23 +38,27 @@
     "build:esm": "rimraf dist/esm dist/types dist/tsconfig.tsbuildinfo && tsc -b --verbose",
     "build:cjs": "rimraf dist/cjs && tsx bundle.esbuild.ts",
     "prepack": "yarn build && cp -r ../../LICENSE ../../README.md .",
-    "publish": "yarn npm publish",
-    "lint": "eslint --ext .js,.jsx,.ts,.tsx ./src",
+    "publish": "yarn npm publish --access public",
+    "lint": "eslint ./src",
     "test": "tsx --test test/**/*.test.ts",
     "watch": "tsc -b -w"
   },
   "dependencies": {
     "@gentrace/core": "^2.2.5",
     "@google-cloud/vertexai": "^0.1.3",
+    "@google/generative-ai": "^0.21.0",
     "@huggingface/inference": "^2.6.4",
     "assemblyai": "^4.6.0",
     "autoevals": "^0.0.26",
+    "cron-parser": "^4.9.0",
     "crypto-js": "^4.2.0",
     "emittery": "^1.0.1",
     "emittery-0-13": "npm:emittery@^0.13.1",
     "gpt-tokenizer": "^2.1.2",
-    "jsonpath-plus": "^7.2.0",
+    "jsonpath-plus": "^10.2.0",
     "lodash-es": "^4.17.21",
+    "mdast-util-gfm-table": "^2.0.0",
+    "mdast-util-to-markdown": "^2.1.2",
     "minimatch": "^9.0.3",
     "nanoid": "^3.3.6",
     "openai": "^4.28.4",
@@ -64,8 +68,8 @@
     "p-retry-4": "npm:p-retry@^4.0.0",
     "safe-stable-stringify": "^2.4.3",
     "ts-dedent": "^2.2.0",
-    "ts-pattern": "^5.0.5",
-    "type-fest": "^4.5.0",
+    "ts-pattern": "^5.6.2",
+    "type-fest": "^4.34.1",
     "yaml": "^2.3.3"
   },
   "devDependencies": {
@@ -76,12 +80,12 @@
     "@types/lodash": "^4.14.200",
     "@types/lodash-es": "^4.17.10",
     "@types/yaml": "^1.9.7",
-    "@typescript-eslint/eslint-plugin": "^6.9.0",
-    "@typescript-eslint/parser": "^6.9.0",
+    "@typescript-eslint/eslint-plugin": "^8.24.0",
+    "@typescript-eslint/parser": "^8.24.0",
     "esbuild": "^0.19.5",
-    "eslint": "^8.52.0",
+    "eslint": "^9.20.1",
     "eslint-import-resolver-typescript": "^3.6.1",
-    "eslint-plugin-import": "npm:eslint-plugin-i@latest",
+    "eslint-plugin-import": "^2.31.0",
     "eslint-plugin-n": "^16.2.0",
     "eslint-plugin-promise": "^6.1.1",
     "lodash": "^4.17.21",
@@ -89,7 +93,7 @@
     "rimraf": "^5.0.5",
     "ts-node": "^10.9.1",
     "tsx": "^4.6.2",
-    "typescript": "^5.2.2"
+    "typescript": "^5.7.3"
   },
   "volta": {
     "node": "20.4.0"