261 lines
9.4 KiB
TypeScript
261 lines
9.4 KiB
TypeScript
import type { Message } from '../types/types';
|
||
import { CozeAPI } from '@coze/api';
|
||
|
||
|
||
// 定义Coze事件数据接口
|
||
interface DeltaEventData {
|
||
content?: string;
|
||
delta?: { content?: string };
|
||
}
|
||
|
||
interface CompletedEventData {
|
||
conversation_id?: string;
|
||
usage?: { token_count?: number };
|
||
token_count?: number;
|
||
messages?: { content?: string }[];
|
||
}
|
||
|
||
// 错误事件数据接口 - 备用
|
||
// interface ErrorEventData {
|
||
// code?: string;
|
||
// message?: string;
|
||
// type?: string;
|
||
// }
|
||
|
||
class CozeService {
|
||
private apiKey: string | null = null;
|
||
private botId: string | null = null;
|
||
private userId: string = `user_${Date.now()}`; // 生成一个临时用户ID
|
||
private apiClient: CozeAPI | null = null;
|
||
private conversationId: string | null = null; // 存储对话ID，用于后续调用
|
||
|
||
constructor() {
|
||
// 初始化服务，实际使用时需要设置API密钥和Bot ID
|
||
}
|
||
|
||
/**
|
||
* 设置API密钥和Bot ID
|
||
* @param apiKey Coze API密钥
|
||
* @param botId Bot ID
|
||
*/
|
||
setCredentials(apiKey: string, botId: string): void {
|
||
this.apiKey = apiKey;
|
||
this.botId = botId;
|
||
|
||
// 初始化Coze API客户端 - 严格按照官方示例实现
|
||
this.apiClient = new CozeAPI({
|
||
token: apiKey,
|
||
baseURL: 'https://api.coze.cn'
|
||
});
|
||
|
||
console.log('Coze API client initialized with official SDK');
|
||
}
|
||
|
||
/**
|
||
* 发送消息给Coze agent
|
||
* 使用官方SDK的stream方法实现流式响应
|
||
* 严格按照api.ts示例实现，使用for await...of循环处理流式响应
|
||
* @param message 用户消息内容
|
||
* @param onStream 流式响应回调函数
|
||
* @returns Promise<Message> 完整的响应消息
|
||
*/
|
||
async sendMessage(
|
||
message: string,
|
||
onStream?: (partialText: string, isDone: boolean) => void
|
||
): Promise<Message> {
|
||
// 检查初始化状态
|
||
if (!this.isInitialized()) {
|
||
console.warn('Coze服务尚未初始化或凭据不完整。使用模拟回复。');
|
||
return this.getMockResponse(message);
|
||
}
|
||
|
||
try {
|
||
console.log('正在使用官方SDK发送消息到Coze服务...');
|
||
console.log('Bot ID:', this.botId);
|
||
console.log('User ID:', this.userId);
|
||
|
||
// 严格按照api.ts示例调用chat.stream方法
|
||
const res = await this.apiClient!.chat.stream({
|
||
bot_id: this.botId!,
|
||
user_id: this.userId,
|
||
conversation_id: this.conversationId || undefined, // 如果有保存的对话ID则使用
|
||
additional_messages: [
|
||
{
|
||
"content": message,
|
||
"content_type": "text",
|
||
"role": "user",
|
||
"type": "question"
|
||
}
|
||
] as any // eslint-disable-line @typescript-eslint/no-explicit-any,
|
||
});
|
||
|
||
let fullContent = '';
|
||
const messageId = Date.now().toString();
|
||
|
||
console.log('开始处理Coze流式响应...');
|
||
|
||
// 严格按照api.ts的实现方式，使用for await...of循环处理流式响应
|
||
for await (const event of res) {
|
||
try {
|
||
console.log('接收到Coze事件:', event.event || 'unknown');
|
||
|
||
// 1. 处理消息增量事件
|
||
if (event.event === 'conversation.message.delta') {
|
||
// 根据TypeScript类型定义，content可能直接在data上
|
||
const deltaData = event.data as DeltaEventData;
|
||
if (deltaData && deltaData.content) {
|
||
const deltaContent = deltaData.content;
|
||
fullContent += deltaContent;
|
||
console.log('增量内容:', fullContent.substring(0, 100) + '...');
|
||
|
||
// 通知流式更新
|
||
if (onStream) {
|
||
onStream(fullContent, false);
|
||
}
|
||
} else if (deltaData && deltaData.delta && deltaData.delta.content) {
|
||
// 兼容另一种可能的数据结构
|
||
const deltaContent = deltaData.delta.content;
|
||
fullContent += deltaContent;
|
||
console.log('增量内容(delta路径):', fullContent.substring(0, 100) + '...');
|
||
|
||
if (onStream) {
|
||
onStream(fullContent, false);
|
||
}
|
||
}
|
||
}
|
||
|
||
// 2. 处理聊天完成事件
|
||
if (event.event === 'conversation.chat.completed') {
|
||
console.log(); // 输出换行
|
||
// 尝试从data中获取conversation_id和usage信息
|
||
const dataObj = event.data as CompletedEventData;
|
||
|
||
// 打印conversation_id并保存
|
||
if (dataObj && dataObj.conversation_id) {
|
||
console.log('Conversation ID:', dataObj.conversation_id);
|
||
this.conversationId = dataObj.conversation_id; // 保存对话ID供下次使用
|
||
}
|
||
|
||
// 打印token使用量
|
||
if (dataObj && dataObj.usage && dataObj.usage.token_count) {
|
||
console.log('token usage:', dataObj.usage.token_count);
|
||
} else if (dataObj && dataObj.token_count) {
|
||
console.log('token usage:', dataObj.token_count);
|
||
}
|
||
|
||
// 尝试从data中获取完整内容（如果有的话）
|
||
if (dataObj && dataObj.messages && dataObj.messages[0] && dataObj.messages[0].content) {
|
||
fullContent = dataObj.messages[0].content || '';
|
||
console.log('使用完成事件中的完整内容');
|
||
}
|
||
}
|
||
|
||
// 3. 处理错误事件
|
||
if (event.event === 'error') {
|
||
console.error('错误事件:', event.data);
|
||
throw new Error(`Coze服务错误: ${JSON.stringify(event.data)}`);
|
||
}
|
||
} catch (err) {
|
||
console.error('处理事件时出错:', err);
|
||
console.log('错误事件完整数据:', JSON.stringify(event, null, 2));
|
||
}
|
||
}
|
||
|
||
// 流式处理完成，检查是否有内容
|
||
console.log('Coze流式响应处理完成，总内容长度:', fullContent.length);
|
||
|
||
if (fullContent.trim()) {
|
||
const botMessage: Message = {
|
||
id: messageId,
|
||
content: fullContent,
|
||
role: 'assistant',
|
||
content_type: 'text',
|
||
timestamp: new Date()
|
||
};
|
||
|
||
// 通知流式更新完成
|
||
if (onStream) {
|
||
onStream(fullContent, true);
|
||
}
|
||
|
||
return botMessage;
|
||
} else {
|
||
throw new Error('Coze服务返回空内容，请检查API配置和网络连接');
|
||
}
|
||
} catch (error) {
|
||
console.error('Error sending message to Coze with official SDK:', error);
|
||
// 获取具体的错误信息
|
||
const errorMsg = error instanceof Error ? error.message : '未知错误';
|
||
|
||
// 即使出错也返回模拟回复
|
||
return this.getMockResponse(message, errorMsg);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取模拟回复，当Coze API调用失败时使用
|
||
*/
|
||
private getMockResponse(message: string, error?: string): Message {
|
||
let mockContent = '';
|
||
|
||
// 根据用户消息内容生成更有针对性的模拟回复
|
||
if (message.includes('你好') || message.includes('hello') || message.includes('hi')) {
|
||
mockContent = '你好！我是你的爆款文案策划师。为了给你提供更有针对性的方案，请告诉我：\n1. 你的行业/产品/服务\n2. 你的目标用户群体\n有了这些信息，我就能帮你打造更精准的文案了！';
|
||
} else if (message.includes('文案') || message.includes('内容') || message.includes('写作')) {
|
||
mockContent = '关于文案创作，我有一些专业建议：\n1. 明确目标受众和核心卖点\n2. 使用吸引人的标题和开头\n3. 突出产品/服务的独特价值\n4. 加入社会证明（案例、数据）\n5. 包含明确的行动号召\n需要针对特定行业的文案模板吗？';
|
||
} else if (message.includes('价格') || message.includes('费用') || message.includes('多少钱')) {
|
||
mockContent = '文案服务的价格因需求复杂度而异。我们提供：\n- 基础文案套餐：¥500-1000\n- 营销策划套餐：¥1000-3000\n- 品牌全案套餐：¥3000+\n欢迎告诉我你的具体需求，我可以为你提供更精准的报价。';
|
||
} else {
|
||
mockContent = '感谢你的提问！作为文案策划师，我可以帮你：\n- 撰写吸引人的营销文案\n- 设计爆款标题\n- 优化产品描述\n- 制定内容营销策略\n请告诉我你具体需要哪方面的帮助？';
|
||
}
|
||
|
||
// 如果有错误信息，可以选择性地在开头添加提示
|
||
if (error) {
|
||
console.log('使用模拟回复，原因：', error);
|
||
// 不直接显示错误，只显示友好提示
|
||
mockContent = '【提示：当前使用的是智能助手的推荐回复】\n\n' + mockContent;
|
||
}
|
||
|
||
return {
|
||
id: Date.now().toString(),
|
||
content: mockContent,
|
||
role: 'assistant',
|
||
content_type: 'text',
|
||
timestamp: new Date()
|
||
};
|
||
}
|
||
|
||
/**
|
||
* 重置对话状态
|
||
*/
|
||
resetConversation(): void {
|
||
this.conversationId = null; // 清除保存的对话ID
|
||
console.log('Conversation reset');
|
||
}
|
||
|
||
/**
|
||
* 检查客户端是否已初始化
|
||
*/
|
||
isInitialized(): boolean {
|
||
return !!this.apiClient && !!this.apiKey && !!this.botId;
|
||
}
|
||
|
||
/**
|
||
* 设置用户ID
|
||
*/
|
||
setUserId(userId: string): void {
|
||
this.userId = userId;
|
||
}
|
||
|
||
/**
|
||
* 获取用户ID
|
||
*/
|
||
getUserId(): string {
|
||
return this.userId;
|
||
}
|
||
}
|
||
|
||
// 导出单例实例
|
||
// 创建并导出单例实例
|
||
const cozeServiceInstance = new CozeService();
|
||
export default cozeServiceInstance; |