Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions src/AnamClient.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ import {
MessageHistoryClient,
PublicEventEmitter,
StreamingClient,
ReasoningHistoryClient,
} from './modules';
import {
AnamClientOptions,
Expand All @@ -38,6 +39,7 @@ export default class AnamClient {
private internalEventEmitter: InternalEventEmitter;

private readonly messageHistoryClient: MessageHistoryClient;
private readonly reasoningHistoryClient: ReasoningHistoryClient;

private personaConfig: PersonaConfig | undefined;
private clientOptions: AnamClientOptions | undefined;
Expand Down Expand Up @@ -104,6 +106,11 @@ export default class AnamClient {
this.publicEventEmitter,
this.internalEventEmitter,
);

this.reasoningHistoryClient = new ReasoningHistoryClient(
this.publicEventEmitter,
this.internalEventEmitter,
);
}

private decodeJwt(token: string): any {
Expand Down
76 changes: 76 additions & 0 deletions src/modules/ReasoningHistoryClient.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
import { PublicEventEmitter, InternalEventEmitter } from '.';
import {
AnamEvent,
InternalEvent,
ReasoningMessage,
ReasoningStreamEvent,
WebRtcReasoningTextMessageEvent,
} from '../types';

export class ReasoningHistoryClient {
private publicEventEmitter: PublicEventEmitter;
private internalEventEmitter: InternalEventEmitter;

private reasoning_messages: ReasoningMessage[] = [];
constructor(
publicEventEmitter: PublicEventEmitter,
internalEventEmitter: InternalEventEmitter,
) {
this.publicEventEmitter = publicEventEmitter;
this.internalEventEmitter = internalEventEmitter;
// register for events
this.internalEventEmitter.addListener(
InternalEvent.WEBRTC_REASONING_TEXT_MESSAGE_RECEIVED,
this.processWebRtcReasoningTextMessageEvent.bind(this),
);
}

private webRtcTextMessageEventToReasoningStreamEvent(
event: WebRtcReasoningTextMessageEvent,
): ReasoningStreamEvent {
return {
id: `${event.role}::${event.message_id}`,
content: event.content,
endOfThought: event.end_of_thought,
role: event.role,
};
}

private processWebRtcReasoningTextMessageEvent(
event: WebRtcReasoningTextMessageEvent,
): void {
const ReasoningStreamEvent: ReasoningStreamEvent =
this.webRtcTextMessageEventToReasoningStreamEvent(event);

this.publicEventEmitter.emit(
AnamEvent.REASONING_STREAM_EVENT_RECEIVED,
ReasoningStreamEvent,
);

const message: ReasoningMessage = {
id: ReasoningStreamEvent.id,
content: ReasoningStreamEvent.content,
role: ReasoningStreamEvent.role,
};

const existingMessageIndex = this.reasoning_messages.findIndex(
(m) => m.id === message.id,
);
if (existingMessageIndex !== -1) {
// update existing message
const existingMessage = this.reasoning_messages[existingMessageIndex];
existingMessage.content += message.content;
this.reasoning_messages[existingMessageIndex] = existingMessage;
} else {
// new message
this.reasoning_messages.push(message);
}

if (ReasoningStreamEvent.endOfThought) {
this.publicEventEmitter.emit(
AnamEvent.REASONING_HISTORY_UPDATED,
this.reasoning_messages,
);
}
}
}
8 changes: 8 additions & 0 deletions src/modules/StreamingClient.ts
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ import {
StreamingClientOptions,
WebRtcClientToolEvent,
WebRtcTextMessageEvent,
WebRtcReasoningTextMessageEvent,
} from '../types';
import { AgentAudioInputStream } from '../types/AgentAudioInputStream';
import { TalkMessageStream } from '../types/TalkMessageStream';
Expand Down Expand Up @@ -693,6 +694,7 @@ export class StreamingClient {
message.data as WebRtcTextMessageEvent,
);
break;

case DataChannelMessage.CLIENT_TOOL_EVENT:
const webRtcToolEvent = message.data as WebRtcClientToolEvent;

Expand All @@ -709,6 +711,12 @@ export class StreamingClient {
clientToolEvent,
);
break;
case DataChannelMessage.REASONING_TEXT:
this.internalEventEmitter.emit(
InternalEvent.WEBRTC_REASONING_TEXT_MESSAGE_RECEIVED,
message.data as WebRtcReasoningTextMessageEvent,
);
break;
// Unknown message types are silently ignored to maintain forward compatibility
default:
break;
Expand Down
1 change: 1 addition & 0 deletions src/modules/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,4 @@ export { MessageHistoryClient } from './MessageHistoryClient';
export { PublicEventEmitter } from './PublicEventEmitter';
export { StreamingClient } from './StreamingClient';
export { ToolCallManager } from './ToolCallManager';
export { ReasoningHistoryClient } from './ReasoningHistoryClient';
4 changes: 3 additions & 1 deletion src/types/AgentAudioInputStream.ts
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,9 @@ export class AgentAudioInputStream {
private arrayBufferToBase64(buffer: ArrayBuffer | Uint8Array): string {
const bytes =
buffer instanceof Uint8Array ? buffer : new Uint8Array(buffer);
const binary = Array.from(bytes, byte => String.fromCharCode(byte)).join('');
const binary = Array.from(bytes, (byte) => String.fromCharCode(byte)).join(
'',
);
return btoa(binary);
}
}
1 change: 1 addition & 0 deletions src/types/events/internal/InternalEvent.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,4 +3,5 @@ export enum InternalEvent {
SIGNAL_MESSAGE_RECEIVED = 'SIGNAL_MESSAGE_RECEIVED',
WEBRTC_CHAT_MESSAGE_RECEIVED = 'WEBRTC_CHAT_MESSAGE_RECEIVED',
WEBRTC_CLIENT_TOOL_EVENT_RECEIVED = 'WEBRTC_CLIENT_TOOL_EVENT_RECEIVED',
WEBRTC_REASONING_TEXT_MESSAGE_RECEIVED = 'WEBRTC_REASONING_TEXT_MESSAGE_RECEIVED',
}
4 changes: 4 additions & 0 deletions src/types/events/internal/InternalEventCallbacks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ import {
SignalMessage,
WebRtcTextMessageEvent,
WebRtcClientToolEvent,
WebRtcReasoningTextMessageEvent,
} from '../../index';

export type InternalEventCallbacks = {
Expand All @@ -16,4 +17,7 @@ export type InternalEventCallbacks = {
[InternalEvent.WEBRTC_CLIENT_TOOL_EVENT_RECEIVED]: (
webRtcToolEvent: WebRtcClientToolEvent,
) => void;
[InternalEvent.WEBRTC_REASONING_TEXT_MESSAGE_RECEIVED]: (
message: WebRtcReasoningTextMessageEvent,
) => void;
};
2 changes: 2 additions & 0 deletions src/types/events/public/AnamEvent.ts
Original file line number Diff line number Diff line change
Expand Up @@ -15,4 +15,6 @@ export enum AnamEvent {
MIC_PERMISSION_DENIED = 'MIC_PERMISSION_DENIED',
INPUT_AUDIO_DEVICE_CHANGED = 'INPUT_AUDIO_DEVICE_CHANGED',
CLIENT_TOOL_EVENT_RECEIVED = 'CLIENT_TOOL_EVENT_RECEIVED',
REASONING_HISTORY_UPDATED = 'REASONING_HISTORY_UPDATED',
REASONING_STREAM_EVENT_RECEIVED = 'REASONING_STREAM_EVENT_RECEIVED',
}
8 changes: 8 additions & 0 deletions src/types/events/public/EventCallbacks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ import {
MessageStreamEvent,
AnamEvent,
ClientToolEvent,
ReasoningMessage,
ReasoningStreamEvent,
} from '../../index';

export type EventCallbacks = {
Expand All @@ -30,4 +32,10 @@ export type EventCallbacks = {
[AnamEvent.CLIENT_TOOL_EVENT_RECEIVED]: (
clientToolEvent: ClientToolEvent,
) => void;
[AnamEvent.REASONING_HISTORY_UPDATED]: (
thoughtMessages: ReasoningMessage[],
) => void;
[AnamEvent.REASONING_STREAM_EVENT_RECEIVED]: (
thoughtEvent: ReasoningStreamEvent,
) => void;
};
5 changes: 5 additions & 0 deletions src/types/messageHistory/ReasoningMessage.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
export interface ReasoningMessage {
id: string;
content: string;
role: string;
}
6 changes: 6 additions & 0 deletions src/types/messageHistory/ReasoningStreamEvent.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
export interface ReasoningStreamEvent {
id: string;
content: string;
endOfThought: boolean;
role: string;
}
2 changes: 2 additions & 0 deletions src/types/messageHistory/index.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
export { MessageRole } from './MessageRole';
export type { Message } from './Message';
export type { MessageStreamEvent } from './MessageStreamEvent';
export type { ReasoningMessage } from './ReasoningMessage';
export type { ReasoningStreamEvent } from './ReasoningStreamEvent';
1 change: 1 addition & 0 deletions src/types/streaming/DataChannelMessage.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
export enum DataChannelMessage {
SPEECH_TEXT = 'speechText',
CLIENT_TOOL_EVENT = 'clientToolEvent',
REASONING_TEXT = 'reasoningText',
}
7 changes: 7 additions & 0 deletions src/types/streaming/WebRtcReasoningTextMessageEvent.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
export interface WebRtcReasoningTextMessageEvent {
message_id: string;
content_index: number;
content: string;
role: string;
end_of_thought: boolean;
}
1 change: 1 addition & 0 deletions src/types/streaming/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,4 @@ export type { WebRtcClientToolEvent } from './WebRtcClientToolEvent';
export type { StreamingClientOptions } from './StreamingClientOptions';
export type { InputAudioOptions } from './InputAudioOptions';
export { DataChannelMessage } from './DataChannelMessage';
export { WebRtcReasoningTextMessageEvent } from './WebRtcReasoningTextMessageEvent';