@livekit/agents 0.7.9 → 1.0.0-next.0

package/src/voice/agent.test.ts

@@ -0,0 +1,80 @@
+// SPDX-FileCopyrightText: 2025 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { describe, expect, it } from 'vitest';
+import { z } from 'zod';
+import { tool } from '../llm/index.js';
+import { Agent } from './agent.js';
+
+describe('Agent', () => {
+  it('should create agent with basic instructions', () => {
+    const instructions = 'You are a helpful assistant';
+    const agent = new Agent({ instructions });
+
+    expect(agent).toBeDefined();
+    expect(agent.instructions).toBe(instructions);
+  });
+
+  it('should create agent with instructions and tools', () => {
+    const instructions = 'You are a helpful assistant with tools';
+
+    // Create mock tools using the tool function
+    const mockTool1 = tool({
+      description: 'First test tool',
+      parameters: z.object({}),
+      execute: async () => 'tool1 result',
+    });
+
+    const mockTool2 = tool({
+      description: 'Second test tool',
+      parameters: z.object({
+        input: z.string().describe('Input parameter'),
+      }),
+      execute: async ({ input }) => `tool2: ${input}`,
+    });
+
+    const agent = new Agent({
+      instructions,
+      tools: {
+        getTool1: mockTool1,
+        getTool2: mockTool2,
+      },
+    });
+
+    expect(agent).toBeDefined();
+    expect(agent.instructions).toBe(instructions);
+
+    // Assert tools are set correctly
+    const agentTools = agent.toolCtx;
+    expect(Object.keys(agentTools)).toHaveLength(2);
+    expect(agentTools).toHaveProperty('getTool1');
+    expect(agentTools).toHaveProperty('getTool2');
+
+    // Verify tool properties with proper checks
+    expect(agentTools.getTool1?.description).toBe('First test tool');
+    expect(agentTools.getTool2?.description).toBe('Second test tool');
+  });
+
+  it('should return a copy of tools, not the original reference', () => {
+    const instructions = 'You are a helpful assistant';
+    const mockTool = tool({
+      description: 'Test tool',
+      parameters: z.object({}),
+      execute: async () => 'result',
+    });
+
+    const tools = { testTool: mockTool };
+    const agent = new Agent({ instructions, tools });
+
+    const tools1 = agent.toolCtx;
+    const tools2 = agent.toolCtx;
+
+    // Should return different object references
+    expect(tools1).not.toBe(tools2);
+    expect(tools1).not.toBe(tools);
+
+    // Should contain the same set of tools
+    expect(tools1).toEqual(tools2);
+    expect(tools1).toEqual(tools);
+  });
+});

package/src/voice/agent.ts

