1 files changed, 5 insertions, 2 deletions
diff --git a/src/types/index.ts b/src/types/index.ts
index 6c16e0d..1e4d57d 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -22,9 +22,12 @@ export interface AIModelAPI {
   tokenizer: (text: string) => number;
   maxTokens: number;
 
-  send: (input: string, systemPrompt?: string) => AsyncRes<string>;
+  send: (
+    input: string | InputToken[],
+    systemPrompt?: string,
+  ) => AsyncRes<string>;
   stream: (
-    input: string,
+    input: string | InputToken[],
     handler: (data: string) => void,
     systemPrompt?: string,
   ) => void;