@h-ai/ai 0.1.0-alpha.10

package/dist/client/index.d.ts

@@ -0,0 +1,266 @@
+import { ac as ChatCompletionRequest, ad as ChatCompletionResponse, aa as ChatCompletionChunk, aD as KnowledgeRetrieveResult, aq as KnowledgeAskResult, az as KnowledgeIngestResult, ar as KnowledgeDocumentInfo, aJ as MemoryEntry, be as StorePage, aK as MemoryEntryInput, aR as MemoryUpdateInput, bc as SessionInfo, ag as ChatRecord, aY as RagResult, b0 as ReasoningResult, O as A2AAgentCardConfig, _ as A2AMessageRecord, V as A2ACallResult } from '../ai-reasoning-types-CGUgkLF7.js';
+import { HaiResult } from '@h-ai/core';
+import '@a2a-js/sdk/server';
+import '@h-ai/datapipe';
+import 'zod';
+import 'openai';
+
+/**
+ * @h-ai/ai — 前端 AI 客户端
+ *
+ * 基于 @h-ai/api-client 的 AI 领域客户端。
+ * 通过 `createAIClient()` 工厂函数创建，消除自建 HTTP 层，
+ * 复用 api-client 提供的 Token 管理、超时、拦截器等基础能力。
+ * @module ai-client
+ */
+
+/**
+ * AI 客户端所需的 API 调用能力
+ *
+ * 结构兼容 `@h-ai/api-client` 的 `api` 单例（鸭子类型）。
+ * 传入 `api` 单例即可，无需额外适配。
+ */
+interface AIApiAdapter {
+    /** POST 请求（返回 HaiResult） */
+    post: <T>(path: string, body?: unknown) => Promise<HaiResult<T>>;
+    /** 流式请求（返回 SSE data 行的 AsyncIterable） */
+    stream: (path: string, body?: unknown) => AsyncIterable<string>;
+}
+/**
+ * AI 客户端配置
+ *
+ * @example
+ * ```ts
+ * import { api } from '@h-ai/api-client'
+ * import { createAIClient } from '@h-ai/ai/client'
+ *
+ * await api.init({ baseUrl: '/api', auth: { ... } })
+ * const aiClient = createAIClient({ api })
+ * ```
+ */
+interface AIClientConfig {
+    /**
+     * API 调用适配器
+     *
+     * 传入 `api` 单例（初始化后）。
+     * baseUrl、Token 管理、超时等通过 api-client 配置。
+     */
+    api: AIApiAdapter;
+}
+/** 流式响应进度（`onProgress` 回调参数） */
+interface StreamProgress {
+    /** 当前累积的文本内容 */
+    content: string;
+    /** 是否已完成 */
+    done: boolean;
+    /** 完成原因（仅 `done=true` 时有值，如 `'stop'`） */
+    finishReason?: string;
+}
+/** 流式响应选项 */
+interface StreamOptions {
+    /** 进度回调（每次收到 chunk 时触发） */
+    onProgress?: (progress: StreamProgress) => void;
+}
+/**
+ * AI 客户端接口
+ *
+ * 提供浏览器端调用 AI API 的能力，HTTP 基础设施委托给 @h-ai/api-client。
+ */
+interface AIClient {
+    /** 发送对话请求（非流式） */
+    chat: (request: ChatCompletionRequest) => Promise<ChatCompletionResponse>;
+    /** 发送流式对话请求，返回异步迭代器 */
+    chatStream: (request: ChatCompletionRequest, options?: StreamOptions) => AsyncIterable<ChatCompletionChunk>;
+    /** 便捷方法：发送纯文本消息并返回回复文本 */
+    sendMessage: (message: string, systemPrompt?: string) => Promise<string>;
+    /** 便捷方法：流式发送纯文本消息并返回完整回复 */
+    sendMessageStream: (message: string, options?: StreamOptions, systemPrompt?: string) => Promise<string>;
+    /** 简单问答（单轮，自动构造 messages） */
+    ask: (question: string, options?: {
+        systemPrompt?: string;
+        model?: string;
+    }) => Promise<string>;
+    /** 流式简单问答，返回异步文本流 */
+    askStream: (question: string, options?: {
+        systemPrompt?: string;
+        model?: string;
+    }) => AsyncIterable<string>;
+    /** 知识检索（向量 + 实体增强） */
+    knowledgeRetrieve: (query: string, options?: {
+        topK?: number;
+        collection?: string;
+    }) => Promise<KnowledgeRetrieveResult>;
+    /** 知识问答（RAG + 信源） */
+    knowledgeAsk: (query: string, options?: {
+        model?: string;
+        collection?: string;
+    }) => Promise<KnowledgeAskResult>;
+    /** 导入文档 */
+    knowledgeIngest: (input: {
+        documentId: string;
+        content: string;
+        title?: string;
+        collection?: string;
+        metadata?: Record<string, unknown>;
+    }) => Promise<KnowledgeIngestResult>;
+    /** 列出已导入文档 */
+    knowledgeListDocuments: (options?: {
+        collection?: string;
+        offset?: number;
+        limit?: number;
+    }) => Promise<KnowledgeDocumentInfo[]>;
+    /** 删除已导入文档 */
+    knowledgeRemoveDocument: (documentId: string, options?: {
+        collection?: string;
+    }) => Promise<void>;
+    /** 检索相关记忆 */
+    recallMemories: (query: string, options?: {
+        topK?: number;
+        types?: string[];
+        minImportance?: number;
+        objectId?: string;
+    }) => Promise<MemoryEntry[]>;
+    /** 列出记忆（分页） */
+    listMemories: (options?: {
+        types?: string[];
+        objectId?: string;
+        offset?: number;
+        limit?: number;
+    }) => Promise<StorePage<MemoryEntry>>;
+    /** 添加记忆 */
+    addMemory: (entry: MemoryEntryInput) => Promise<MemoryEntry>;
+    /** 更新记忆 */
+    updateMemory: (memoryId: string, updates: MemoryUpdateInput) => Promise<MemoryEntry>;
+    /** 删除记忆 */
+    removeMemory: (memoryId: string) => Promise<void>;
+    /** 列出会话 */
+    listSessions: (objectId: string) => Promise<SessionInfo[]>;
+    /** 重命名会话 */
+    renameSession: (sessionId: string, title: string) => Promise<void>;
+    /** 删除会话 */
+    removeSession: (sessionId: string) => Promise<void>;
+    /** 获取对话历史 */
+    chatHistory: (objectId: string, sessionId: string, options?: {
+        limit?: number;
+        order?: 'asc' | 'desc';
+    }) => Promise<ChatRecord[]>;
+    /** RAG 查询（检索 + 生成） */
+    ragQuery: (query: string, options?: {
+        collection?: string;
+        topK?: number;
+        model?: string;
+    }) => Promise<RagResult>;
+    /** RAG 流式查询，返回 SSE 数据行 */
+    ragQueryStream: (query: string, options?: {
+        collection?: string;
+        topK?: number;
+        model?: string;
+    }) => AsyncIterable<string>;
+    /** 推理（多步思考） */
+    reasoningRun: (query: string, options?: {
+        model?: string;
+        maxSteps?: number;
+    }) => Promise<ReasoningResult>;
+    /** 推理流式，返回 SSE 数据行 */
+    reasoningRunStream: (query: string, options?: {
+        model?: string;
+        maxSteps?: number;
+    }) => AsyncIterable<string>;
+    /** 估算文本的 Token 数 */
+    estimateTokens: (text: string) => Promise<{
+        tokens: number;
+    }>;
+}
+/**
+ * 创建 AI 客户端
+ *
+ * 通过 @h-ai/api-client 的 ApiClient 实例调用后端 AI API。
+ * HTTP 层（Token、超时、拦截器）全部复用 api-client，本模块仅负责：
+ * - 非流式 / 流式 ChatCompletion 协议适配
+ * - SSE 解析与 ChatCompletionChunk 类型化
+ * - onProgress 进度回调
+ *
+ * @param config - 客户端配置
+ * @returns AI 客户端实例
+ *
+ * @example
+ * ```ts
+ * import { api } from '@h-ai/api-client'
+ * import { createAIClient } from '@h-ai/ai/client'
+ *
+ * await api.init({ baseUrl: '/api' })
+ * const client = createAIClient({ api })
+ * const reply = await client.sendMessage('你好')
+ * ```
+ */
+declare function createAIClient(config: AIClientConfig): AIClient;
+/**
+ * 从 fetch Response 中解析 SSE data 字段
+ *
+ * 自动处理分片缓冲和 `[DONE]` 结束标记。
+ *
+ * @param response - fetch 响应对象
+ * @yields SSE data 字符串（不含 `[DONE]`）
+ *
+ * @example
+ * ```ts
+ * const resp = await fetch('/api/chat', { method: 'POST', body })
+ * for await (const data of parseSSE(resp)) {
+ *   const chunk = JSON.parse(data) // ChatCompletionChunk
+ * }
+ * ```
+ */
+declare function parseSSE(response: Response): AsyncIterable<string>;
+/**
+ * 收集流式响应的完整文本内容
+ *
+ * 完整消费流并拼接所有 `delta.content` 片段。
+ *
+ * @param stream - 聊天响应块流
+ * @returns 完整文本
+ *
+ * @example
+ * ```ts
+ * const client = createAIClient({ api })
+ * const stream = client.chatStream({ messages })
+ * const fullText = await collectStreamContent(stream)
+ * ```
+ */
+declare function collectStreamContent(stream: AsyncIterable<ChatCompletionChunk>): Promise<string>;
+
+/**
+ * @h-ai/ai — 前端 A2A 客户端
+ *
+ * 浏览器端 A2A 操作客户端，通过 API 服务调用后端 A2A 能力。
+ * @module ai-a2a-client
+ */
+
+/**
+ * A2A 客户端接口
+ *
+ * 提供浏览器端调用后端 A2A API 的能力。
+ */
+interface A2AClientOperations {
+    /** 获取当前 Agent Card 配置 */
+    getAgentCard: () => Promise<A2AAgentCardConfig>;
+    /** 查询 A2A 消息记录 */
+    listMessages: (filter?: {
+        contextId?: string;
+        status?: string;
+        limit?: number;
+        offset?: number;
+    }) => Promise<StorePage<A2AMessageRecord>>;
+    /** 作为客户端调用远端 Agent */
+    callRemoteAgent: (remoteUrl: string, message: string, options?: {
+        timeout?: number;
+    }) => Promise<A2ACallResult>;
+}
+/**
+ * 创建 A2A 客户端
+ *
+ * @param api - API 适配器（来自 @h-ai/api-client）
+ * @returns A2A 客户端操作实例
+ */
+declare function createA2AClient(api: AIApiAdapter): A2AClientOperations;
+
+export { type A2AClientOperations, type AIApiAdapter, type AIClient, type AIClientConfig, type StreamOptions, type StreamProgress, collectStreamContent, createA2AClient, createAIClient, parseSSE };

