chores(ts) remove aimessage alias

Author: sslava

src/agent.ts

@@ -1,10 +1,10 @@
-import { LanguageModel, Message as AIMessage } from 'ai';
+import { LanguageModel, Message } from 'ai';
 import { Schema, z } from 'zod';
 
 import { GenericToolSet, inferParameters, ToolParameters } from './tools.js';
 import { Context } from './context.js';
 
-export type PromptType = { messages?: AIMessage[]; prompt?: string };
+export type PromptType = { messages?: Message[]; prompt?: string };
 
 export type LlmAgent<C extends Context, P extends ToolParameters> = {
   isLlmAgent: true;

src/context.ts

@@ -1,4 +1,4 @@
-import { Message as AIMessage, CoreAssistantMessage, CoreToolMessage, DataStreamWriter } from 'ai';
+import { Message, CoreAssistantMessage, CoreToolMessage, DataStreamWriter } from 'ai';
 
 import { VercelStreamWriter } from './stream.js';
 
@@ -8,15 +8,15 @@ export type AIResponseMessage = (CoreAssistantMessage | CoreToolMessage) & {
 
 export type Context = {
   readonly dataStream?: DataStreamWriter;
-  readonly history?: AIMessage[];
+  readonly history?: Message[];
 };
 
 export interface IRunContext<C extends Context> {
   readonly steps: RunStepContext<C>[];
 
   get writer(): VercelStreamWriter;
   get inner(): C;
-  get history(): AIMessage[] | undefined;
+  get history(): Message[] | undefined;
 
   step(): IRunContext<C>;
 
@@ -28,7 +28,7 @@ class RunStepContext<C extends Context> implements IRunContext<C> {
 
   public data: Record<string, unknown> = {};
 
-  public get history(): AIMessage[] | undefined {
+  public get history(): Message[] | undefined {
     return this.context.history;
   }
 
@@ -56,7 +56,7 @@ class RunStepContext<C extends Context> implements IRunContext<C> {
 export class RunFlowContext<C extends Context> implements IRunContext<C> {
   public writer: VercelStreamWriter;
 
-  public messages: AIMessage[] = [];
+  public messages: Message[] = [];
 
   public responseMessages: AIResponseMessage[] = [];
 
@@ -75,7 +75,7 @@ export class RunFlowContext<C extends Context> implements IRunContext<C> {
     return step;
   }
 
-  public get history(): AIMessage[] | undefined {
+  public get history(): Message[] | undefined {
     return this.inner.history;
   }
 

src/flow.ts

@@ -7,7 +7,7 @@ import {
   ToolSet,
   Output,
   generateText,
-  Message as AIMessage,
+  Message,
   StepResult,
   StreamTextOnStepFinishCallback,
   GenerateTextOnStepFinishCallback,
@@ -62,7 +62,7 @@ export abstract class AgentFlow<C extends Context> {
   protected async agentStreamText<P extends ToolParameters>(
     agent: LlmAgent<C, P>,
     ctx: IRunContext<C>,
-    messages: AIMessage[],
+    messages: Message[],
     onFinish?: StreamTextOnFinishCallback<ToolSet>,
     onStepFinish?: StreamTextOnStepFinishCallback<ToolSet>
   ) {