@valerypopoff/rivet2-cli 2.0.9 → 2.0.11

package/bin/commands/serve.js

@@ -147,13 +147,13 @@ function buildGraphRunArgs(args, project, inputs, graph) {
     };
 }
 async function streamGraph({ project, inputs, graph, openaiApiKey, openaiEndpoint, openaiOrganization, stream, streamNode, }) {
-    const { run, processor, getSSEStream } = createProcessor(project, {
-        inputs,
+    const { run, processor, getSSEStream } = createProcessor(project, buildStreamingGraphProcessorOptions({
         graph,
-        openAiKey: openaiApiKey,
-        openAiEndpoint: openaiEndpoint,
-        openAiOrganization: openaiOrganization,
-    });
+        inputs,
+        openaiApiKey,
+        openaiEndpoint,
+        openaiOrganization,
+    }));
     const responseStream = streamNode
         ? getSingleNodeStream(processor, streamNode)
         : getSSEStream(buildStreamEventFilter(stream));
@@ -162,6 +162,21 @@ async function streamGraph({ project, inputs, graph, openaiApiKey, openaiEndpoin
     });
     return responseStream;
 }
+export function buildGraphProcessorOptions({ inputs, graph, openaiApiKey, openaiEndpoint, openaiOrganization, }) {
+    return {
+        inputs,
+        graph,
+        openAiKey: openaiApiKey,
+        openAiEndpoint: openaiEndpoint,
+        openAiOrganization: openaiOrganization,
+    };
+}
+export function buildStreamingGraphProcessorOptions(args) {
+    return {
+        ...buildGraphProcessorOptions(args),
+        runtimeProfile: 'compatible',
+    };
+}
 export function buildStreamEventFilter(stream) {
     const streamTarget = stream?.trim();
     if (!streamTarget) {
@@ -178,13 +193,13 @@ export function buildStreamEventFilter(stream) {
     };
 }
 async function runGraph({ project, inputs, graph, openaiApiKey, openaiEndpoint, openaiOrganization, exposeCost, }) {
-    const { run } = createProcessor(project, {
-        inputs,
+    const { run } = createProcessor(project, buildGraphProcessorOptions({
         graph,
-        openAiKey: openaiApiKey,
-        openAiEndpoint: openaiEndpoint,
-        openAiOrganization: openaiOrganization,
-    });
+        inputs,
+        openaiApiKey,
+        openaiEndpoint,
+        openaiOrganization,
+    }));
     const outputs = await run();
     if (!exposeCost) {
         delete outputs.cost;

package/dist/types/commands/serve.d.ts

@@ -1,4 +1,4 @@
-import type { RivetEventStreamFilterSpec } from '@valerypopoff/rivet2-node';
+import type { LooseDataValue, NodeCreateProcessorOptions, Project, RivetEventStreamFilterSpec } from '@valerypopoff/rivet2-node';
 import type * as yargs from 'yargs';
 type ServeArgs = {
     port: number;
@@ -13,6 +13,16 @@ type ServeArgs = {
     stream: string | undefined;
     streamNode: string | undefined;
 };
+type GraphRunArgs = {
+    exposeCost: boolean;
+    graph: string | undefined;
+    inputs: Record<string, LooseDataValue>;
+    openaiApiKey: string | undefined;
+    openaiEndpoint: string | undefined;
+    openaiOrganization: string | undefined;
+    project: Project;
+};
+type GraphProcessorArgs = Omit<GraphRunArgs, 'exposeCost' | 'project'>;
 export declare function makeCommand<T>(y: yargs.Argv<T>): yargs.Argv<T & {
     port: number;
 } & {
@@ -37,5 +47,7 @@ export declare function makeCommand<T>(y: yargs.Argv<T>): yargs.Argv<T & {
     projectFile: string | undefined;
 }>;
 export declare function serve(args: ServeArgs): Promise<void>;
+export declare function buildGraphProcessorOptions({ inputs, graph, openaiApiKey, openaiEndpoint, openaiOrganization, }: GraphProcessorArgs): NodeCreateProcessorOptions;
+export declare function buildStreamingGraphProcessorOptions(args: GraphProcessorArgs): NodeCreateProcessorOptions;
 export declare function buildStreamEventFilter(stream: string | undefined): RivetEventStreamFilterSpec;
 export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@valerypopoff/rivet2-cli",
-  "version": "2.0.9",
+  "version": "2.0.11",
   "license": "MIT",
   "types": "dist/types/cli.d.ts",
   "type": "module",
@@ -24,7 +24,7 @@
   "dependencies": {
     "@hono/node-server": "^1.13.8",
     "@types/dotenv": "^8.2.3",
-    "@valerypopoff/rivet2-node": "^2.0.9",
+    "@valerypopoff/rivet2-node": "^2.0.11",
     "chalk": "^5.4.1",
     "didyoumean2": "^7.0.4",
     "dotenv": "^16.4.7",