koishi-plugin-chatluna-qwen-adapter
Version:
qwen adapter for chatluna
10 lines (9 loc) • 1.13 kB
TypeScript
import { AIMessageChunk, BaseMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, MessageType, SystemMessageChunk, ToolMessageChunk } from '@langchain/core/messages';
import { StructuredTool } from '@langchain/core/tools';
import { ChatCompletionResponseMessage, ChatCompletionResponseMessageRoleEnum, ChatCompletionTool } from './types';
import { ChatLunaPlugin } from 'koishi-plugin-chatluna/services/chat';
export declare function formatToolsToQWenTools(tools: StructuredTool[]): ChatCompletionTool[];
export declare function formatToolToQWenTool(tool: StructuredTool): ChatCompletionTool;
export declare function langchainMessageToQWenMessage(messages: BaseMessage[], plugin: ChatLunaPlugin, model: string): Promise<ChatCompletionResponseMessage[]>;
export declare function messageTypeToQWenRole(type: MessageType): ChatCompletionResponseMessageRoleEnum;
export declare function convertDeltaToMessageChunk(delta: Record<string, any>, defaultRole?: ChatCompletionResponseMessageRoleEnum): AIMessageChunk | HumanMessageChunk | SystemMessageChunk | FunctionMessageChunk | ToolMessageChunk | ChatMessageChunk;