function-manager.ts

import APIClient from './api-client';
import { LLMMessage, Metadata } from './index';

class FunctionManager {
  private apiClient: APIClient;

  constructor() {
    this.apiClient = new APIClient();
  }

  async maskPrompt(messages: LLMMessage[], metadata?: Metadata): Promise<any> {
    try {
      const response = await this.apiClient.request({
        method: 'post',
        url: '/mask/prompt',
        data: JSON.stringify({ messages, metadata }),
      });
      return response;
    } catch (err: any) {
      throw new Error(err.response?.data?.error || err.message);
    }
  }

  async logError(error: any, messages: LLMMessage[], metadata?: Metadata): Promise<any> {
    try {
      const response = await this.apiClient.request({
        method: 'post',
        url: '/log/error',
        data: JSON.stringify({ error, messages, metadata }),
      });
      return response;
    } catch (err: any) {
      throw new Error(err.response?.data?.error || err.message);
    }
  }

  async executeGuardrails(guardrails: string[], messages: LLMMessage[], untrusted_input?: string, metadata?: Metadata): Promise<any> {
    try {
      const response = await this.apiClient.request({
        method: 'post',
        url: '/guardrails/execute',
        data: JSON.stringify({ guardrails, messages, untrusted_input, metadata }),
      });
      return response;
    } catch (err: any) {
      throw new Error(err.response?.data?.error || err.message);
    }
  }
}

export default FunctionManager;