package/dist/client/index.js

@@ -0,0 +1,3 @@
+export { collectStreamContent, createA2AClient, createAIClient, parseSSE } from '../chunk-ABP7RKIP.js';
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/client/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"index.js"}

package/dist/index.d.ts

@@ -0,0 +1,40 @@
+export { O as A2AAgentCardConfig, P as A2AApiKeySecurity, Q as A2AAuthenticator, U as A2ACallOptions, V as A2ACallResult, W as A2ACallerIdentity, X as A2AClientCallRecord, A as A2AConfig, a as A2AConfigSchema, Y as A2AContextInfo, Z as A2AHandleResult, _ as A2AMessageRecord, $ as A2AOperations, a0 as A2ASecurityConfig, b as A2ASkillConfigSchema, a1 as A2ATaskFilter, c as AIConfig, d as AIConfigInput, e as AIConfigSchema, a2 as AILLMFunctionsDeps, a3 as AIRelStore, a4 as AIRelStoreOptions, a5 as AIStoreProvider, a6 as AIVectorStore, a7 as AskOptions, a8 as AssistantMessage, a9 as ChatCompletionChoice, aa as ChatCompletionChunk, ab as ChatCompletionDelta, ac as ChatCompletionRequest, ad as ChatCompletionResponse, ae as ChatHistoryOptions, af as ChatMessage, ag as ChatRecord, ah as Citation, C as CompressConfig, f as CompressConfigSchema, ai as DefineToolOptions, E as EmbeddingConfig, g as EmbeddingConfigSchema, aj as EntityDocumentRelation, ak as EntityDocumentResult, al as EntityListOptions, am as EntityQueryOptions, h as EntityType, i as EntityTypeSchema, F as FileConfig, j as FileConfigSchema, an as ImageContent, ao as InteractionScope, ap as KnowledgeAskOptions, aq as KnowledgeAskResult, K as KnowledgeConfig, k as KnowledgeConfigSchema, ar as KnowledgeDocumentInfo, as as KnowledgeDocumentListOptions, at as KnowledgeDocumentRemoveOptions, au as KnowledgeEntity, av as KnowledgeIngestBatchProgress, aw as KnowledgeIngestBatchResult, ax as KnowledgeIngestFileInput, ay as KnowledgeIngestInput, az as KnowledgeIngestResult, aA as KnowledgeOperations, aB as KnowledgeRetrieveItem, aC as KnowledgeRetrieveOptions, aD as KnowledgeRetrieveResult, aE as KnowledgeSetupOptions, aF as KnowledgeStore, L as LLMConfig, l as LLMConfigSchema, aG as LLMOperations, aH as LLMProvider, M as MCPConfig, m as MCPConfigSchema, n as MCPServerCapabilities, o as MCPServerCapabilitiesSchema, p as MCPServerConfig, q as MCPServerConfigSchema, aI as MemoryClearOptions, r as MemoryConfig, s as MemoryConfigSchema, aJ as MemoryEntry, aK as MemoryEntryInput, aL as MemoryExtractOptions, aM as MemoryInjectionOptions, aN as MemoryListOptions, aO as MemoryListPageOptions, aP as MemoryOperations, aQ as MemoryRecallOptions, t as MemoryType, u as MemoryTypeSchema, aR as MemoryUpdateInput, aS as MessageContent, aT as MessageRole, v as ModelEntry, w as ModelEntrySchema, x as ModelScenario, y as ModelScenarioSchema, aU as ObjectRef, aV as RagContextItem, aW as RagOperations, aX as RagOptions, aY as RagResult, aZ as RagStreamEvent, a_ as ReasoningOperations, a$ as ReasoningOptions, b0 as ReasoningResult, b1 as ReasoningStep, b2 as ReasoningStepType, b3 as ReasoningStrategy, b4 as ReasoningStreamEvent, R as ResolveRequiredModelEntryOptions, z as ResolvedModelConfig, B as RetrievalConfig, D as RetrievalConfigSchema, b5 as RetrievalOperations, b6 as RetrievalRequest, b7 as RetrievalResult, b8 as RetrievalResultItem, b9 as RetrievalSource, G as RetrievalSourceConfig, H as RetrievalSourceSchema, ba as SSEDecoder, bb as SSEEvent, bc as SessionInfo, bd as StoreFilter, be as StorePage, bf as StoreScope, bg as StreamOperations, bh as StreamProcessor, bi as StreamResult, S as SummaryConfig, I as SummaryConfigSchema, bj as SystemMessage, bk as TextContent, T as TokenConfig, J as TokenConfigSchema, bl as TokenUsage, bm as Tool, bn as ToolCall, bo as ToolDefinition, bp as ToolErrorType, bq as ToolMessage, br as ToolRegistryOperations, bs as ToolsOperations, bt as UserMessage, bu as WhereClause, bv as WhereOperator, bw as WhereValue, N as resolveModelEntry } from './ai-reasoning-types-CGUgkLF7.js';
+import { A as AIFunctions } from './ai-types-CRCjluXI.js';
+export { a as AIInitOptions, c as AIMCPFunctionsDeps, d as CompressOperations, e as CompressOptions, f as CompressResult, C as CompressionStrategy, b as CompressionStrategySchema, g as ContextChatOptions, h as ContextChatResult, i as ContextDeps, j as ContextManager, k as ContextManagerOptions, l as ContextOperations, m as ContextStreamEvent, E as EmbeddingItem, n as EmbeddingOperations, o as EmbeddingProvider, p as EmbeddingRequest, q as EmbeddingResponse, F as FileOperations, r as FileParseMethod, s as FileParseOptions, t as FileParseRequest, u as FileParseResult, H as HaiAIError, M as MCPContext, v as MCPOperations, w as MCPPrompt, x as MCPPromptArgument, y as MCPPromptContent, z as MCPPromptMessage, B as MCPProvider, D as MCPResource, G as MCPResourceContent, I as MCPToolDefinition, J as MCPToolHandler, K as McpServerOptions, O as OutputFormat, R as RerankDocument, L as RerankItem, N as RerankOperations, P as RerankRequest, Q as RerankResponse, S as SummaryOperations, T as SummaryOptions, U as SummaryResult, V as TokenOperations } from './ai-types-CRCjluXI.js';
+export { TaskStore as A2ATaskStore, AgentExecutor, ExecutionEventBus, RequestContext, ServerCallContext } from '@a2a-js/sdk/server';
+export { AgentCapabilities, AgentCard, AgentProvider, AgentSkill, Artifact, DataPart, FilePart, FileWithBytes, FileWithUri, Message, MessageSendParams, Part, PushNotificationConfig, Task, TaskArtifactUpdateEvent, TaskIdParams, TaskQueryParams, TaskState, TaskStatus, TaskStatusUpdateEvent, TextPart } from '@a2a-js/sdk';
+import '@h-ai/core';
+import '@h-ai/datapipe';
+import 'zod';
+import 'openai';
+import 'node:buffer';
+
+/**
+ * @h-ai/ai — AI 服务主入口
+ *
+ * 提供统一的 `ai` 对象，聚合所有 AI 操作功能。
+ * @module ai-main
+ */
+
+/**
+ * AI 服务对象，统一的 AI 访问入口
+ *
+ * @example
+ * ```ts
+ * import { ai } from '@h-ai/ai'
+ *
+ * // 初始化
+ * await ai.init({ llm: { model: 'gpt-4o-mini', apiKey: process.env.HAI_AI_LLM_API_KEY } })
+ *
+ * // LLM 调用
+ * const result = await ai.llm.chat({
+ *   messages: [{ role: 'user', content: '你好' }],
+ * })
+ *
+ * // 关闭
+ * ai.close()
+ * ```
+ */
+declare const ai: AIFunctions;
+
+export { AIFunctions, ai };