feat(core): workspace embedding settings

Author: yoyoyohamapi

blocksuite/affine/components/src/icons/file-icons-rc.ts

@@ -0,0 +1,174 @@
+import {
+  ImageIcon,
+  FileIconJpegIcon,
+  FileIconPngIcon,
+  FileIconWebpIcon,
+  FileIconTiffIcon,
+  FileIconGifIcon,
+  FileIconSvgIcon,
+  FileIconEpsIcon,
+  FileIconPdfIcon,
+  FileIconDocxIcon,
+  FileIconTxtIcon,
+  FileIconCsvIcon,
+  FileIconDocIcon,
+  FileIconXlsIcon,
+  FileIconXlsxIcon,
+  FileIconPptIcon,
+  FileIconPptxIcon,
+  FileIconAiIcon,
+  FileIconPsdIcon,
+  FileIconInddIcon,
+  FileIconAepIcon,
+  FileIconFigIcon,
+  FileIconMp3Icon,
+  FileIconWavIcon,
+  FileIconMpegIcon,
+  FileIconAviIcon,
+  FileIconMp4Icon,
+  FileIconMkvIcon,
+  FileIconHtmlIcon,
+  FileIconRssIcon,
+  FileIconSqlIcon,
+  FileIconJsIcon,
+  FileIconCssIcon,
+  FileIconJavaIcon,
+  FileIconXmlIcon,
+  FileIconDmgIcon,
+  FileIconZipIcon,
+  FileIconRarIcon,
+  FileIconExeIcon,
+  FileIconJsonIcon,
+  FileIconNoneIcon,
+} from '@blocksuite/icons/rc';
+
+export function getAttachmentFileIconRC(filetype: string) {
+  switch (filetype) {
+    case 'img':
+      return ImageIcon;
+    case 'image/jpeg':
+    case 'jpg':
+    case 'jpeg':
+      return FileIconJpegIcon;
+    case 'image/png':
+    case 'png':
+      return FileIconPngIcon;
+    case 'image/webp':
+    case 'webp':
+      return FileIconWebpIcon;
+    case 'image/tiff':
+    case 'tiff':
+      return FileIconTiffIcon;
+    case 'image/gif':
+    case 'gif':
+      return FileIconGifIcon;
+    case 'image/svg':
+    case 'svg':
+      return FileIconSvgIcon;
+    case 'image/eps':
+    case 'eps':
+      return FileIconEpsIcon;
+    case 'application/pdf':
+    case 'pdf':
+      return FileIconPdfIcon;
+    case 'application/msword':
+    case 'application/x-iwork-pages-sffpages':
+    case 'doc':
+      return FileIconDocIcon;
+    case 'application/vnd.openxmlformats-officedocument.wordprocessingml.document':
+    case 'docx':
+      return FileIconDocxIcon;
+    case 'text/plain':
+    case 'txt':
+      return FileIconTxtIcon;
+    case 'csv':
+      return FileIconCsvIcon;
+    case 'application/vnd.ms-excel':
+    case 'xls':
+      return FileIconXlsIcon;
+    case 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet':
+    case 'application/x-iwork-numbers-sffnumbers':
+    case 'xlsx':
+      return FileIconXlsxIcon;
+    case 'application/vnd.ms-powerpoint':
+    case 'application/x-iwork-keynote-sffkeynote':
+    case 'ppt':
+      return FileIconPptIcon;
+    case 'application/vnd.openxmlformats-officedocument.presentationml.presentation':
+    case 'pptx':
+      return FileIconPptxIcon;
+    case 'application/illustrator':
+    case 'fig':
+      return FileIconFigIcon;
+    case 'application/postscript':
+    case 'ai':
+      return FileIconAiIcon;
+    case 'application/vnd.adobe.photoshop':
+    case 'psd':
+      return FileIconPsdIcon;
+    case 'application/vnd.adobe.indesign':
+    case 'indd':
+      return FileIconInddIcon;
+    case 'application/vnd.adobe.afterfx':
+    case 'aep':
+      return FileIconAepIcon;
+    case 'audio/mpeg':
+    case 'audio/mp3':
+    case 'mp3':
+      return FileIconMp3Icon;
+    case 'audio/wav':
+    case 'wav':
+      return FileIconWavIcon;
+    case 'video/mpeg':
+    case 'video/mp4':
+    case 'mp4':
+      return FileIconMp4Icon;
+    case 'video/mpeg':
+    case 'mpeg':
+      return FileIconMpegIcon;
+    case 'video/avi':
+    case 'avi':
+      return FileIconAviIcon;
+    case 'video/mkv':
+    case 'mkv':
+      return FileIconMkvIcon;
+    case 'text/html':
+    case 'html':
+      return FileIconHtmlIcon;
+    case 'text/css':
+    case 'css':
+      return FileIconCssIcon;
+    case 'application/rss+xml':
+    case 'rss':
+      return FileIconRssIcon;
+    case 'application/sql':
+    case 'sql':
+      return FileIconSqlIcon;
+    case 'application/javascript':
+    case 'js':
+      return FileIconJsIcon;
+    case 'application/json':
+    case 'json':
+      return FileIconJsonIcon;
+    case 'application/java':
+    case 'java':
+      return FileIconJavaIcon;
+    case 'application/xml':
+    case 'xml':
+      return FileIconXmlIcon;
+    case 'application/x-msdos-program':
+    case 'exe':
+      return FileIconExeIcon;
+    case 'application/x-apple-diskimage':
+    case 'dmg':
+      return FileIconDmgIcon;
+    case 'application/zip':
+    case 'zip':
+      return FileIconZipIcon;
+    case 'application/x-rar-compressed':
+    case 'rar':
+      return FileIconRarIcon;
+    default:
+      return FileIconNoneIcon;
+  }
+}

