@threadplane/ag-ui 0.0.55 → 0.0.57

package/types/threadplane-ag-ui.d.ts

@@ -1,5 +1,5 @@
 import { Signal, Provider } from '@angular/core';
-import { AbstractAgent, RunAgentInput, BaseEvent } from '@ag-ui/client';
+import { AbstractAgent, BaseEvent, RunAgentInput } from '@ag-ui/client';
 import { Agent, ClientToolsCapability, Subagent, AgentRuntimeTelemetrySink, AgentRef, Message } from '@threadplane/chat';
 import { Observable } from 'rxjs';
 import { FakeAgentConfig } from '@threadplane/chat/testing';
@@ -21,10 +21,12 @@ interface ToAgentOptions {
     telemetry?: AgentRuntimeTelemetrySink | false;
 }
 /**
- * The neutral Agent contract, widened with the AG-UI adapter's optional
+ * The neutral Agent contract, widened with the AG-UI adapter's
  * `customEvents` signal (the chat composition feature-detects it to enable
- * live a2ui streaming) and the optional `clientTools` capability.
- * Mirrors langgraph's LangGraphAgent extension.
+ * live a2ui streaming), the browser client-tools capability, and concrete
+ * ACTIVITY-backed `subagents`.
+ * Mirrors langgraph's LangGraphAgent extension where the protocol surfaces
+ * overlap.
  */
 interface AgUiAgent<TState = Record<string, unknown>> extends Agent<TState> {
     customEvents: Signal<CustomStreamEvent[]>;
@@ -47,6 +49,14 @@ interface AgUiAgent<TState = Record<string, unknown>> extends Agent<TState> {
  * of toAgent() should treat the returned object's lifecycle as tied to the
  * agent instance they constructed. The subscriber registered via
  * source.subscribe() will fire for the lifetime of source.
+ *
+ * @example
+ * ```ts
+ * import { HttpAgent } from '@ag-ui/client';
+ * import { toAgent } from '@threadplane/ag-ui';
+ *
+ * const agent = toAgent(new HttpAgent({ url: '/api/agent' }));
+ * ```
  */
 declare function toAgent(source: AbstractAgent, options?: ToAgentOptions): AgUiAgent;
 
@@ -116,6 +126,13 @@ declare function provideAgent(configOrFactory: AgentConfig | (() => AgentConfig)
 declare function injectAgent(): AgUiAgent;
 declare function injectAgent<T>(ref: AgentRef<T>): AgUiAgent<T>;
 
+type FakeAgentScriptWhen = 'initial' | {
+    toolMessageFor: string;
+};
+interface FakeAgentScriptBranch {
+    when: FakeAgentScriptWhen;
+    events: readonly BaseEvent[];
+}
 /**
  * In-process AG-UI agent that emits a canned streaming response.
  *
@@ -134,13 +151,18 @@ declare class FakeAgent extends AbstractAgent {
     private readonly reasoningTokens;
     /** Milliseconds between successive token emissions. */
     private readonly delayMs;
+    /** Optional deterministic event branches for tests that need exact streams. */
+    private readonly script;
     constructor(opts?: {
         tokens?: string[];
         /** Optional reasoning chunks emitted before the text reply. */
         reasoningTokens?: string[];
         delayMs?: number;
+        script?: readonly FakeAgentScriptBranch[];
     });
     run(input: RunAgentInput): Observable<BaseEvent>;
+    private scriptedSequence;
+    private emitSequence;
 }
 
 /**
@@ -152,7 +174,7 @@ declare class FakeAgent extends AbstractAgent {
  * @example
  * ```ts
  * TestBed.configureTestingModule({
- *   providers: [provideFakeAgent({ responses: ['Hello from the fake agent'] })],
+ *   providers: [provideFakeAgent({ tokens: ['Hello from the fake agent'] })],
  * });
  * ```
  */