@fe-free/ai 6.0.23 → 6.0.25

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @fe-free/ai
 
+## 6.0.25
+
+### Patch Changes
+
+- sender
+  - @fe-free/core@6.0.25
+  - @fe-free/icons@6.0.25
+  - @fe-free/tool@6.0.25
+
+## 6.0.24
+
+### Patch Changes
+
+- feat: fetchStream
+  - @fe-free/core@6.0.24
+  - @fe-free/icons@6.0.24
+  - @fe-free/tool@6.0.24
+
 ## 6.0.23
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fe-free/ai",
-  "version": "6.0.23",
+  "version": "6.0.25",
   "description": "",
   "license": "ISC",
   "author": "",
@@ -17,7 +17,7 @@
     "lodash-es": "^4.17.21",
     "uuid": "^13.0.0",
     "zustand": "^4.5.7",
-    "@fe-free/core": "6.0.23"
+    "@fe-free/core": "6.0.25"
   },
   "peerDependencies": {
     "antd": "^6.2.1",
@@ -27,8 +27,8 @@
     "i18next-icu": "^2.4.1",
     "react": "^19.2.0",
     "react-i18next": "^16.4.0",
-    "@fe-free/icons": "6.0.23",
-    "@fe-free/tool": "6.0.23"
+    "@fe-free/icons": "6.0.25",
+    "@fe-free/tool": "6.0.25"
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",

package/src/sender/files.tsx

@@ -100,7 +100,7 @@ function FileUpload(
   },
 ) {
   const { allowUpload, refUpload, fileList, setFileList, uploadMaxCount } = props;
-  const { uploadAction, filesMaxCount, accept } = allowUpload || {};
+  const { uploadAction, uploadHeaders, filesMaxCount, accept } = allowUpload || {};
 
   const { message } = App.useApp();
   const { t } = useTranslation();
@@ -108,6 +108,7 @@ function FileUpload(
   return (
     <Upload.Dragger
       action={uploadAction}
+      headers={uploadHeaders}
       accept={accept}
       fileList={fileList}
       multiple

package/src/sender/sender.stories.tsx

@@ -72,6 +72,20 @@ export const AllowUploadWithAccept: Story = {
   render: (props) => <Component {...props} />,
 };
 
+export const AllowUploadWithHeaders: Story = {
+  args: {
+    allowUpload: {
+      uploadAction: '/api/ai-service/v1/file_upload/upload',
+      filesMaxCount: 3,
+      uploadHeaders: {
+        Authorization: 'Bearer mock-token',
+        'X-Trace-Id': 'storybook-demo',
+      },
+    },
+  },
+  render: (props) => <Component {...props} />,
+};
+
 export const RenderUpload: Story = {
   args: {
     allowUpload: {

package/src/sender/types.ts

@@ -1,5 +1,5 @@
-import type { ReactNode } from 'react';
-import { RefObject } from 'react';
+import type { UploadProps } from 'antd';
+import type { ReactNode, RefObject } from 'react';
 
 interface SenderRef {
   focus: () => void;
@@ -22,6 +22,8 @@ interface SenderProps {
   allowUpload?: {
     /** 上传文件的接口地址，约定返回的 {data: {url: string}} */
     uploadAction?: string;
+    /** 上传文件时附带的请求头 */
+    uploadHeaders?: UploadProps['headers'];
     /** 限制可上传文件类型，语法同 input.accept，如 image/*,.pdf */
     accept?: string;
     /** files 最大数量 */

package/src/stream/index.ts

@@ -1,5 +1,7 @@
 import { XRequest } from '@ant-design/x-sdk';
 
+type StreamMessage = { event: string; data: string };
+
 async function fetchStream(
   url: string,
   {
@@ -11,24 +13,25 @@ async function fetchStream(
     params?: Record<string, any>;
     headers?: Record<string, string>;
     callbacks?: {
-      onUpdate?: ({ event, data }: { event: string; data: string }, headers: Headers) => void;
+      onUpdate?: ({ event, data }: StreamMessage, headers: Headers) => void;
+      onSuccess?: (chunks: StreamMessage[], headers: Headers) => void;
       onError?: (error: Error, errorInfo?: any) => void;
     };
   },
 ) {
-  await XRequest<any, { event: string; data: string }>(url, {
+  XRequest<any, StreamMessage>(url, {
     ...rest,
     params: params,
     headers: headers,
     callbacks: {
-      onUpdate: (message, headers) => {
+      onUpdate: (message, hs) => {
         // 会存在 message 为 undefined 的情况
         if (message) {
-          callbacks?.onUpdate?.(message, headers);
+          callbacks?.onUpdate?.(message, hs);
         }
       },
-      onSuccess: () => {
-        // nothing
+      onSuccess: (chunks, hs) => {
+        callbacks?.onSuccess?.(chunks, hs);
       },
       onError: (error, errorInfo) => {
         console.log('XRequest onError', error);