@sqlrooms/ai 0.19.2 → 0.21.0

package/dist/schemas.d.ts

@@ -13,304 +13,231 @@ export declare const QueryToolParameters: z.ZodObject<{
     reasoning: string;
 }>;
 export type QueryToolParameters = z.infer<typeof QueryToolParameters>;
-export declare const AnalysisSchema: z.ZodString;
-export type AnalysisSchema = z.infer<typeof AnalysisSchema>;
-export declare const ToolCallSchema: z.ZodObject<{
-    toolName: z.ZodString;
-    toolCallId: z.ZodString;
-    args: z.ZodObject<{
-        type: z.ZodLiteral<"query">;
-        sqlQuery: z.ZodString;
-        reasoning: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        sqlQuery: string;
-        type: "query";
-        reasoning: string;
-    }, {
-        sqlQuery: string;
-        type: "query";
-        reasoning: string;
-    }>;
-}, "strip", z.ZodTypeAny, {
-    toolName: string;
-    toolCallId: string;
-    args: {
-        sqlQuery: string;
-        type: "query";
-        reasoning: string;
-    };
-}, {
-    toolName: string;
-    toolCallId: string;
-    args: {
-        sqlQuery: string;
-        type: "query";
-        reasoning: string;
-    };
-}>;
-export type ToolCallSchema = z.infer<typeof ToolCallSchema>;
-export declare const QueryResultElementSchema: z.ZodObject<{
-    type: z.ZodLiteral<"query-result">;
-    title: z.ZodString;
-    sqlQuery: z.ZodString;
+export declare const ErrorMessageSchema: z.ZodObject<{
+    error: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    title: string;
-    sqlQuery: string;
-    type: "query-result";
+    error: string;
 }, {
-    title: string;
-    sqlQuery: string;
-    type: "query-result";
+    error: string;
 }>;
-export type QueryResultElementSchema = z.infer<typeof QueryResultElementSchema>;
-export declare const ElementSchema: z.ZodUnion<[z.ZodObject<{
-    type: z.ZodLiteral<"query-result">;
-    title: z.ZodString;
-    sqlQuery: z.ZodString;
-}, "strip", z.ZodTypeAny, {
-    title: string;
-    sqlQuery: string;
-    type: "query-result";
-}, {
-    title: string;
-    sqlQuery: string;
-    type: "query-result";
-}>, z.ZodString]>;
-export type ElementSchema = z.infer<typeof ElementSchema>;
-export declare const ToolCallMessageSchema: z.ZodObject<{
-    toolCallId: z.ZodString;
-    element: z.ZodUnion<[z.ZodObject<{
-        type: z.ZodLiteral<"query-result">;
-        title: z.ZodString;
-        sqlQuery: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        title: string;
-        sqlQuery: string;
-        type: "query-result";
-    }, {
-        title: string;
-        sqlQuery: string;
-        type: "query-result";
-    }>, z.ZodString]>;
+export type ErrorMessageSchema = z.infer<typeof ErrorMessageSchema>;
+export declare const StreamMessagePartSchema: z.ZodUnion<[z.ZodObject<{
+    type: z.ZodLiteral<"text">;
+    text: z.ZodString;
+    additionalData: z.ZodOptional<z.ZodAny>;
+    isCompleted: z.ZodOptional<z.ZodBoolean>;
 }, "strip", z.ZodTypeAny, {
-    toolCallId: string;
-    element: string | {
-        title: string;
-        sqlQuery: string;
-        type: "query-result";
-    };
+    type: "text";
+    text: string;
+    additionalData?: any;
+    isCompleted?: boolean | undefined;
 }, {
-    toolCallId: string;
-    element: string | {
-        title: string;
-        sqlQuery: string;
-        type: "query-result";
-    };
-}>;
-export type ToolCallMessageSchema = z.infer<typeof ToolCallMessageSchema>;
-export declare const ToolResultSchema: z.ZodObject<{
-    toolName: z.ZodString;
-    toolCallId: z.ZodString;
-    args: z.ZodRecord<z.ZodString, z.ZodAny>;
-    result: z.ZodUnion<[z.ZodObject<{
-        success: z.ZodLiteral<false>;
-        error: z.ZodString;
+    type: "text";
+    text: string;
+    additionalData?: any;
+    isCompleted?: boolean | undefined;
+}>, z.ZodObject<{
+    type: z.ZodLiteral<"tool-invocation">;
+    toolInvocation: z.ZodObject<{
+        toolCallId: z.ZodString;
+        toolName: z.ZodString;
+        args: z.ZodAny;
+        state: z.ZodString;
+        result: z.ZodOptional<z.ZodAny>;
     }, "strip", z.ZodTypeAny, {
-        success: false;
-        error: string;
+        toolCallId: string;
+        toolName: string;
+        state: string;
+        result?: any;
+        args?: any;
     }, {
-        success: false;
-        error: string;
-    }>, z.ZodObject<{
-        success: z.ZodLiteral<true>;
-        data: z.ZodRecord<z.ZodString, z.ZodAny>;
-    }, "strip", z.ZodTypeAny, {
-        success: true;
-        data: Record<string, any>;
-    }, {
-        success: true;
-        data: Record<string, any>;
-    }>]>;
+        toolCallId: string;
+        toolName: string;
+        state: string;
+        result?: any;
+        args?: any;
+    }>;
+    additionalData: z.ZodOptional<z.ZodAny>;
+    isCompleted: z.ZodOptional<z.ZodBoolean>;
 }, "strip", z.ZodTypeAny, {
-    result: {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
+    type: "tool-invocation";
+    toolInvocation: {
+        toolCallId: string;
+        toolName: string;
+        state: string;
+        result?: any;
+        args?: any;
     };
-    toolName: string;
-    toolCallId: string;
-    args: Record<string, any>;
+    additionalData?: any;
+    isCompleted?: boolean | undefined;
 }, {
-    result: {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
+    type: "tool-invocation";
+    toolInvocation: {
+        toolCallId: string;
+        toolName: string;
+        state: string;
+        result?: any;
+        args?: any;
     };
-    toolName: string;
-    toolCallId: string;
-    args: Record<string, any>;
-}>;
-export type ToolResultSchema = z.infer<typeof ToolResultSchema>;
-export declare const ErrorMessageSchema: z.ZodObject<{
-    error: z.ZodString;
-}, "strip", z.ZodTypeAny, {
-    error: string;
-}, {
-    error: string;
-}>;
-export type ErrorMessageSchema = z.infer<typeof ErrorMessageSchema>;
+    additionalData?: any;
+    isCompleted?: boolean | undefined;
+}>, z.ZodObject<{
+    type: z.ZodString;
+    additionalData: z.ZodOptional<z.ZodAny>;
+    isCompleted: z.ZodOptional<z.ZodBoolean>;
+}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+    type: z.ZodString;
+    additionalData: z.ZodOptional<z.ZodAny>;
+    isCompleted: z.ZodOptional<z.ZodBoolean>;
+}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+    type: z.ZodString;
+    additionalData: z.ZodOptional<z.ZodAny>;
+    isCompleted: z.ZodOptional<z.ZodBoolean>;
+}, z.ZodTypeAny, "passthrough">>]>;
 export declare const AnalysisResultSchema: z.ZodObject<{
     id: z.ZodString;
     prompt: z.ZodString;
-    streamMessage: z.ZodObject<{
-        reasoning: z.ZodOptional<z.ZodString>;
-        toolCallMessages: z.ZodOptional<z.ZodArray<z.ZodObject<{
-            toolName: z.ZodString;
-            toolCallId: z.ZodString;
-            args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-            isCompleted: z.ZodBoolean;
-            llmResult: z.ZodOptional<z.ZodUnknown>;
-            additionalData: z.ZodOptional<z.ZodUnknown>;
-            text: z.ZodOptional<z.ZodString>;
-        }, "strip", z.ZodTypeAny, {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }, {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }>, "many">>;
-        analysis: z.ZodOptional<z.ZodString>;
-        text: z.ZodOptional<z.ZodString>;
+    streamMessage: z.ZodEffects<z.ZodObject<{
         parts: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodObject<{
             type: z.ZodLiteral<"text">;
             text: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
         }, "strip", z.ZodTypeAny, {
-            text: string;
             type: "text";
-        }, {
             text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        }, {
             type: "text";
+            text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
         }>, z.ZodObject<{
-            type: z.ZodLiteral<"tool">;
-            toolCallMessages: z.ZodArray<z.ZodObject<{
-                toolName: z.ZodString;
+            type: z.ZodLiteral<"tool-invocation">;
+            toolInvocation: z.ZodObject<{
                 toolCallId: z.ZodString;
-                args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-                isCompleted: z.ZodBoolean;
-                llmResult: z.ZodOptional<z.ZodUnknown>;
-                additionalData: z.ZodOptional<z.ZodUnknown>;
-                text: z.ZodOptional<z.ZodString>;
+                toolName: z.ZodString;
+                args: z.ZodAny;
+                state: z.ZodString;
+                result: z.ZodOptional<z.ZodAny>;
             }, "strip", z.ZodTypeAny, {
-                toolName: string;
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }, {
                 toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            }, {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }>, "many">;
-        }, "strip", z.ZodTypeAny, {
-            type: "tool";
-            toolCallMessages: {
                 toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            }>;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, "strip", z.ZodTypeAny, {
+            type: "tool-invocation";
+            toolInvocation: {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        }, {
-            type: "tool";
-            toolCallMessages: {
                 toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        }, {
+            type: "tool-invocation";
+            toolInvocation: {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        }>]>, "many">>;
+                toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        }>, z.ZodObject<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">>]>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        text?: string | undefined;
-        toolCallMessages?: {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }[] | undefined;
-        reasoning?: string | undefined;
-        analysis?: string | undefined;
         parts?: ({
-            text: string;
             type: "text";
+            text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
         } | {
-            type: "tool";
-            toolCallMessages: {
-                toolName: string;
+            type: "tool-invocation";
+            toolInvocation: {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        })[] | undefined;
+                toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        } | z.objectOutputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">)[] | undefined;
     }, {
-        text?: string | undefined;
-        toolCallMessages?: {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }[] | undefined;
-        reasoning?: string | undefined;
-        analysis?: string | undefined;
         parts?: ({
-            text: string;
             type: "text";
+            text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
         } | {
-            type: "tool";
-            toolCallMessages: {
+            type: "tool-invocation";
+            toolInvocation: {
+                toolCallId: string;
                 toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        } | z.objectInputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">)[] | undefined;
+    }>, {
+        parts?: ({
+            type: "text";
+            text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        } | {
+            type: "tool-invocation";
+            toolInvocation: {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        })[] | undefined;
-    }>;
+                toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        } | z.objectOutputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">)[] | undefined;
+    }, unknown>;
     errorMessage: z.ZodOptional<z.ZodObject<{
         error: z.ZodString;
     }, "strip", z.ZodTypeAny, {
@@ -321,73 +248,39 @@ export declare const AnalysisResultSchema: z.ZodObject<{
     isCompleted: z.ZodBoolean;
 }, "strip", z.ZodTypeAny, {
     id: string;
+    isCompleted: boolean;
     prompt: string;
     streamMessage: {
-        text?: string | undefined;
-        toolCallMessages?: {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }[] | undefined;
-        reasoning?: string | undefined;
-        analysis?: string | undefined;
         parts?: ({
-            text: string;
             type: "text";
+            text: string;
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
         } | {
-            type: "tool";
-            toolCallMessages: {
-                toolName: string;
+            type: "tool-invocation";
+            toolInvocation: {
                 toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        })[] | undefined;
+                toolName: string;
+                state: string;
+                result?: any;
+                args?: any;
+            };
+            additionalData?: any;
+            isCompleted?: boolean | undefined;
+        } | z.objectOutputType<{
+            type: z.ZodString;
+            additionalData: z.ZodOptional<z.ZodAny>;
+            isCompleted: z.ZodOptional<z.ZodBoolean>;
+        }, z.ZodTypeAny, "passthrough">)[] | undefined;
     };
-    isCompleted: boolean;
     errorMessage?: {
         error: string;
     } | undefined;
 }, {
     id: string;
-    prompt: string;
-    streamMessage: {
-        text?: string | undefined;
-        toolCallMessages?: {
-            toolName: string;
-            toolCallId: string;
-            args: Record<string, unknown>;
-            isCompleted: boolean;
-            text?: string | undefined;
-            llmResult?: unknown;
-            additionalData?: unknown;
-        }[] | undefined;
-        reasoning?: string | undefined;
-        analysis?: string | undefined;
-        parts?: ({
-            text: string;
-            type: "text";
-        } | {
-            type: "tool";
-            toolCallMessages: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[];
-        })[] | undefined;
-    };
     isCompleted: boolean;
+    prompt: string;
+    streamMessage?: unknown;
     errorMessage?: {
         error: string;
     } | undefined;
@@ -401,151 +294,147 @@ export declare const AnalysisSessionSchema: z.ZodObject<{
     analysisResults: z.ZodArray<z.ZodObject<{
         id: z.ZodString;
         prompt: z.ZodString;
-        streamMessage: z.ZodObject<{
-            reasoning: z.ZodOptional<z.ZodString>;
-            toolCallMessages: z.ZodOptional<z.ZodArray<z.ZodObject<{
-                toolName: z.ZodString;
-                toolCallId: z.ZodString;
-                args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-                isCompleted: z.ZodBoolean;
-                llmResult: z.ZodOptional<z.ZodUnknown>;
-                additionalData: z.ZodOptional<z.ZodUnknown>;
-                text: z.ZodOptional<z.ZodString>;
-            }, "strip", z.ZodTypeAny, {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }, {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }>, "many">>;
-            analysis: z.ZodOptional<z.ZodString>;
-            text: z.ZodOptional<z.ZodString>;
+        streamMessage: z.ZodEffects<z.ZodObject<{
             parts: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodObject<{
                 type: z.ZodLiteral<"text">;
                 text: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
             }, "strip", z.ZodTypeAny, {
-                text: string;
                 type: "text";
-            }, {
                 text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            }, {
                 type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
             }>, z.ZodObject<{
-                type: z.ZodLiteral<"tool">;
-                toolCallMessages: z.ZodArray<z.ZodObject<{
-                    toolName: z.ZodString;
+                type: z.ZodLiteral<"tool-invocation">;
+                toolInvocation: z.ZodObject<{
                     toolCallId: z.ZodString;
-                    args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-                    isCompleted: z.ZodBoolean;
-                    llmResult: z.ZodOptional<z.ZodUnknown>;
-                    additionalData: z.ZodOptional<z.ZodUnknown>;
-                    text: z.ZodOptional<z.ZodString>;
+                    toolName: z.ZodString;
+                    args: z.ZodAny;
+                    state: z.ZodString;
+                    result: z.ZodOptional<z.ZodAny>;
                 }, "strip", z.ZodTypeAny, {
-                    toolName: string;
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }, {
                     toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                }, {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }>, "many">;
-            }, "strip", z.ZodTypeAny, {
-                type: "tool";
-                toolCallMessages: {
                     toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                }>;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, "strip", z.ZodTypeAny, {
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            }, {
-                type: "tool";
-                toolCallMessages: {
                     toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            }, {
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            }>]>, "many">>;
+                    toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            }>, z.ZodObject<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">>]>, "many">>;
         }, "strip", z.ZodTypeAny, {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
             parts?: ({
-                text: string;
                 type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
             } | {
-                type: "tool";
-                toolCallMessages: {
-                    toolName: string;
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
+                    toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | z.objectOutputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">)[] | undefined;
         }, {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
             parts?: ({
-                text: string;
                 type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
             } | {
-                type: "tool";
-                toolCallMessages: {
+                type: "tool-invocation";
+                toolInvocation: {
+                    toolCallId: string;
                     toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | z.objectInputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">)[] | undefined;
+        }>, {
+            parts?: ({
+                type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | {
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
-        }>;
+                    toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | z.objectOutputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">)[] | undefined;
+        }, unknown>;
         errorMessage: z.ZodOptional<z.ZodObject<{
             error: z.ZodString;
         }, "strip", z.ZodTypeAny, {
@@ -556,73 +445,39 @@ export declare const AnalysisSessionSchema: z.ZodObject<{
         isCompleted: z.ZodBoolean;
     }, "strip", z.ZodTypeAny, {
         id: string;
+        isCompleted: boolean;
         prompt: string;
         streamMessage: {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
             parts?: ({
-                text: string;
                 type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
             } | {
-                type: "tool";
-                toolCallMessages: {
-                    toolName: string;
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
+                    toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | z.objectOutputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">)[] | undefined;
         };
-        isCompleted: boolean;
         errorMessage?: {
             error: string;
         } | undefined;
     }, {
         id: string;
-        prompt: string;
-        streamMessage: {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
-            parts?: ({
-                text: string;
-                type: "text";
-            } | {
-                type: "tool";
-                toolCallMessages: {
-                    toolName: string;
-                    toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
-        };
         isCompleted: boolean;
+        prompt: string;
+        streamMessage?: unknown;
         errorMessage?: {
             error: string;
         } | undefined;
@@ -635,37 +490,31 @@ export declare const AnalysisSessionSchema: z.ZodObject<{
     model: string;
     analysisResults: {
         id: string;
+        isCompleted: boolean;
         prompt: string;
         streamMessage: {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
             parts?: ({
-                text: string;
                 type: "text";
+                text: string;
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
             } | {
-                type: "tool";
-                toolCallMessages: {
-                    toolName: string;
+                type: "tool-invocation";
+                toolInvocation: {
                     toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
+                    toolName: string;
+                    state: string;
+                    result?: any;
+                    args?: any;
+                };
+                additionalData?: any;
+                isCompleted?: boolean | undefined;
+            } | z.objectOutputType<{
+                type: z.ZodString;
+                additionalData: z.ZodOptional<z.ZodAny>;
+                isCompleted: z.ZodOptional<z.ZodBoolean>;
+            }, z.ZodTypeAny, "passthrough">)[] | undefined;
         };
-        isCompleted: boolean;
         errorMessage?: {
             error: string;
         } | undefined;
@@ -678,37 +527,9 @@ export declare const AnalysisSessionSchema: z.ZodObject<{
     model: string;
     analysisResults: {
         id: string;
-        prompt: string;
-        streamMessage: {
-            text?: string | undefined;
-            toolCallMessages?: {
-                toolName: string;
-                toolCallId: string;
-                args: Record<string, unknown>;
-                isCompleted: boolean;
-                text?: string | undefined;
-                llmResult?: unknown;
-                additionalData?: unknown;
-            }[] | undefined;
-            reasoning?: string | undefined;
-            analysis?: string | undefined;
-            parts?: ({
-                text: string;
-                type: "text";
-            } | {
-                type: "tool";
-                toolCallMessages: {
-                    toolName: string;
-                    toolCallId: string;
-                    args: Record<string, unknown>;
-                    isCompleted: boolean;
-                    text?: string | undefined;
-                    llmResult?: unknown;
-                    additionalData?: unknown;
-                }[];
-            })[] | undefined;
-        };
         isCompleted: boolean;
+        prompt: string;
+        streamMessage?: unknown;
         errorMessage?: {
             error: string;
         } | undefined;