daggerai/dist/src/llm/anthropic.js

A simple and powerful Typescript based agent framework to help businesses thrive in the AI Agent revolution.

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChatAnthropic = void 0;
const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
const prompts_1 = require("../core/prompts");
class ChatAnthropic {
    api;
    params;
    constructor(params) {
        this.api = new sdk_1.default({
            apiKey: params.apiKey || process.env.ANTHROPIC_API_KEY,
        });
        this.params = params;
    }
    async invoke(prompt) {
        const completion = await this.api.messages.create({
            model: this.params.model,
            max_tokens: this.params.maxTokens || 4000,
            stop_sequences: [prompts_1.SQUAD_PROMPTS.observation],
            temperature: 0,
            messages: [{ role: 'user', content: prompt }],
        });
        return completion.content[0]?.text || '';
    }
}
exports.ChatAnthropic = ChatAnthropic;
//# sourceMappingURL=anthropic.js.map