34 lines
1.2 KiB
TypeScript
34 lines
1.2 KiB
TypeScript
import { AnthropicInput, ChatAnthropic as LangchainChatAnthropic } from '@langchain/anthropic'
|
|
import { IVisionChatModal, IMultiModalOption } from '../../../src'
|
|
import { BaseLLMParams } from '@langchain/core/language_models/llms'
|
|
|
|
export class ChatAnthropic extends LangchainChatAnthropic implements IVisionChatModal {
|
|
configuredModel: string
|
|
configuredMaxToken: number
|
|
multiModalOption: IMultiModalOption
|
|
id: string
|
|
|
|
constructor(id: string, fields: Partial<AnthropicInput> & BaseLLMParams & { anthropicApiKey?: string }) {
|
|
super(fields)
|
|
this.id = id
|
|
this.configuredModel = fields?.modelName || 'claude-3-opus-20240229'
|
|
this.configuredMaxToken = fields?.maxTokens ?? 256
|
|
}
|
|
|
|
revertToOriginalModel(): void {
|
|
super.modelName = this.configuredModel
|
|
super.maxTokens = this.configuredMaxToken
|
|
}
|
|
|
|
setMultiModalOption(multiModalOption: IMultiModalOption): void {
|
|
this.multiModalOption = multiModalOption
|
|
}
|
|
|
|
setVisionModel(): void {
|
|
if (!this.modelName.startsWith('claude-3')) {
|
|
super.modelName = 'claude-3-opus-20240229'
|
|
super.maxTokens = 1024
|
|
}
|
|
}
|
|
}
|