Advisors API

import {
  ChatClientMessageAggregator,
  type CallAdvisor,
  type CallAdvisorChain,
  type ChatClientRequest,
  type ChatClientResponse,
  type StreamAdvisor,
  type StreamAdvisorChain,
} from '@nestjs-ai/client-chat';
import { LoggerFactory, type Logger } from '@nestjs-port/core';
import type { Observable } from 'rxjs';

export class SimpleLoggerAdvisor implements CallAdvisor, StreamAdvisor {
  private readonly logger: Logger = LoggerFactory.getLogger(
    SimpleLoggerAdvisor.name,
  );

  private readonly aggregator = new ChatClientMessageAggregator();

  get name(): string { (1)
    return this.constructor.name;
  }

  get order(): number { (2)
    return 0;
  }

  async adviseCall(
    chatClientRequest: ChatClientRequest,
    callAdvisorChain: CallAdvisorChain,
  ): Promise<ChatClientResponse> {
    this.logRequest(chatClientRequest);

    const chatClientResponse = await callAdvisorChain.nextCall(chatClientRequest);

    this.logResponse(chatClientResponse);

    return chatClientResponse;
  }

  adviseStream(
    chatClientRequest: ChatClientRequest,
    streamAdvisorChain: StreamAdvisorChain,
  ): Observable<ChatClientResponse> {
    this.logRequest(chatClientRequest);

    const chatClientResponses = streamAdvisorChain.nextStream(chatClientRequest);

    return this.aggregator.aggregateChatClientResponse( (3)
      chatClientResponses,
      (chatClientResponse) => this.logResponse(chatClientResponse),
    );
  }

  private logRequest(request: ChatClientRequest): void {
    this.logger.debug(`request: ${JSON.stringify(request)}`);
  }

  private logResponse(chatClientResponse: ChatClientResponse): void {
    this.logger.debug(`response: ${JSON.stringify(chatClientResponse)}`);
  }
}

{Input_Query}
Read the question again: {Input_Query}

import {
  BaseAdvisor,
  type AdvisorChain,
  type ChatClientRequest,
  type ChatClientResponse,
} from '@nestjs-ai/client-chat';
import { PromptTemplate } from '@nestjs-ai/model';

export class ReReadingAdvisor extends BaseAdvisor {
  private static readonly DEFAULT_RE2_ADVISE_TEMPLATE = `{re2_input_query}
Read the question again: {re2_input_query}`;

  private readonly re2AdviseTemplate: string;

  private _order = 0;

  constructor(
    re2AdviseTemplate: string = ReReadingAdvisor.DEFAULT_RE2_ADVISE_TEMPLATE,
  ) {
    super();
    this.re2AdviseTemplate = re2AdviseTemplate;
  }

  override async before( (1)
    chatClientRequest: ChatClientRequest,
    _advisorChain: AdvisorChain,
  ): Promise<ChatClientRequest> {
    const augmentedUserText = PromptTemplate.builder()
      .template(this.re2AdviseTemplate)
      .variables({
        re2_input_query: chatClientRequest.prompt.userMessage.text ?? '',
      })
      .build()
      .render();

    return chatClientRequest
      .mutate()
      .prompt(chatClientRequest.prompt.augmentUserMessage(augmentedUserText))
      .build();
  }

  override async after(
    chatClientResponse: ChatClientResponse,
    _advisorChain: AdvisorChain,
  ): Promise<ChatClientResponse> {
    return chatClientResponse;
  }

  override get order(): number { (2)
    return this._order;
  }

  withOrder(order: number): this {
    this._order = order;
    return this;
  }
}