@future-agi/sdk/dist/esm/prompt/types.js

We help GenAI teams maintain high-accuracy for their Models in production.

export class MessageBase {
    constructor(role, content) {
        this.role = role;
        this.content = content;
    }
}
export class UserMessage extends MessageBase {
    constructor(content, variable_names) {
        super('user', content);
        this.role = 'user';
        this.variable_names = variable_names;
    }
}
export class SystemMessage extends MessageBase {
    constructor(content) {
        super('system', content);
        this.role = 'system';
    }
}
export class AssistantMessage extends MessageBase {
    constructor(content) {
        super('assistant', content);
        this.role = 'assistant';
    }
}
export class ModelConfig {
    constructor(config = {}) {
        this.model_name = config.model_name ?? 'gpt-4o-mini';
        this.temperature = config.temperature ?? 0.7;
        this.frequency_penalty = config.frequency_penalty ?? 0;
        this.presence_penalty = config.presence_penalty ?? 0;
        this.max_tokens = config.max_tokens;
        this.top_p = config.top_p ?? 1.0;
        this.response_format = config.response_format ?? null;
        this.tool_choice = config.tool_choice ?? null;
        this.tools = config.tools ?? null;
    }
}
export class PromptTemplate {
    constructor(template = {}) {
        this.messages = template.messages || [];
        Object.assign(this, template);
    }
}
//# sourceMappingURL=types.js.map