dtamind-components
Version:
DTAmindai Components
30 lines • 1.11 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChatAnthropic = void 0;
const anthropic_1 = require("@langchain/anthropic");
const DEFAULT_IMAGE_MODEL = 'claude-3-5-haiku-latest';
const DEFAULT_IMAGE_MAX_TOKEN = 2048;
class ChatAnthropic extends anthropic_1.ChatAnthropic {
constructor(id, fields) {
// @ts-ignore
super(fields ?? {});
this.id = id;
this.configuredModel = fields?.modelName || '';
this.configuredMaxToken = fields?.maxTokens ?? 2048;
}
revertToOriginalModel() {
this.modelName = this.configuredModel;
this.maxTokens = this.configuredMaxToken;
}
setMultiModalOption(multiModalOption) {
this.multiModalOption = multiModalOption;
}
setVisionModel() {
if (!this.modelName.startsWith('claude-3')) {
this.modelName = DEFAULT_IMAGE_MODEL;
this.maxTokens = this.configuredMaxToken ? this.configuredMaxToken : DEFAULT_IMAGE_MAX_TOKEN;
}
}
}
exports.ChatAnthropic = ChatAnthropic;
//# sourceMappingURL=FlowiseChatAnthropic.js.map