@celljs/ai-anthropic

Cell - AI Anthropic Component

概览

AI Anthropic 模块是一个用于与 Anthropic API 交互的库，提供了生成聊天响应和嵌入向量的功能。通过简单易用的 API 接口，支持消息的创建、请求的发送和响应的处理。是 @celljs/ai-core 模块中所有模型服务接口抽象的一种实现。

特性

• 生成聊天响应
• 生成嵌入向量
• 支持流式响应
• 支持多种模型参数配置

安装

使用 npm 安装 AI Anthropic 模块：

npm install @celljs/ai-anthropic

或者使用 yarn：

yarn add @celljs/ai-anthropic

快速开始

以下是一个简单的示例，展示如何使用 AI Anthropic 模块生成聊天响应：

import { AssistantMessage, PromptTemplate } from '@celljs/ai-core';
import { OllamChatModel, OllamModel, } from '@celljs/ai-anthropic';
import { Component Autowired } from '@celljs/core';

@Component()
export class AnthropicDemo {
    @Autowired(AnthropicChatModel)
    private anthropicChatModel: AnthropicChatModel;

    @Autowired(PromptTemplate)
    private promptTemplate: PromptTemplate;

    /**
     * Chat with Anthropic
     */
    async chat() {
        const prompt = await this.promptTemplate.create(
            'Hello {name}',
            { 
                chatOptions: { model: AnthropicModel.CLAUDE_3_5_SONNET },
                variables: { name: 'Anthropic' }
            }
        );
        const response = await this.anthropicChatModel.call(prompt);
        console.log(response.result.output);
    }

    /**
     * Stream chat response
     */
    async stream() {
        const prompt = await this.promptTemplate.create(
            'Hello {name}',
            { 
                chatOptions: { model: AnthropicModel.CLAUDE_3_5_SONNET },
                variables: { name: 'Anthropic' }
            }
        );
        const response$ = await this.anthropicChatModel.stream(prompt);
        response$.subscribe({
            next: response => console.log(response.result.output),
            complete: () => console.log('Chat completed!')
        });
    }
}

许可证

本项目采用 MIT 许可证。