From e99c8bdd502ccaee8ed4b1038b4fd75d97a2d701 Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Sat, 6 Dec 2025 15:25:15 +0530
Subject: [PATCH] feat(models-types): update to use `Message`

---
 src/lib/models/types.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/lib/models/types.ts b/src/lib/models/types.ts
index 54c3e1e..8abefd7 100644
--- a/src/lib/models/types.ts
+++ b/src/lib/models/types.ts
@@ -1,5 +1,5 @@
 import z from 'zod';
-import { ChatTurnMessage } from '../types';
+import { Message } from '../types';
 
 type Model = {
   name: string;
@@ -44,13 +44,13 @@ type Tool = {
 };
 
 type ToolCall = {
-  id?: string;
+  id: string;
   name: string;
   arguments: Record<string, any>;
 };
 
 type GenerateTextInput = {
-  messages: ChatTurnMessage[];
+  messages: Message[];
   tools?: Tool[];
   options?: GenerateOptions;
 };
@@ -63,14 +63,14 @@ type GenerateTextOutput = {
 
 type StreamTextOutput = {
   contentChunk: string;
-  toolCallChunk: Partial<ToolCall>[];
+  toolCallChunk: ToolCall[];
   additionalInfo?: Record<string, any>;
   done?: boolean;
 };
 
 type GenerateObjectInput = {
   schema: z.ZodTypeAny;
-  messages: ChatTurnMessage[];
+  messages: Message[];
   options?: GenerateOptions;
 };