blocksuite/affine/components/src/icons/index.ts

@@ -1,5 +1,6 @@
 export * from './ai.js';
 export * from './file-icons.js';
+export * from './file-icons-rc.js';
 export * from './import-export.js';
 export * from './list.js';
 export * from './misc.js';

packages/frontend/component/src/components/setting-components/setting-row.tsx

@@ -5,7 +5,7 @@ import { settingRow } from './share.css';
 
 export type SettingRowProps = PropsWithChildren<{
   name: ReactNode;
-  desc: ReactNode;
+  desc?: ReactNode;
   style?: CSSProperties;
   onClick?: () => void;
   spreadCol?: boolean;
@@ -41,7 +41,7 @@ export const SettingRow = ({
     >
       <div className="left-col">
         <div className="name">{name}</div>
-        <div className="desc">{desc}</div>
+        {desc && <div className="desc">{desc}</div>}
       </div>
       {spreadCol ? <div className="right-col">{children}</div> : children}
     </div>

packages/frontend/component/src/components/setting-components/wrapper.tsx

@@ -7,16 +7,22 @@ interface SettingWrapperProps {
   id?: string;
   title?: ReactNode;
   disabled?: boolean;
+  testId?: string;
 }
 
 export const SettingWrapper = ({
   id,
   title,
   children,
   disabled,
+  testId,
 }: PropsWithChildren<SettingWrapperProps>) => {
   return (
-    <div id={id} className={clsx(wrapper, disabled && wrapperDisabled)}>
+    <div
+      id={id}
+      className={clsx(wrapper, disabled && wrapperDisabled)}
+      data-testid={testId}
+    >
       {title ? <div className="title">{title}</div> : null}
       {children}
     </div>

packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts

@@ -627,9 +627,6 @@ export class AIChatInput extends SignalWatcher(WithDisposable(LitElement)) {
 
   private async _getMatchedContexts(userInput: string) {
     const contextId = await this.getContextId();
-    if (!contextId) {
-      return { files: [], docs: [] };
-    }
 
     const docContexts = new Map<
       string,
@@ -641,7 +638,7 @@ export class AIChatInput extends SignalWatcher(WithDisposable(LitElement)) {
     >();
 
     const { files: matchedFiles = [], docs: matchedDocs = [] } =
-      (await AIProvider.context?.matchContext(contextId, userInput)) ?? {};
+      (await AIProvider.context?.matchContext(userInput, contextId)) ?? {};
 
     matchedDocs.forEach(doc => {
       docContexts.set(doc.docId, {
@@ -655,17 +652,12 @@ export class AIChatInput extends SignalWatcher(WithDisposable(LitElement)) {
       if (context) {
         context.fileContent += `\n${file.content}`;
       } else {
-        const fileChip = this.chips.find(
-          chip => isFileChip(chip) && chip.fileId === file.fileId
-        ) as FileChip | undefined;
-        if (fileChip && fileChip.blobId) {
-          fileContexts.set(file.fileId, {
-            blobId: fileChip.blobId,
-            fileName: fileChip.file.name,
-            fileType: fileChip.file.type,
-            fileContent: file.content,
-          });
-        }
+        fileContexts.set(file.fileId, {
+          blobId: file.blobId,
+          fileName: file.name,
+          fileType: file.type,
+          fileContent: file.content,
+        });
       }
     });
 

packages/frontend/core/src/blocksuite/ai/provider/copilot-client.ts

@@ -333,7 +333,7 @@ export class CopilotClient {
     return res.currentUser?.copilot?.contexts?.[0];
   }
 
-  async matchContext(contextId: string, content: string, limit?: number) {
+  async matchContext(content: string, contextId?: string, limit?: number) {
     const res = await this.gql({
       query: matchContextQuery,
       variables: {

packages/frontend/core/src/blocksuite/ai/settings/indexer-embedding/constants.ts

@@ -0,0 +1 @@
+export const COUNT_PER_PAGE = 10;