@assistant-ui/react 0.4.3 → 0.4.5

package/dist/edge.d.mts

@@ -1,8 +1,99 @@
-import { LanguageModelV1 } from '@ai-sdk/provider';
+import { LanguageModelV1, LanguageModelV1ToolChoice, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning } from '@ai-sdk/provider';
+import { z } from 'zod';
+import { JSONSchema7 } from 'json-schema';
 
-declare const createEdgeRuntimeAPI: ({ model }: {
+type TextContentPart = {
+    type: "text";
+    text: string;
+};
+type ImageContentPart = {
+    type: "image";
+    image: string;
+};
+type ToolCallContentPart<TArgs = unknown, TResult = unknown> = {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    argsText: string;
+    args: TArgs;
+    result?: TResult;
+    isError?: boolean;
+};
+/** Core Message Types (without UI content parts) */
+type CoreUserContentPart = TextContentPart | ImageContentPart;
+type CoreAssistantContentPart = TextContentPart | ToolCallContentPart;
+type CoreSystemMessage = {
+    role: "system";
+    content: [TextContentPart];
+};
+type CoreUserMessage = {
+    role: "user";
+    content: CoreUserContentPart[];
+};
+type CoreAssistantMessage = {
+    role: "assistant";
+    content: CoreAssistantContentPart[];
+};
+type CoreMessage = CoreSystemMessage | CoreUserMessage | CoreAssistantMessage;
+
+type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
+type Tool<TArgs = unknown, TResult = unknown> = {
+    description?: string | undefined;
+    parameters: z.ZodSchema<TArgs> | JSONSchema7;
+    execute?: ToolExecuteFunction<TArgs, TResult>;
+};
+
+declare const LanguageModelSettingsSchema: z.ZodObject<{
+    maxTokens: z.ZodOptional<z.ZodNumber>;
+    temperature: z.ZodOptional<z.ZodNumber>;
+    topP: z.ZodOptional<z.ZodNumber>;
+    presencePenalty: z.ZodOptional<z.ZodNumber>;
+    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+    headers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodOptional<z.ZodString>>>;
+}, "strip", z.ZodTypeAny, {
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    presencePenalty?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    seed?: number | undefined;
+    headers?: Record<string, string | undefined> | undefined;
+}, {
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    presencePenalty?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    seed?: number | undefined;
+    headers?: Record<string, string | undefined> | undefined;
+}>;
+type LanguageModelSettings = z.infer<typeof LanguageModelSettingsSchema>;
+type FinishResult = {
+    finishReason: LanguageModelV1FinishReason;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+    };
+    logProbs?: LanguageModelV1LogProbs | undefined;
+    messages: CoreMessage[];
+    rawCall: {
+        rawPrompt: unknown;
+        rawSettings: Record<string, unknown>;
+    };
+    warnings?: LanguageModelV1CallWarning[] | undefined;
+    rawResponse?: {
+        headers?: Record<string, string>;
+    } | undefined;
+};
+type CreateEdgeRuntimeAPIOptions = LanguageModelSettings & {
     model: LanguageModelV1;
-}) => {
+    system?: string;
+    tools?: Record<string, Tool<any, any>>;
+    toolChoice?: LanguageModelV1ToolChoice;
+    onFinish?: (result: FinishResult) => void;
+};
+declare const createEdgeRuntimeAPI: ({ model, system: serverSystem, tools: serverTools, toolChoice, onFinish, ...unsafeSettings }: CreateEdgeRuntimeAPIOptions) => {
     POST: (request: Request) => Promise<Response>;
 };
 

package/dist/edge.d.ts

@@ -1,8 +1,99 @@
-import { LanguageModelV1 } from '@ai-sdk/provider';
+import { LanguageModelV1, LanguageModelV1ToolChoice, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning } from '@ai-sdk/provider';
+import { z } from 'zod';
+import { JSONSchema7 } from 'json-schema';
 
-declare const createEdgeRuntimeAPI: ({ model }: {
+type TextContentPart = {
+    type: "text";
+    text: string;
+};
+type ImageContentPart = {
+    type: "image";
+    image: string;
+};
+type ToolCallContentPart<TArgs = unknown, TResult = unknown> = {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    argsText: string;
+    args: TArgs;
+    result?: TResult;
+    isError?: boolean;
+};
+/** Core Message Types (without UI content parts) */
+type CoreUserContentPart = TextContentPart | ImageContentPart;
+type CoreAssistantContentPart = TextContentPart | ToolCallContentPart;
+type CoreSystemMessage = {
+    role: "system";
+    content: [TextContentPart];
+};
+type CoreUserMessage = {
+    role: "user";
+    content: CoreUserContentPart[];
+};
+type CoreAssistantMessage = {
+    role: "assistant";
+    content: CoreAssistantContentPart[];
+};
+type CoreMessage = CoreSystemMessage | CoreUserMessage | CoreAssistantMessage;
+
+type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
+type Tool<TArgs = unknown, TResult = unknown> = {
+    description?: string | undefined;
+    parameters: z.ZodSchema<TArgs> | JSONSchema7;
+    execute?: ToolExecuteFunction<TArgs, TResult>;
+};
+
+declare const LanguageModelSettingsSchema: z.ZodObject<{
+    maxTokens: z.ZodOptional<z.ZodNumber>;
+    temperature: z.ZodOptional<z.ZodNumber>;
+    topP: z.ZodOptional<z.ZodNumber>;
+    presencePenalty: z.ZodOptional<z.ZodNumber>;
+    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+    headers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodOptional<z.ZodString>>>;
+}, "strip", z.ZodTypeAny, {
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    presencePenalty?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    seed?: number | undefined;
+    headers?: Record<string, string | undefined> | undefined;
+}, {
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    presencePenalty?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    seed?: number | undefined;
+    headers?: Record<string, string | undefined> | undefined;
+}>;
+type LanguageModelSettings = z.infer<typeof LanguageModelSettingsSchema>;
+type FinishResult = {
+    finishReason: LanguageModelV1FinishReason;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+    };
+    logProbs?: LanguageModelV1LogProbs | undefined;
+    messages: CoreMessage[];
+    rawCall: {
+        rawPrompt: unknown;
+        rawSettings: Record<string, unknown>;
+    };
+    warnings?: LanguageModelV1CallWarning[] | undefined;
+    rawResponse?: {
+        headers?: Record<string, string>;
+    } | undefined;
+};
+type CreateEdgeRuntimeAPIOptions = LanguageModelSettings & {
     model: LanguageModelV1;
-}) => {
+    system?: string;
+    tools?: Record<string, Tool<any, any>>;
+    toolChoice?: LanguageModelV1ToolChoice;
+    onFinish?: (result: FinishResult) => void;
+};
+declare const createEdgeRuntimeAPI: ({ model, system: serverSystem, tools: serverTools, toolChoice, onFinish, ...unsafeSettings }: CreateEdgeRuntimeAPIOptions) => {
     POST: (request: Request) => Promise<Response>;
 };