remove stream.ts

Author: mme

typescript-sdk/packages/client/src/agent/agent.ts

@@ -1,5 +1,5 @@
 import { defaultApplyEvents } from "@/apply/default";
-import { Message, State, RunAgentInput, RunAgent, BaseEvent, AgentState } from "@ag-ui/core";
+import { Message, State, RunAgentInput, BaseEvent } from "@ag-ui/core";
 
 import { AgentConfig, RunAgentParameters } from "./types";
 import { v4 as uuidv4 } from "uuid";
@@ -12,7 +12,7 @@ import { convertToLegacyEvents } from "@/legacy/convert";
 import { LegacyRuntimeProtocolEvent } from "@/legacy/types";
 import { lastValueFrom, of } from "rxjs";
 import { transformChunks } from "@/chunks";
-import { RunAgentSubscriber } from "./subscriber";
+import { AgentStateMutation, RunAgentSubscriber } from "./subscriber";
 
 export abstract class AbstractAgent {
   public agentId?: string;
@@ -48,7 +48,7 @@ export abstract class AbstractAgent {
     };
   }
 
-  protected abstract run(...args: Parameters<RunAgent>): ReturnType<RunAgent>;
+  protected abstract run(input: RunAgentInput): Observable<BaseEvent>;
 
   public async runAgent(parameters?: RunAgentParameters): Promise<void> {
     this.agentId = this.agentId ?? uuidv4();
@@ -74,14 +74,17 @@ export abstract class AbstractAgent {
 
   public abortRun() {}
 
-  protected apply(input: RunAgentInput, events$: Observable<BaseEvent>): Observable<AgentState> {
+  protected apply(
+    input: RunAgentInput,
+    events$: Observable<BaseEvent>,
+  ): Observable<AgentStateMutation> {
     return defaultApplyEvents(input, events$);
   }
 
   protected processApplyEvents(
     input: RunAgentInput,
-    events$: Observable<AgentState>,
-  ): Observable<AgentState> {
+    events$: Observable<AgentStateMutation>,
+  ): Observable<AgentStateMutation> {
     return events$.pipe(
       tap((event) => {
         if (event.messages) {

typescript-sdk/packages/client/src/agent/http.ts

@@ -1,7 +1,7 @@
 import { AbstractAgent } from "./agent";
-import { runHttpRequest, HttpEvent } from "@/run/http-request";
+import { runHttpRequest } from "@/run/http-request";
 import { HttpAgentConfig, RunAgentParameters } from "./types";
-import { RunAgent, RunAgentInput, BaseEvent } from "@ag-ui/core";
+import { RunAgentInput, BaseEvent } from "@ag-ui/core";
 import { structuredClone_ } from "@/utils";
 import { transformHttpEventStream } from "@/transform/http";
 import { Observable } from "rxjs";

typescript-sdk/packages/client/src/apply/default.ts

@@ -19,7 +19,7 @@ import { mergeMap } from "rxjs/operators";
 import { structuredClone_ } from "../utils";
 import { applyPatch } from "fast-json-patch";
 import untruncateJson from "untruncate-json";
-import { AgentState } from "@ag-ui/core";
+import { AgentStateMutation } from "@/agent/subscriber";
 import { Observable } from "rxjs";
 
 interface PredictStateValue {
@@ -31,13 +31,13 @@ interface PredictStateValue {
 export const defaultApplyEvents = (
   input: RunAgentInput,
   events$: Observable<BaseEvent>,
-): Observable<AgentState> => {
+): Observable<AgentStateMutation> => {
   let messages = structuredClone_(input.messages);
   let state = structuredClone_(input.state);
   let predictState: PredictStateValue[] | undefined;
 
   // Helper function to emit state updates with proper cloning
-  const emitUpdate = (agentState: AgentState) => [structuredClone_(agentState)];
+  const emitUpdate = (agentState: AgentStateMutation) => [structuredClone_(agentState)];
 
   const emitNoUpdate = () => [];
 

typescript-sdk/packages/core/src/index.ts

@@ -3,6 +3,3 @@ export * from "./types";
 
 // Export all event-related types and schemas
 export * from "./events";
-
-// Export all stream-related types and schemas
-export * from "./stream";