@llumiverse/core 0.15.0 → 0.17.0

package/lib/types/async.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"async.d.ts","sourceRoot":"","sources":["../../src/async.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAExD,wBAAuB,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC,6CAI9G;AAED,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAM9D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,cAAc,CAAC,eAAe,CAAC,EAAE,SAAS,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,MAAM,CAAC,CAc/J;AAED,qBAAa,WAAW,CAAC,CAAC,EAAE,OAAO,GAAG,GAAG,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC;IAElE,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAAC,CAGd;IACF,OAAO,CAAC,IAAI,CAAS;IAGrB,IAAI,CAAC,KAAK,EAAE,CAAC;IAYb;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,EAAE,OAAO;IAQrB,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CA0BlE;AAID;;;;IAII;AACJ,wBAAuB,sBAAsB,CAAC,CAAC,EAAE,CAAC,EAC9C,iBAAiB,EAAE,aAAa,CAAC,CAAC,CAAC,EACnC,SAAS,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EACvC,YAAY,CAAC,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACpC,aAAa,CAAC,CAAC,CAAC,CAOlB"}
+{"version":3,"file":"async.d.ts","sourceRoot":"","sources":["../../src/async.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE7C,wBAAuB,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC,6CAI9G;AAED,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAM9D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,cAAc,CAAC,eAAe,CAAC,EAAE,SAAS,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,eAAe,GAAG,cAAc,CAAC,eAAe,CAAC,GAAG,aAAa,CAAC,eAAe,CAAC,CAe1L;AAED,qBAAa,WAAW,CAAC,CAAC,EAAE,OAAO,GAAG,GAAG,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC;IAElE,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAAC,CAGd;IACF,OAAO,CAAC,IAAI,CAAS;IAGrB,IAAI,CAAC,KAAK,EAAE,CAAC;IAYb;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,EAAE,OAAO;IAQrB,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CA0BlE;AAID;;;;IAII;AACJ,wBAAuB,sBAAsB,CAAC,CAAC,EAAE,CAAC,EAC9C,iBAAiB,EAAE,aAAa,CAAC,CAAC,CAAC,EACnC,SAAS,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EACvC,YAAY,CAAC,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACpC,aAAa,CAAC,CAAC,CAAC,CAOlB"}

package/lib/types/formatters/commons.d.ts

@@ -1,3 +1,3 @@
-import { JSONSchema4 } from "json-schema";
-export declare function getJSONSafetyNotice(schema: JSONSchema4): string;
+import { JSONSchema } from "../types.js";
+export declare function getJSONSafetyNotice(schema: JSONSchema): string;
 //# sourceMappingURL=commons.d.ts.map

package/lib/types/formatters/commons.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"commons.d.ts","sourceRoot":"","sources":["../../../src/formatters/commons.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,WAAW,UAEtD"}
+{"version":3,"file":"commons.d.ts","sourceRoot":"","sources":["../../../src/formatters/commons.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,UAAU,UAErD"}

package/lib/types/formatters/generic.d.ts

@@ -1,4 +1,4 @@
-import { JSONSchema4 } from "json-schema";
+import { JSONSchema } from "../types.js";
 import { PromptSegment } from "../types.js";
 interface Labels {
     user: string;
@@ -7,7 +7,7 @@ interface Labels {
     safety: string;
     instruction: string;
 }
-export declare function createTextPromptFormatter(labels?: Labels): (segments: PromptSegment[], schema?: JSONSchema4) => string;
-declare const formatTextPrompt: (segments: PromptSegment[], schema?: JSONSchema4) => string;
+export declare function createTextPromptFormatter(labels?: Labels): (segments: PromptSegment[], schema?: JSONSchema) => string;
+declare const formatTextPrompt: (segments: PromptSegment[], schema?: JSONSchema) => string;
 export { formatTextPrompt };
 //# sourceMappingURL=generic.d.ts.map

package/lib/types/formatters/generic.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generic.d.ts","sourceRoot":"","sources":["../../../src/formatters/generic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAc,aAAa,EAAE,MAAM,aAAa,CAAC;AAGxD,UAAU,MAAM;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAA;CACtB;AAED,wBAAgB,yBAAyB,CAAC,MAAM,GAAE,MAMjD,cAC+C,aAAa,EAAE,WAAW,WAAW,KAAG,MAAM,CA2C7F;AAED,QAAA,MAAM,gBAAgB,aA7C0B,aAAa,EAAE,WAAW,WAAW,KAAG,MA6CpC,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,CAAC"}
+{"version":3,"file":"generic.d.ts","sourceRoot":"","sources":["../../../src/formatters/generic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAc,aAAa,EAAE,MAAM,aAAa,CAAC;AAGxD,UAAU,MAAM;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAA;CACtB;AAED,wBAAgB,yBAAyB,CAAC,MAAM,GAAE,MAMjD,IACqC,UAAU,aAAa,EAAE,EAAE,SAAS,UAAU,KAAG,MAAM,CA2C5F;AAED,QAAA,MAAM,gBAAgB,aA7C0B,aAAa,EAAE,WAAW,UAAU,KAAG,MA6CnC,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,CAAC"}

package/lib/types/formatters/index.d.ts

@@ -1,10 +1,8 @@
-import { JSONSchema4 } from "json-schema";
+import { JSONSchema } from "../types.js";
 import { PromptSegment } from "../types.js";
-export type PromptFormatter<T = any> = (messages: PromptSegment[], schema?: JSONSchema4) => T;
-export * from "./claude.js";
+export type PromptFormatter<T = any> = (messages: PromptSegment[], schema?: JSONSchema) => T;
 export * from "./commons.js";
 export * from "./generic.js";
-export * from "./llama2.js";
-export * from "./llama3.js";
 export * from "./openai.js";
+export * from "./nova.js";
 //# sourceMappingURL=index.d.ts.map

package/lib/types/formatters/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/formatters/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,MAAM,eAAe,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,CAAC,CAAC;AAE9F,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/formatters/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,MAAM,eAAe,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,MAAM,CAAC,EAAE,UAAU,KAAK,CAAC,CAAC;AAE7F,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC"}

package/lib/types/formatters/nova.d.ts

@@ -0,0 +1,40 @@
+import { JSONSchema } from "../types.js";
+import { PromptSegment } from "../index.js";
+export interface NovaMessage {
+    role: 'user' | 'assistant';
+    content: NovaMessagePart[];
+}
+export interface NovaSystemMessage {
+    text: string;
+}
+interface NovaMessagePart {
+    text?: string;
+    image?: {
+        format: "jpeg" | "png" | "gif" | "webp";
+        source: {
+            bytes: string;
+        };
+    };
+    video?: {
+        format: "mkv" | "mov" | "mp4" | "webm" | "three_gp" | "flv" | "mpeg" | "mpg" | "wmv";
+        source: {
+            s3Location?: {
+                uri: string;
+                bucketOwner: string;
+            };
+            bytes?: string;
+        };
+    };
+}
+export interface NovaMessagesPrompt {
+    system?: NovaSystemMessage[];
+    messages: NovaMessage[];
+    negative?: string;
+    mask?: string;
+}
+/**
+ * A formatter used by Bedrock to format prompts for nova related models
+ */
+export declare function formatNovaPrompt(segments: PromptSegment[], schema?: JSONSchema): Promise<NovaMessagesPrompt>;
+export {};
+//# sourceMappingURL=nova.d.ts.map

package/lib/types/formatters/nova.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"nova.d.ts","sourceRoot":"","sources":["../../../src/formatters/nova.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAc,aAAa,EAAsB,MAAM,aAAa,CAAC;AAI5E,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,eAAe,EAAE,CAAA;CAC7B;AAED,MAAM,WAAW,iBAAiB;IAC9B,IAAI,EAAE,MAAM,CAAA;CACf;AAED,UAAU,eAAe;IACrB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,KAAK,CAAC,EAAE;QACJ,MAAM,EAAE,MAAM,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;QACxC,MAAM,EAAE;YACJ,KAAK,EAAE,MAAM,CAAA;SAChB,CAAA;KACJ,CAAA;IACD,KAAK,CAAC,EAAE;QACJ,MAAM,EAAE,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,UAAU,GAAG,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,KAAK,CAAC;QACrF,MAAM,EAAE;YAEJ,UAAU,CAAC,EAAE;gBACT,GAAG,EAAE,MAAM,CAAC;gBACZ,WAAW,EAAE,MAAM,CAAA;aACtB,CAAA;YAED,KAAK,CAAC,EAAE,MAAM,CAAA;SACjB,CAAA;KACJ,CAAA;CACJ;AAED,MAAM,WAAW,kBAAkB;IAC/B,MAAM,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAC7B,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AAEH,wBAAsB,gBAAgB,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,MAAM,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA6FlH"}

package/lib/types/formatters/openai.d.ts

@@ -1,3 +1,4 @@
+import { PromptOptions } from "../index.js";
 import { PromptSegment } from "../types.js";
 export interface OpenAITextMessage {
     content: string;
@@ -8,6 +9,12 @@ export interface OpenAIMessage {
     role: "system" | "user" | "assistant";
     name?: string;
 }
+export interface OpenAIToolMessage {
+    role: "tool";
+    tool_call_id: string;
+    content: string;
+}
+export type OpenAIInputMessage = OpenAIMessage | OpenAIToolMessage;
 export interface OpenAIContentPartText {
     type: "text";
     text: string;
@@ -25,5 +32,10 @@ export interface OpenAIContentPartImage {
  * @returns
  */
 export declare function formatOpenAILikeTextPrompt(segments: PromptSegment[]): OpenAITextMessage[];
-export declare function formatOpenAILikeMultimodalPrompt(segments: PromptSegment[]): Promise<OpenAIMessage[]>;
+export declare function formatOpenAILikeMultimodalPrompt(segments: PromptSegment[], opts: PromptOptions & OpenAIPromptFormatterOptions): Promise<OpenAIInputMessage[]>;
+export interface OpenAIPromptFormatterOptions {
+    multimodal?: boolean;
+    useToolForFormatting?: boolean;
+    schema?: Object;
+}
 //# sourceMappingURL=openai.d.ts.map

package/lib/types/formatters/openai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/formatters/openai.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG5C,MAAM,WAAW,iBAAiB;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;CACzC;AAED,MAAM,WAAW,aAAa;IAC1B,OAAO,EAAE,CAAC,qBAAqB,GAAG,sBAAsB,CAAC,EAAE,CAAA;IAC3D,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,sBAAsB;IACnC,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE;QACP,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,GAAG,MAAM,CAAA;QAChC,GAAG,EAAE,MAAM,CAAA;KACd,CAAA;CACJ;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,iBAAiB,EAAE,CAoBzF;AAED,wBAAsB,gCAAgC,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC,CA8D1G"}
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/formatters/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,aAAa,EAAE,MAAM,aAAa,CAAC;AAExD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG5C,MAAM,WAAW,iBAAiB;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;CACzC;AAED,MAAM,WAAW,aAAa;IAC1B,OAAO,EAAE,CAAC,qBAAqB,GAAG,sBAAsB,CAAC,EAAE,CAAA;IAC3D,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,IAAI,CAAC,EAAE,MAAM,CAAC;CACjB;AACD,MAAM,WAAW,iBAAiB;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CACnB;AACD,MAAM,MAAM,kBAAkB,GAAG,aAAa,GAAG,iBAAiB,CAAC;AAEnE,MAAM,WAAW,qBAAqB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,sBAAsB;IACnC,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE;QACP,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,GAAG,MAAM,CAAA;QAChC,GAAG,EAAE,MAAM,CAAA;KACd,CAAA;CACJ;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,iBAAiB,EAAE,CAoBzF;AAGD,wBAAsB,gCAAgC,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,aAAa,GAAG,4BAA4B,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAyFnK;AAED,MAAM,WAAW,4BAA4B;IACzC,UAAU,CAAC,EAAE,OAAO,CAAA;IACpB,oBAAoB,CAAC,EAAE,OAAO,CAAA;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAA;CAClB"}

package/lib/types/index.d.ts

@@ -2,4 +2,5 @@ export * from "./Driver.js";
 export * from "./json.js";
 export * from "./stream.js";
 export * from "./types.js";
+export * from "./options.js";
 //# sourceMappingURL=index.d.ts.map

package/lib/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC;AAC3B,cAAc,cAAc,CAAC"}

package/lib/types/options/bedrock.d.ts

@@ -0,0 +1,32 @@
+import { ModelOptionsInfo, ModelOptions } from "../types.js";
+export type BedrockOptions = NovaCanvasOptions | BaseConverseOptions | BedrockClaudeOptions;
+export interface NovaCanvasOptions {
+    _option_id: "bedrock-nova-canvas";
+    taskType: "TEXT_IMAGE" | "TEXT_IMAGE_WITH_IMAGE_CONDITIONING" | "COLOR_GUIDED_GENERATION" | "IMAGE_VARIATION" | "INPAINTING" | "OUTPAINTING" | "BACKGROUND_REMOVAL";
+    width?: number;
+    height?: number;
+    quality?: "standard" | "premium";
+    cfgScale?: number;
+    seed?: number;
+    numberOfImages?: number;
+    controlMode?: "CANNY_EDGE" | "SEGMENTATION";
+    controlStrength?: number;
+    colors?: string[];
+    similarityStrength?: number;
+    outPaintingMode?: "DEFAULT" | "PRECISE";
+}
+export interface BaseConverseOptions {
+    _option_id: "bedrock-converse" | "bedrock-claude" | "bedrock-nova" | "bedrock-mistral" | "bedrock-ai21" | "bedrock-cohere-command";
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    stop_sequence?: string[];
+}
+export interface BedrockClaudeOptions extends BaseConverseOptions {
+    _option_id: "bedrock-claude";
+    top_k?: number;
+    thinking_mode?: boolean;
+    thinking_budget_tokens?: number;
+}
+export declare function getBedrockOptions(model: string, option?: ModelOptions): ModelOptionsInfo;
+//# sourceMappingURL=bedrock.d.ts.map

package/lib/types/options/bedrock.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/options/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAmC,MAAM,aAAa,CAAC;AAI9F,MAAM,MAAM,cAAc,GAAG,iBAAiB,GAAG,mBAAmB,GAAG,oBAAoB,CAAC;AAE5F,MAAM,WAAW,iBAAiB;IAC9B,UAAU,EAAE,qBAAqB,CAAA;IACjC,QAAQ,EAAE,YAAY,GAAG,oCAAoC,GAAG,yBAAyB,GAAG,iBAAiB,GAAG,YAAY,GAAG,aAAa,GAAG,oBAAoB,CAAC;IACpK,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,UAAU,GAAG,SAAS,CAAC;IACjC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,YAAY,GAAG,cAAc,CAAC;IAC5C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;CAC3C;AAED,MAAM,WAAW,mBAAmB;IAChC,UAAU,EAAE,kBAAkB,GAAG,gBAAgB,GAAG,cAAc,GAAG,iBAAiB,GAAG,cAAc,GAAG,wBAAwB,CAAC;IACnI,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,oBAAqB,SAAQ,mBAAmB;IAC7D,UAAU,EAAE,gBAAgB,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACnC;AA4ED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,GAAG,gBAAgB,CAoRxF"}

package/lib/types/options/groq.d.ts

@@ -0,0 +1,12 @@
+import { ModelOptionsInfo, ModelOptions } from "../types.js";
+export type GroqOptions = GroqDeepseekThinkingOptions;
+export interface GroqDeepseekThinkingOptions {
+    _option_id: "groq-deepseek-thinking";
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    stop_sequence?: string[];
+    reasonsing_format: 'parsed' | 'raw' | 'hidden';
+}
+export declare function getGroqOptions(model: string, _option?: ModelOptions): ModelOptionsInfo;
+//# sourceMappingURL=groq.d.ts.map

package/lib/types/options/groq.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"groq.d.ts","sourceRoot":"","sources":["../../../src/options/groq.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAuB,YAAY,EAA6B,MAAM,aAAa,CAAC;AAI7G,MAAM,MAAM,WAAW,GAAG,2BAA2B,CAAC;AAEtD,MAAM,WAAW,2BAA2B;IACxC,UAAU,EAAE,wBAAwB,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,iBAAiB,EAAE,QAAQ,GAAG,KAAK,GAAG,QAAQ,CAAC;CAClD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,gBAAgB,CA+BtF"}

package/lib/types/options/openai.d.ts

@@ -0,0 +1,21 @@
+import { ModelOptionsInfo, ModelOptions } from "../types.js";
+export type OpenAiOptions = OpenAiThinkingOptions | OpenAiTextOptions;
+export interface OpenAiThinkingOptions {
+    _option_id: "openai-thinking";
+    max_tokens?: number;
+    stop_sequence?: string[];
+    reasoning_effort?: "low" | "medium" | "high";
+    image_detail?: "low" | "high" | "auto";
+}
+export interface OpenAiTextOptions {
+    _option_id: "openai-text";
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    presence_penalty?: number;
+    frequency_penalty?: number;
+    stop_sequence?: string[];
+    image_detail?: "low" | "high" | "auto";
+}
+export declare function getOpenAiOptions(model: string, _option?: ModelOptions): ModelOptionsInfo;
+//# sourceMappingURL=openai.d.ts.map

package/lib/types/options/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/options/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAuB,YAAY,EAA6B,MAAM,aAAa,CAAC;AAI7G,MAAM,MAAM,aAAa,GAAG,qBAAqB,GAAG,iBAAiB,CAAC;AAEtE,MAAM,WAAW,qBAAqB;IAClC,UAAU,EAAE,iBAAiB,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;CAC1C;AAED,MAAM,WAAW,iBAAiB;IAC9B,UAAU,EAAE,aAAa,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;CAC1C;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,gBAAgB,CA4GxF"}

package/lib/types/options/vertexai.d.ts

@@ -0,0 +1,52 @@
+import { ModelOptionsInfo, ModelOptions } from "../types.js";
+export type VertexAIOptions = ImagenOptions | VertexAIClaudeOptions;
+export declare enum ImagenTaskType {
+    TEXT_IMAGE = "TEXT_IMAGE",
+    EDIT_MODE_INPAINT_REMOVAL = "EDIT_MODE_INPAINT_REMOVAL",
+    EDIT_MODE_INPAINT_INSERTION = "EDIT_MODE_INPAINT_INSERTION",
+    EDIT_MODE_BGSWAP = "EDIT_MODE_BGSWAP",
+    EDIT_MODE_OUTPAINT = "EDIT_MODE_OUTPAINT",
+    CUSTOMIZATION_SUBJECT = "CUSTOMIZATION_SUBJECT",
+    CUSTOMIZATION_STYLE = "CUSTOMIZATION_STYLE",
+    CUSTOMIZATION_CONTROLLED = "CUSTOMIZATION_CONTROLLED",
+    CUSTOMIZATION_INSTRUCT = "CUSTOMIZATION_INSTRUCT"
+}
+export declare enum ImagenMaskMode {
+    MASK_MODE_USER_PROVIDED = "MASK_MODE_USER_PROVIDED",
+    MASK_MODE_BACKGROUND = "MASK_MODE_BACKGROUND",
+    MASK_MODE_FOREGROUND = "MASK_MODE_FOREGROUND",
+    MASK_MODE_SEMANTIC = "MASK_MODE_SEMANTIC"
+}
+export interface ImagenOptions {
+    _option_id: "vertexai-imagen";
+    number_of_images?: number;
+    seed?: number;
+    person_generation?: "dont_allow" | "allow_adults" | "allow_all";
+    safety_setting?: "block_none" | "block_only_high" | "block_medium_and_above" | "block_low_and_above";
+    image_file_type?: "image/jpeg" | "image/png";
+    jpeg_compression_quality?: number;
+    aspect_ratio?: "1:1" | "4:3" | "3:4" | "16:9" | "9:16";
+    add_watermark?: boolean;
+    enhance_prompt?: boolean;
+    edit_mode?: ImagenTaskType;
+    guidance_scale?: number;
+    edit_steps?: number;
+    mask_mode?: ImagenMaskMode;
+    mask_dilation?: number;
+    mask_class?: number[];
+    controlType: "CONTROL_TYPE_FACE_MESH" | "CONTROL_TYPE_CANNY" | "CONTROL_TYPE_SCRIBBLE";
+    controlImageComputation?: boolean;
+    subjectType: "SUBJECT_TYPE_PERSON" | "SUBJECT_TYPE_ANIMAL" | "SUBJECT_TYPE_PRODUCT" | "SUBJECT_TYPE_DEFAULT";
+}
+export interface VertexAIClaudeOptions {
+    _option_id: "vertexai-claude";
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    top_k?: number;
+    stop_sequence?: string[];
+    thinking_mode?: boolean;
+    thinking_budget_tokens?: number;
+}
+export declare function getVertexAiOptions(model: string, option?: ModelOptions): ModelOptionsInfo;
+//# sourceMappingURL=vertexai.d.ts.map

package/lib/types/options/vertexai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertexai.d.ts","sourceRoot":"","sources":["../../../src/options/vertexai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAkD,YAAY,EAAE,MAAM,aAAa,CAAC;AAI7G,MAAM,MAAM,eAAe,GAAG,aAAa,GAAG,qBAAqB,CAAC;AAEpE,oBAAY,cAAc;IACtB,UAAU,eAAe;IACzB,yBAAyB,8BAA8B;IACvD,2BAA2B,gCAAgC;IAC3D,gBAAgB,qBAAqB;IACrC,kBAAkB,uBAAuB;IACzC,qBAAqB,0BAA0B;IAC/C,mBAAmB,wBAAwB;IAC3C,wBAAwB,6BAA6B;IACrD,sBAAsB,2BAA2B;CACpD;AAED,oBAAY,cAAc;IACtB,uBAAuB,4BAA4B;IACnD,oBAAoB,yBAAyB;IAC7C,oBAAoB,yBAAyB;IAC7C,kBAAkB,uBAAuB;CAC5C;AAED,MAAM,WAAW,aAAa;IAC1B,UAAU,EAAE,iBAAiB,CAAA;IAG7B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,YAAY,GAAG,cAAc,GAAG,WAAW,CAAC;IAChE,cAAc,CAAC,EAAE,YAAY,GAAG,iBAAiB,GAAG,wBAAwB,GAAG,qBAAqB,CAAC;IACrG,eAAe,CAAC,EAAE,YAAY,GAAG,WAAW,CAAC;IAC7C,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,YAAY,CAAC,EAAE,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;IACvD,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,OAAO,CAAC;IAGzB,SAAS,CAAC,EAAE,cAAc,CAAA;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,cAAc,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IAGtB,WAAW,EAAE,wBAAwB,GAAG,oBAAoB,GAAG,uBAAuB,CAAC;IACvF,uBAAuB,CAAC,EAAE,OAAO,CAAC;IAClC,WAAW,EAAE,qBAAqB,GAAG,qBAAqB,GAAG,sBAAsB,GAAG,sBAAsB,CAAC;CAChH;AAED,MAAM,WAAW,qBAAqB;IAClC,UAAU,EAAE,iBAAiB,CAAA;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACnC;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,GAAG,gBAAgB,CA8NzF"}

package/lib/types/options.d.ts

@@ -0,0 +1,14 @@
+import { ModelOptions, ModelOptionsInfo } from "./types.js";
+export interface TextFallbackOptions {
+    _option_id: "text-fallback";
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    top_k?: number;
+    presence_penalty?: number;
+    frequency_penalty?: number;
+    stop_sequence?: string[];
+}
+export declare const textOptionsFallback: ModelOptionsInfo;
+export declare function getOptions(provider?: string, model?: string, options?: ModelOptions): ModelOptionsInfo;
+//# sourceMappingURL=options.d.ts.map

package/lib/types/options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.d.ts","sourceRoot":"","sources":["../../src/options.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAA6B,MAAM,YAAY,CAAC;AAMvF,MAAM,WAAW,mBAAmB;IAChC,UAAU,EAAE,eAAe,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,eAAO,MAAM,mBAAmB,EAAE,gBA6BjC,CAAC;AAEF,wBAAgB,UAAU,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,gBAAgB,CAatG"}

package/lib/types/types.d.ts

@@ -1,20 +1,22 @@
-import { JSONSchema4 } from "json-schema";
-import { PromptFormatter } from "./formatters/index.js";
-import { JSONObject } from "./json.js";
+import { PromptFormatter } from './formatters/index.js';
+import { JSONObject } from './json.js';
+import { TextFallbackOptions } from './options.js';
+import { BedrockOptions } from './options/bedrock.js';
+import { OpenAiOptions } from './options/openai.js';
+import { VertexAIOptions } from './options/vertexai.js';
 export interface EmbeddingsOptions {
     /**
-     * The content to generate the embeddings for. Required.
+     * The text to generate the embeddings for. One of text or image is required.
      */
-    content: string;
+    text?: string;
     /**
-     * The model to use to generate the embeddings. Optional.
+     * The image to generate embeddings for
      */
-    model?: string;
+    image?: string;
     /**
-     * Additional options for the embeddings generation. Optional.
-     * The supported properties depends on the target implementation.
+     * The model to use to generate the embeddings. Optional.
      */
-    [key: string]: any;
+    model?: string;
 }
 export interface EmbeddingsResult {
     /**
@@ -24,28 +26,52 @@ export interface EmbeddingsResult {
     /**
      * The model used to hgenerate the embeddings.
      */
-    model?: string;
+    model: string;
     /**
      * Number of tokens of the input text.
      */
     token_count?: number;
-    /**
-     * Additional properties. Depends on the target implementation.
-     */
-    [key: string]: any;
 }
 export interface ResultValidationError {
-    code: 'validation_error' | 'json_error';
+    code: 'validation_error' | 'json_error' | 'content_policy_violation';
     message: string;
     data?: string;
 }
+export interface CompletionChunkObject<ResultT = any> {
+    result: ResultT;
+    token_usage?: ExecutionTokenUsage;
+    finish_reason?: "stop" | "length" | string;
+}
+export type CompletionChunk = CompletionChunkObject | string;
+export interface ToolDefinition {
+    name: string;
+    description?: string;
+    input_schema: {
+        type: 'object';
+        properties?: unknown | null | undefined;
+        [k: string]: unknown;
+    };
+}
+/**
+ * A tool use instance represents a call to a tool.
+ * The id property is used to identify the tool call.
+ */
+export interface ToolUse<ParamsT = JSONObject> {
+    id: string;
+    tool_name: string;
+    tool_input: ParamsT | null;
+}
 export interface Completion<ResultT = any> {
     result: ResultT;
     token_usage?: ExecutionTokenUsage;
+    /**
+     * Contains the tools from which the model awaits information.
+     */
+    tool_use?: ToolUse[];
     /**
      * The finish reason as reported by the model: stop | length or other model specific values
      */
-    finish_reason?: "stop" | "length" | string;
+    finish_reason?: "stop" | "length" | "tool_use" | string;
     /**
      * Set only if a result validation error occured, otherwise if the result is valid the error field is undefined
      * This can only be set if the result_schema is set and the reuslt could not be parsed as a json or if the result does not match the schema
@@ -55,6 +81,13 @@ export interface Completion<ResultT = any> {
      * The original response. Only included if the option include_original_response is set to true and the request is made using execute. Not supported when streaming.
      */
     original_response?: Record<string, any>;
+    /**
+     * The conversation context. This is an opaque structure that can be passed to the next request to restore the context.
+     */
+    conversation?: unknown;
+}
+export interface ImageGeneration {
+    images?: string[];
 }
 export interface ExecutionResponse<PromptT = any> extends Completion {
     prompt: PromptT;
@@ -62,6 +95,10 @@ export interface ExecutionResponse<PromptT = any> extends Completion {
      * The time it took to execute the request in seconds
      */
     execution_time?: number;
+    /**
+     * The number of chunks for streamed executions
+     */
+    chunks?: number;
 }
 export interface CompletionStream<PromptT = any> extends AsyncIterable<string> {
     completion: ExecutionResponse<PromptT> | undefined;
@@ -75,6 +112,14 @@ export interface Logger {
 export interface DriverOptions {
     logger?: Logger | "console";
 }
+export type JSONSchema4TypeName = "string" | "number" | "integer" | "boolean" | "object" | "array" | "null" | "any";
+export interface JSONSchema {
+    type?: JSONSchema4TypeName | JSONSchema4TypeName[];
+    description?: string;
+    properties?: Record<string, JSONSchema>;
+    required?: string[];
+    [k: string]: any;
+}
 export interface PromptOptions {
     model: string;
     /**
@@ -82,56 +127,103 @@ export interface PromptOptions {
      * If no one is specified the driver will choose a formatter compatible with the target model
      */
     format?: PromptFormatter;
-    result_schema?: JSONSchema4;
+    result_schema?: JSONSchema;
 }
-export interface ExecutionOptions extends PromptOptions {
-    temperature?: number;
-    max_tokens?: number;
-    stop_sequence?: string | string[];
-    /**
-     * restricts the selection of tokens to the “k” most likely options, based on their probabilities
-     * Lower values make the model more deterministic, more focused. Examples:
-     * - 10 - result will be highly controlled anc contextually relevant
-     * - 50 - result will be more creative but maintaining a balance between control and creativity
-     * - 100 - will lead to more creative and less predictable outputs
-     * It will be ignored on OpenAI since it does not support it
-     */
-    top_k?: number;
-    /**
-     * An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered.
-     * Either use temperature or top_p, not both
-     */
-    top_p?: number;
-    /**
-     * Only supported for OpenAI. Look at OpenAI documentation for more detailsx
-     */
-    top_logprobs?: number;
-    /**
-     * Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far, increasing the model's likelihood to talk about new topics.
-     * Ignored for models which doesn;t support it
-     */
-    presence_penalty?: number;
-    /**
-     * Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.
-     * Ignored for models which doesn;t support it
-     */
-    frequency_penalty?: number;
+export interface StatelessExecutionOptions extends PromptOptions {
     /**
      * If set to true the original response from the target LLM will be included in the response under the original_response field.
      * This is useful for debugging and for some advanced use cases.
      * It is ignored on streaming requests
      */
     include_original_response?: boolean;
+    model_options?: ModelOptions;
+    output_modality: Modalities;
+}
+export interface ExecutionOptions extends StatelessExecutionOptions {
+    /**
+     * Available tools for the request
+     */
+    tools?: ToolDefinition[];
+    /**
+     * This is an opaque structure that provides a conversation context
+     * Each driver implementation will return a conversation property in the execution response
+     * that can be passed here to restore the context when a new prompt is sent to the model.
+     */
+    conversation?: unknown | null;
+}
+export declare enum SharedOptions {
+    max_tokens = "max_tokens",
+    temperature = "temperature",
+    top_p = "top_p",
+    top_k = "top_k",
+    presence_penalty = "presence_penalty",
+    frequency_penalty = "frequency_penalty",
+    stop_sequence = "stop_sequence",
+    seed = "seed",
+    number_of_images = "number_of_images"
+}
+export declare enum OptionType {
+    numeric = "numeric",
+    enum = "enum",
+    boolean = "boolean",
+    string_list = "string_list"
+}
+export type ModelOptions = TextFallbackOptions | VertexAIOptions | BedrockOptions | OpenAiOptions;
+export interface ModelOptionsInfo {
+    options: ModelOptionInfoItem[];
+    _option_id: string;
+}
+export type ModelOptionInfoItem = NumericOptionInfo | EnumOptionInfo | BooleanOptionInfo | StringListOptionInfo;
+interface OptionInfoPrototype {
+    type: OptionType;
+    name: string;
+    description?: string;
+    refresh?: boolean;
+}
+export interface NumericOptionInfo extends OptionInfoPrototype {
+    type: OptionType.numeric;
+    value?: number;
+    min?: number;
+    max?: number;
+    step?: number;
+    integer?: boolean;
+    default?: number;
+}
+export interface EnumOptionInfo extends OptionInfoPrototype {
+    type: OptionType.enum;
+    value?: string;
+    enum: Record<string, string>;
+    default?: string;
+}
+export interface BooleanOptionInfo extends OptionInfoPrototype {
+    type: OptionType.boolean;
+    value?: boolean;
+    default?: boolean;
+}
+export interface StringListOptionInfo extends OptionInfoPrototype {
+    type: OptionType.string_list;
+    value?: string[];
+    default?: string[];
 }
 export declare enum PromptRole {
     safety = "safety",
     system = "system",
     user = "user",
-    assistant = "assistant"
+    assistant = "assistant",
+    negative = "negative",
+    mask = "mask",
+    /**
+     * Used to send the response of a tool
+     */
+    tool = "tool"
 }
 export interface PromptSegment {
     role: PromptRole;
     content: string;
+    /**
+     * The tool use id if the segment is a tool response
+     */
+    tool_use_id?: string;
     files?: DataSource[];
 }
 export interface ExecutionTokenUsage {
@@ -139,6 +231,10 @@ export interface ExecutionTokenUsage {
     result?: number;
     total?: number;
 }
+export declare enum Modalities {
+    text = "text",
+    image = "image"
+}
 export interface AIModel<ProviderKeys = string> {
     id: string;
     name: string;
@@ -152,6 +248,8 @@ export interface AIModel<ProviderKeys = string> {
     can_stream?: boolean;
     is_custom?: boolean;
     is_multimodal?: boolean;
+    input_modalities?: string[];
+    environment?: string;
 }
 export declare enum AIModelStatus {
     Available = "available",
@@ -207,7 +305,7 @@ export interface TrainingPromptOptions {
     segments: PromptSegment[];
     completion: string | JSONObject;
     model: string;
-    schema?: JSONSchema4;
+    schema?: JSONSchema;
 }
 export declare enum TrainingJobStatus {
     running = "running",
@@ -221,4 +319,5 @@ export interface TrainingJob {
     details?: string;
     model?: string;
 }
+export {};
 //# sourceMappingURL=types.d.ts.map