@@ -0,0 +1,332 @@
+// SPDX-FileCopyrightText: 2025 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import type { AudioFrame } from '@livekit/rtc-node';
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { ReadableStream } from 'node:stream/web';
+import { ReadonlyChatContext } from '../llm/chat_context.js';
+import type { ChatMessage, FunctionCall, RealtimeModel } from '../llm/index.js';
+import {
+  type ChatChunk,
+  ChatContext,
+  LLM,
+  type ToolChoice,
+  type ToolContext,
+} from '../llm/index.js';
+import type { STT, SpeechEvent } from '../stt/index.js';
+import { StreamAdapter as STTStreamAdapter } from '../stt/index.js';
+import { SentenceTokenizer as BasicSentenceTokenizer } from '../tokenize/basic/index.js';
+import type { TTS } from '../tts/index.js';
+import { SynthesizeStream, StreamAdapter as TTSStreamAdapter } from '../tts/index.js';
+import type { VAD } from '../vad.js';
+import type { AgentActivity } from './agent_activity.js';
+import type { AgentSession, TurnDetectionMode } from './agent_session.js';
+
+export const asyncLocalStorage = new AsyncLocalStorage<{ functionCall?: FunctionCall }>();
+export const STOP_RESPONSE_SYMBOL = Symbol('StopResponse');
+
+export class StopResponse extends Error {
+  constructor() {
+    super();
+    this.name = 'StopResponse';
+
+    Object.defineProperty(this, STOP_RESPONSE_SYMBOL, {
+      value: true,
+    });
+  }
+}
+
+export function isStopResponse(value: unknown): value is StopResponse {
+  return (
+    value !== undefined &&
+    value !== null &&
+    typeof value === 'object' &&
+    STOP_RESPONSE_SYMBOL in value
+  );
+}
+
+export interface ModelSettings {
+  /* The tool choice to use when calling the LLM. */
+  toolChoice?: ToolChoice;
+}
+
+export interface AgentOptions<UserData> {
+  instructions: string;
+  chatCtx?: ChatContext;
+  tools?: ToolContext<UserData>;
+  turnDetection?: TurnDetectionMode;
+  stt?: STT;
+  vad?: VAD;
+  llm?: LLM | RealtimeModel;
+  tts?: TTS;
+  allowInterruptions?: boolean;
+  minConsecutiveSpeechDelay?: number;
+}
+
+export class Agent<UserData = any> {
+  private turnDetection?: TurnDetectionMode;
+  private _stt?: STT;
+  private _vad?: VAD;
+  private _llm?: LLM | RealtimeModel;
+  private _tts?: TTS;
+
+  /** @internal */
+  _agentActivity?: AgentActivity;
+
+  /** @internal */
+  _chatCtx: ChatContext;
+
+  /** @internal */
+  _instructions: string;
+
+  /** @internal */
+  _tools?: ToolContext<UserData>;
+
+  constructor({
+    instructions,
+    chatCtx,
+    tools,
+    turnDetection,
+    stt,
+    vad,
+    llm,
+    tts,
+  }: AgentOptions<UserData>) {
+    this._instructions = instructions;
+    this._tools = { ...tools };
+    this._chatCtx = chatCtx
+      ? chatCtx.copy({
+          toolCtx: this._tools,
+        })
+      : ChatContext.empty();
+
+    this.turnDetection = turnDetection;
+    this._stt = stt;
+    this._vad = vad;
+    this._llm = llm;
+    this._tts = tts;
+    this._agentActivity = undefined;
+  }
+
+  get vad(): VAD | undefined {
+    return this._vad;
+  }
+
+  get stt(): STT | undefined {
+    return this._stt;
+  }
+
+  get llm(): LLM | RealtimeModel | undefined {
+    return this._llm;
+  }
+
+  get tts(): TTS | undefined {
+    return this._tts;
+  }
+
+  get chatCtx(): ReadonlyChatContext {
+    return new ReadonlyChatContext(this._chatCtx.items);
+  }
+
+  get instructions(): string {
+    return this._instructions;
+  }
+
+  get toolCtx(): ToolContext<UserData> {
+    return { ...this._tools };
+  }
+
+  get session(): AgentSession<UserData> {
+    return this.getActivityOrThrow().agentSession as AgentSession<UserData>;
+  }
+
+  async onEnter(): Promise<void> {}
+
+  async onExit(): Promise<void> {}
+
+  async transcriptionNode(
+    text: ReadableStream<string>,
+    modelSettings: ModelSettings,
+  ): Promise<ReadableStream<string> | null> {
+    return Agent.default.transcriptionNode(this, text, modelSettings);
+  }
+
+  async onUserTurnCompleted(_chatCtx: ChatContext, _newMessage: ChatMessage): Promise<void> {}
+
+  async sttNode(
+    audio: ReadableStream<AudioFrame>,
+    modelSettings: ModelSettings,
+  ): Promise<ReadableStream<SpeechEvent | string> | null> {
+    return Agent.default.sttNode(this, audio, modelSettings);
+  }
+
+  async llmNode(
+    chatCtx: ChatContext,
+    toolCtx: ToolContext,
+    modelSettings: ModelSettings,
+  ): Promise<ReadableStream<ChatChunk | string> | null> {
+    return Agent.default.llmNode(this, chatCtx, toolCtx, modelSettings);
+  }
+
+  async ttsNode(
+    text: ReadableStream<string>,
+    modelSettings: ModelSettings,
+  ): Promise<ReadableStream<AudioFrame> | null> {
+    return Agent.default.ttsNode(this, text, modelSettings);
+  }
+
+  async realtimeAudioOutputNode(
+    audio: ReadableStream<AudioFrame>,
+    modelSettings: ModelSettings,
+  ): Promise<ReadableStream<AudioFrame> | null> {
+    return Agent.default.realtimeAudioOutputNode(this, audio, modelSettings);
+  }
+
+  // realtime_audio_output_node
+
+  getActivityOrThrow(): AgentActivity {
+    if (!this._agentActivity) {
+      throw new Error('Agent activity not found');
+    }
+    return this._agentActivity;
+  }
+
+  async updateChatCtx(chatCtx: ChatContext): Promise<void> {
+    if (!this._agentActivity) {
+      this._chatCtx = chatCtx.copy({ toolCtx: this.toolCtx });
+      return;
+    }
+
+    this._agentActivity.updateChatCtx(chatCtx);
+  }
+
+  static default = {
+    async sttNode(
+      agent: Agent,
+      audio: ReadableStream<AudioFrame>,
+      _modelSettings: ModelSettings,
+    ): Promise<ReadableStream<SpeechEvent | string> | null> {
+      const activity = agent.getActivityOrThrow();
+      if (!activity.stt) {
+        throw new Error('sttNode called but no STT node is available');
+      }
+
+      let wrapped_stt = activity.stt;
+
+      if (!wrapped_stt.capabilities.streaming) {
+        if (!agent.vad) {
+          throw new Error(
+            'STT does not support streaming, add a VAD to the AgentTask/VoiceAgent to enable streaming',
+          );
+        }
+        wrapped_stt = new STTStreamAdapter(wrapped_stt, agent.vad);
+      }
+
+      const stream = wrapped_stt.stream();
+      stream.updateInputStream(audio);
+
+      return new ReadableStream({
+        async start(controller) {
+          for await (const event of stream) {
+            controller.enqueue(event);
+          }
+          controller.close();
+        },
+        cancel() {
+          stream.detachInputStream();
+          stream.close();
+        },
+      });
+    },
+
+    async llmNode(
+      agent: Agent,
+      chatCtx: ChatContext,
+      toolCtx: ToolContext,
+      modelSettings: ModelSettings,
+    ): Promise<ReadableStream<ChatChunk | string> | null> {
+      const activity = agent.getActivityOrThrow();
+      if (!activity.llm) {
+        throw new Error('llmNode called but no LLM node is available');
+      }
+
+      if (!(activity.llm instanceof LLM)) {
+        throw new Error(
+          'llmNode should only be used with LLM (non-multimodal/realtime APIs) nodes',
+        );
+      }
+
+      // TODO(brian): make parallelToolCalls configurable
+      const { toolChoice } = modelSettings;
+
+      const stream = activity.llm.chat({
+        chatCtx,
+        toolCtx,
+        toolChoice,
+        parallelToolCalls: true,
+      });
+      return new ReadableStream({
+        async start(controller) {
+          for await (const chunk of stream) {
+            controller.enqueue(chunk);
+          }
+          controller.close();
+        },
+        cancel() {
+          stream.close();
+        },
+      });
+    },
+
+    async ttsNode(
+      agent: Agent,
+      text: ReadableStream<string>,
+      _modelSettings: ModelSettings,
+    ): Promise<ReadableStream<AudioFrame> | null> {
+      const activity = agent.getActivityOrThrow();
+      if (!activity.tts) {
+        throw new Error('ttsNode called but no TTS node is available');
+      }
+
+      let wrapped_tts = activity.tts;
+
+      if (!activity.tts.capabilities.streaming) {
+        wrapped_tts = new TTSStreamAdapter(wrapped_tts, new BasicSentenceTokenizer());
+      }
+
+      const stream = wrapped_tts.stream();
+      stream.updateInputStream(text);
+
+      return new ReadableStream({
+        async start(controller) {
+          for await (const chunk of stream) {
+            if (chunk === SynthesizeStream.END_OF_STREAM) {
+              break;
+            }
+            controller.enqueue(chunk.frame);
+          }
+          controller.close();
+        },
+        cancel() {
+          stream.close();
+        },
+      });
+    },
+
+    async transcriptionNode(
+      agent: Agent,
+      text: ReadableStream<string>,
+      _modelSettings: ModelSettings,
+    ): Promise<ReadableStream<string> | null> {
+      return text;
+    },
+
+    async realtimeAudioOutputNode(
+      _agent: Agent,
+      audio: ReadableStream<AudioFrame>,
+      _modelSettings: ModelSettings,
+    ): Promise<ReadableStream<AudioFrame> | null> {
+      return audio;
+    },
+  };
+}