Class AIClient

SDK for interacting with the AI Services API's such as Text Generation and Text to SQL

import { AIClient } from '@domoinc/toolkit';

interface PromptTemplate {
     template: string;
}

interface Context {
     input: string;
     promptTemplate?: PromptTemplate;
     parameters?: Record<string, string>;
     model?: string;
     modelConfiguration?: Record<string, Object>;
}
const aiClient = new AIClient<Context>();

const generateText = async (input: string, promptTemplate?: PromptTemplate, parameters?: Record<string, string>, model?: string, modelConfiguration?: Record<string, Object>) =>
    aiClient.generate_text(input, promptTemplate, parameters, model, modelConfiguration);

import { AIClient } from '@domoinc/toolkit';

interface PromptTemplate {
     template: string;
}

interface DataSourceSchema {
     dataSourceName: string;
     description: string;
     columns: Column[];
}


const aiClient = new AIClient<aiClient>();

const textToSQL = async (input: string, dataSourceSchemas: DataSourceSchema[], promptTemplate?: PromptTemplate, parameters?: Record<string, string>, model?: string, modelConfiguration?: Record<string, Object>) =>
   aiClient.text_to_sql(input, dataSourceSchemas, promptTemplate, parameters, model, modelConfiguration);