#4 refactored code to fix issue with close-ended chat + added tests

[pranav-kural]

Added Jest tests for close-ended chat endpoint.

Also:

Refactored the getFormattedInput method to accept object instead of positional arguments.

/**
   * Gets the formatted input based on the agent type.
   * @param agentType - The type of agent.
   * @param topic - Topic for close-ended and rag agents.
   * @param query - The query string.
   * @param context - The context object.
   * @returns Returns the formatted input.
   * @throws Throws an error if the agent type is invalid.
   */
  private static getFormattedInput({
    agentType,
    query,
    context,
    topic,
  }: {
    agentType?: ChatAgentType;
    query?: string;
    topic?: string;
    context?: string;
  }) {
    switch (agentType) {
      case "open-ended":
        return {
          query,
        };
      case "close-ended":
        return {
          query,
          topic: topic,
        };
      case "rag":
        return {
          query,
          topic: topic,
          context,
        };
      default:
        throw new Error("Invalid agent type");
    }
  }

  /**
   * Generates a response using the system prompt.
   * @param agentType - The type of agent.
   * @param model - The supported model name.
   * @param modelConfig - The model configuration.
   * @param query - The query string.
   * @param context - The context string.
   * @param topic - The topic string.
   * @param tools - The tool arguments.
   * @returns Returns the generated response.
   */
  private static generateSystemPromptResponse({
    agentType,
    prompt,
    model,
    modelConfig,
    query,
    context,
    topic,
    tools,
  }: GenerateSystemPromptResponseParams) {
    // generate the response
    const res = prompt.generate({
      // if undefined, will use model defined in the dotprompt
      model: model,
      config: modelConfig,
      input: ChatAgent.getFormattedInput({ agentType, query, context, topic }),
      tools: tools,
    });
    // return the response
    return res;
  }

[github-actions]

Dependency Review

✅ No vulnerabilities or license issues or OpenSSF Scorecard issues found.

OpenSSF Scorecard

Package	Version	Score	Details

Scanned Manifest Files