@future-agi/sdk
Version:
We help GenAI teams maintain high-accuracy for their Models in production.
55 lines • 2.14 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.PromptTemplate = exports.ModelConfig = exports.AssistantMessage = exports.SystemMessage = exports.UserMessage = exports.MessageBase = void 0;
class MessageBase {
constructor(role, content) {
this.role = role;
this.content = content;
}
}
exports.MessageBase = MessageBase;
class UserMessage extends MessageBase {
constructor(content, variable_names) {
super('user', content);
this.role = 'user';
this.variable_names = variable_names;
}
}
exports.UserMessage = UserMessage;
class SystemMessage extends MessageBase {
constructor(content) {
super('system', content);
this.role = 'system';
}
}
exports.SystemMessage = SystemMessage;
class AssistantMessage extends MessageBase {
constructor(content) {
super('assistant', content);
this.role = 'assistant';
}
}
exports.AssistantMessage = AssistantMessage;
class ModelConfig {
constructor(config = {}) {
var _a, _b, _c, _d, _e, _f, _g, _h;
this.model_name = (_a = config.model_name) !== null && _a !== void 0 ? _a : 'gpt-4o-mini';
this.temperature = (_b = config.temperature) !== null && _b !== void 0 ? _b : 0.7;
this.frequency_penalty = (_c = config.frequency_penalty) !== null && _c !== void 0 ? _c : 0;
this.presence_penalty = (_d = config.presence_penalty) !== null && _d !== void 0 ? _d : 0;
this.max_tokens = config.max_tokens;
this.top_p = (_e = config.top_p) !== null && _e !== void 0 ? _e : 1.0;
this.response_format = (_f = config.response_format) !== null && _f !== void 0 ? _f : null;
this.tool_choice = (_g = config.tool_choice) !== null && _g !== void 0 ? _g : null;
this.tools = (_h = config.tools) !== null && _h !== void 0 ? _h : null;
}
}
exports.ModelConfig = ModelConfig;
class PromptTemplate {
constructor(template = {}) {
this.messages = template.messages || [];
Object.assign(this, template);
}
}
exports.PromptTemplate = PromptTemplate;
//# sourceMappingURL=types.js.map