@lobehub/chat 1.11.2 → 1.11.4

package/src/components/PanelTitle/index.tsx

@@ -0,0 +1,41 @@
+'use client';
+
+import { createStyles } from 'antd-style';
+import { CSSProperties, memo } from 'react';
+import { Flexbox } from 'react-layout-kit';
+
+const useStyles = createStyles(({ token, css }) => ({
+  desc: css`
+    line-height: 1.4;
+    color: ${token.colorTextDescription};
+  `,
+  header: css`
+    padding-block: 24px 0;
+    padding-inline: 0.75rem;
+  `,
+  title: css`
+    margin: 0;
+    font-size: 26px;
+    font-weight: 600;
+    line-height: 1.3;
+  `,
+}));
+
+interface PanelTitleProps {
+  desc?: string;
+  style?: CSSProperties;
+  title?: string;
+}
+
+const PanelTitle = memo(({ title, desc, style }: PanelTitleProps) => {
+  const { styles } = useStyles();
+
+  return (
+    <Flexbox className={styles.header} gap={4} style={style}>
+      <h1 className={styles.title}>{title}</h1>
+      <p className={styles.desc}>{desc}</p>
+    </Flexbox>
+  );
+});
+
+export default PanelTitle;

package/src/components/RepoIcon/index.tsx

@@ -0,0 +1,22 @@
+import { Icon } from '@lobehub/ui';
+import { useTheme } from 'antd-style';
+import { BookTextIcon } from 'lucide-react';
+import { memo } from 'react';
+
+interface RepoIconProps {
+  size?: number;
+}
+const RepoIcon = memo<RepoIconProps>(({ size = 20 }) => {
+  const theme = useTheme();
+
+  return (
+    <Icon
+      color={theme.geekblue}
+      fill={theme.geekblue3}
+      icon={BookTextIcon}
+      size={{ fontSize: size }}
+    />
+  );
+});
+
+export default RepoIcon;

package/src/components/SidebarHeader/index.tsx

@@ -1,5 +1,5 @@
 import { createStyles } from 'antd-style';
-import { type ReactNode, memo } from 'react';
+import { CSSProperties, type ReactNode, memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
 const useStyles = createStyles(({ css }) => ({
@@ -10,10 +10,11 @@ const useStyles = createStyles(({ css }) => ({
 
 interface SidebarHeaderProps {
   actions?: ReactNode;
+  style?: CSSProperties;
   title: ReactNode;
 }
 
-const SidebarHeader = memo<SidebarHeaderProps>(({ title, actions }) => {
+const SidebarHeader = memo<SidebarHeaderProps>(({ title, style, actions }) => {
   const { styles } = useStyles();
 
   return (
@@ -24,6 +25,7 @@ const SidebarHeader = memo<SidebarHeaderProps>(({ title, actions }) => {
       horizontal
       padding={14}
       paddingInline={16}
+      style={style}
     >
       <Flexbox align={'center'} gap={4} horizontal>
         {title}

package/src/components/TipGuide/index.tsx

@@ -0,0 +1,150 @@
+import { ActionIcon } from '@lobehub/ui';
+import { ConfigProvider, Popover, TooltipProps } from 'antd';
+import { createStyles, useTheme } from 'antd-style';
+import { XIcon } from 'lucide-react';
+import { CSSProperties, type FC, type ReactNode } from 'react';
+import { Flexbox } from 'react-layout-kit';
+
+const useStyle = createStyles(({ css }) => {
+  return {
+    close: css`
+      color: white;
+    `,
+    container: css`
+      position: relative;
+    `,
+    footer: css`
+      display: flex;
+      justify-content: end;
+      width: 100%;
+    `,
+    overlay: css`
+      .ant-popover-inner {
+        border: none;
+      }
+    `,
+    tip: css`
+      position: absolute;
+      inset-inline-start: 50%;
+      transform: translate(-50%);
+    `,
+  };
+});
+
+export interface TipGuideProps {
+  /**
+   * 引导内容
+   */
+  children?: ReactNode;
+  /**
+   * 类名
+   */
+  className?: string;
+  /**
+   * 默认时候的打开状态
+   */
+  defaultOpen?: boolean;
+  /**
+   * 用于自定义 footer 部分的 render api
+   */
+  footerRender?: (dom: ReactNode) => ReactNode;
+  /**
+   * 最大宽度
+   */
+  maxWidth?: number;
+  /**
+   * 纵向偏移值
+   */
+  offsetY?: number;
+  /**
+   * 当 open 属性变化时候的触发
+   */
+  onOpenChange: (open: boolean) => void;
+  /**
+   * 受控的 open 属性
+   */
+  open?: boolean;
+  /**
+   * Tooltip 位置，默认为 bottom
+   */
+  placement?: TooltipProps['placement'];
+  /**
+   * style
+   */
+  style?: CSSProperties;
+  tip?: boolean;
+  /**
+   * 引导标题
+   */
+  title: string;
+}
+
+const TipGuide: FC<TipGuideProps> = ({
+  children,
+  placement = 'bottom',
+  title,
+  offsetY,
+  maxWidth = 300,
+  className,
+  style,
+  open,
+  onOpenChange: setOpen,
+}) => {
+  const token = useTheme();
+  const { styles, cx } = useStyle();
+
+  return (
+    <ConfigProvider
+      theme={{
+        components: {
+          Badge: { fontSize: 12, lineHeight: 1 },
+          Button: { colorPrimary: token.blue7 },
+          Checkbox: {
+            colorPrimary: token.blue7,
+            colorText: token.colorTextLightSolid,
+          },
+          Popover: { colorText: token.colorTextLightSolid },
+        },
+      }}
+    >
+      {open ? (
+        <div className={styles.container}>
+          <div
+            style={{
+              marginTop: offsetY,
+            }}
+          >
+            <Popover
+              arrow={{ pointAtCenter: true }}
+              color={'blue'}
+              content={
+                <Flexbox gap={24} horizontal style={{ userSelect: 'none' }}>
+                  <div>{title}</div>
+                  <ActionIcon
+                    className={styles.close}
+                    icon={XIcon}
+                    onClick={() => {
+                      setOpen(false);
+                    }}
+                    size={'small'}
+                  />
+                </Flexbox>
+              }
+              open={open}
+              overlayClassName={cx(className, styles.overlay)}
+              overlayStyle={{ maxWidth, zIndex: 1000, ...style }}
+              placement={placement}
+              trigger="hover"
+            >
+              {children}
+            </Popover>
+          </div>
+        </div>
+      ) : (
+        children
+      )}
+    </ConfigProvider>
+  );
+};
+
+export default TipGuide;

package/src/config/llm.ts

@@ -90,6 +90,8 @@ export const getLLMConfig = () => {
 
       ENABLED_SILICONCLOUD: z.boolean(),
       SILICONCLOUD_API_KEY: z.string().optional(),
+      SILICONCLOUD_MODEL_LIST: z.string().optional(),
+      SILICONCLOUD_PROXY_URL: z.string().optional(),
     },
     runtimeEnv: {
       API_KEY_SELECT_MODE: process.env.API_KEY_SELECT_MODE,
@@ -177,6 +179,8 @@ export const getLLMConfig = () => {
 
       ENABLED_SILICONCLOUD: !!process.env.SILICONCLOUD_API_KEY,
       SILICONCLOUD_API_KEY: process.env.SILICONCLOUD_API_KEY,
+      SILICONCLOUD_MODEL_LIST: process.env.SILICONCLOUD_MODEL_LIST,
+      SILICONCLOUD_PROXY_URL: process.env.SILICONCLOUD_PROXY_URL,
     },
   });
 };

package/src/config/modelProviders/siliconcloud.ts

@@ -122,6 +122,9 @@ const SiliconCloud: ModelProviderCard = {
   id: 'siliconcloud',
   modelList: { showModelFetcher: true },
   name: 'SiliconCloud',
+  proxyUrl: {
+    placeholder: 'https://api.siliconflow.cn/v1',
+  },
 };
 
 export default SiliconCloud;

package/src/database/client/models/message.ts

@@ -6,9 +6,14 @@ import { DB_Message, DB_MessageSchema } from '@/database/client/schemas/message'
 import { ChatMessage } from '@/types/message';
 import { nanoid } from '@/utils/uuid';
 
+/**
+ * use `@/types/message` instead
+ * @deprecated
+ */
 export interface CreateMessageParams
   extends Partial<Omit<ChatMessage, 'content' | 'role'>>,
     Pick<ChatMessage, 'content' | 'role'> {
+  files?: string[];
   fromModel?: string;
   fromProvider?: string;
   sessionId: string;

package/src/features/Conversation/Messages/Default.tsx

@@ -1,10 +1,9 @@
 import { ReactNode, memo } from 'react';
 
+import BubblesLoading from '@/components/BubblesLoading';
 import { LOADING_FLAT } from '@/const/message';
 import { ChatMessage } from '@/types/message';
 
-import BubblesLoading from '@/components/BubblesLoading';
-
 export const DefaultMessage = memo<
   ChatMessage & {
     editableContent: ReactNode;

package/src/libs/agent-runtime/AgentRuntime.ts

@@ -27,6 +27,8 @@ import { LobeTogetherAI } from './togetherai';
 import {
   ChatCompetitionOptions,
   ChatStreamPayload,
+  EmbeddingsOptions,
+  EmbeddingsPayload,
   ModelProvider,
   TextToImagePayload,
 } from './types';
@@ -78,6 +80,7 @@ class AgentRuntime {
   async chat(payload: ChatStreamPayload, options?: ChatCompetitionOptions) {
     return this._runtime.chat(payload, options);
   }
+
   async textToImage(payload: TextToImagePayload) {
     return this._runtime.textToImage?.(payload);
   }
@@ -86,6 +89,10 @@ class AgentRuntime {
     return this._runtime.models?.();
   }
 
+  async embeddings(payload: EmbeddingsPayload, options?: EmbeddingsOptions) {
+    return this._runtime.embeddings?.(payload, options);
+  }
+
   /**
    * @description Initialize the runtime with the provider and the options
    * @param provider choose a model provider
@@ -247,12 +254,12 @@ class AgentRuntime {
 
       case ModelProvider.Ai360: {
         runtimeModel = new LobeAi360AI(params.ai360 ?? {});
-        break
+        break;
       }
 
       case ModelProvider.SiliconCloud: {
         runtimeModel = new LobeSiliconCloudAI(params.siliconcloud ?? {});
-        break
+        break;
       }
     }
 

package/src/libs/agent-runtime/BaseAI.ts

@@ -3,12 +3,20 @@ import OpenAI from 'openai';
 import { TextToImagePayload } from '@/libs/agent-runtime/types/textToImage';
 import { ChatModelCard } from '@/types/llm';
 
-import { ChatCompetitionOptions, ChatStreamPayload } from './types';
+import {
+  ChatCompetitionOptions,
+  ChatStreamPayload,
+  EmbeddingItem,
+  EmbeddingsOptions,
+  EmbeddingsPayload,
+} from './types';
 
 export interface LobeRuntimeAI {
   baseURL?: string;
   chat(payload: ChatStreamPayload, options?: ChatCompetitionOptions): Promise<Response>;
 
+  embeddings?(payload: EmbeddingsPayload, options?: EmbeddingsOptions): Promise<EmbeddingItem[]>;
+
   models?(): Promise<any>;
 
   textToImage?: (payload: TextToImagePayload) => Promise<string[]>;
@@ -21,4 +29,9 @@ export abstract class LobeOpenAICompatibleRuntime {
   abstract chat(payload: ChatStreamPayload, options?: ChatCompetitionOptions): Promise<Response>;
 
   abstract models(): Promise<ChatModelCard[]>;
+
+  abstract embeddings(
+    payload: EmbeddingsPayload,
+    options?: EmbeddingsOptions,
+  ): Promise<EmbeddingItem[]>;
 }

package/src/libs/agent-runtime/qwen/index.ts

@@ -3,7 +3,7 @@ import OpenAI, { ClientOptions } from 'openai';
 
 import Qwen from '@/config/modelProviders/qwen';
 
-import { LobeOpenAICompatibleRuntime, LobeRuntimeAI } from '../BaseAI';
+import { LobeRuntimeAI } from '../BaseAI';
 import { AgentRuntimeErrorType } from '../error';
 import { ChatCompetitionOptions, ChatStreamPayload, ModelProvider } from '../types';
 import { AgentRuntimeError } from '../utils/createError';
@@ -23,7 +23,7 @@ const DEFAULT_BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1';
  * or
  * 2. Set S3-* enviroment variables properly to store all uploaded files.
  */
-export class LobeQwenAI extends LobeOpenAICompatibleRuntime implements LobeRuntimeAI {
+export class LobeQwenAI implements LobeRuntimeAI {
   client: OpenAI;
   baseURL: string;
 
@@ -32,7 +32,6 @@ export class LobeQwenAI extends LobeOpenAICompatibleRuntime implements LobeRunti
     baseURL = DEFAULT_BASE_URL,
     ...res
   }: ClientOptions & Record<string, any> = {}) {
-    super();
     if (!apiKey) throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidProviderAPIKey);
     this.client = new OpenAI({ apiKey, baseURL, ...res });
     this.baseURL = this.client.baseURL;
@@ -108,7 +107,7 @@ export class LobeQwenAI extends LobeOpenAICompatibleRuntime implements LobeRunti
       ...payload,
       messages,
       result_format: 'message',
-      stream: !!tools?.length ? false : stream ?? true,
+      stream: !!tools?.length ? false : (stream ?? true),
       top_p: top_p && top_p >= 1 ? 0.999 : top_p,
     };
 

package/src/libs/agent-runtime/types/embeddings.ts

@@ -0,0 +1,43 @@
+export interface EmbeddingsPayload {
+  /**
+   * The number of dimensions the resulting output embeddings should have. Only
+   * supported in `text-embedding-3` and later models.
+   */
+  dimensions?: number;
+  /**
+   * Input text to embed, encoded as a string or array of tokens. To embed multiple
+   * inputs in a single request, pass an array of strings .
+   * The input must not exceed the max input tokens for the model (8192 tokens for
+   * `text-embedding-ada-002`), cannot be an empty string, and any array must be 2048
+   * dimensions or less.
+   */
+  input: string | Array<string>;
+
+  model: string;
+}
+
+export interface EmbeddingsOptions {
+  headers?: Record<string, any>;
+  signal?: AbortSignal;
+  /**
+   * userId for the embeddings
+   */
+  user?: string;
+}
+
+export interface EmbeddingItem {
+  /**
+   * The embedding vector, which is a list of floats. The length of vector depends on
+   * the model as listed in the
+   * [embedding guide](https://platform.openai.com/docs/guides/embeddings).
+   */
+  embedding: Array<number>;
+  /**
+   * The index of the embedding in the list of embeddings.
+   */
+  index: number;
+  /**
+   * The object type, which is always "embedding".
+   */
+  object: 'embedding';
+}

package/src/libs/agent-runtime/types/index.ts

@@ -1,3 +1,4 @@
 export * from './chat';
+export * from './embeddings';
 export * from './textToImage';
 export * from './type';

package/src/libs/agent-runtime/utils/openaiCompatibleFactory/index.ts

@@ -6,7 +6,14 @@ import { ChatModelCard } from '@/types/llm';
 
 import { LobeRuntimeAI } from '../../BaseAI';
 import { AgentRuntimeErrorType, ILobeAgentRuntimeErrorType } from '../../error';
-import { ChatCompetitionOptions, ChatCompletionErrorPayload, ChatStreamPayload } from '../../types';
+import {
+  ChatCompetitionOptions,
+  ChatCompletionErrorPayload,
+  ChatStreamPayload,
+  EmbeddingItem,
+  EmbeddingsOptions,
+  EmbeddingsPayload,
+} from '../../types';
 import { AgentRuntimeError } from '../createError';
 import { debugResponse, debugStream } from '../debugStream';
 import { desensitizeUrl } from '../desensitizeUrl';
@@ -154,7 +161,7 @@ export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>
         const response = await this.client.chat.completions.create(
           {
             ...postPayload,
-            ...(chatCompletion?.noUserId ? {} : { user: options?.user }) 
+            ...(chatCompletion?.noUserId ? {} : { user: options?.user }),
           },
           {
             // https://github.com/lobehub/lobe-chat/pull/318
@@ -215,6 +222,22 @@ export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>
         .filter(Boolean) as ChatModelCard[];
     }
 
+    async embeddings(
+      payload: EmbeddingsPayload,
+      options?: EmbeddingsOptions,
+    ): Promise<EmbeddingItem[]> {
+      try {
+        const res = await this.client.embeddings.create(
+          { ...payload, user: options?.user },
+          { headers: options?.headers, signal: options?.signal },
+        );
+
+        return res.data;
+      } catch (error) {
+        throw this.handleError(error);
+      }
+    }
+
     async textToImage(payload: TextToImagePayload) {
       try {
         const res = await this.client.images.generate(payload);

package/src/libs/langchain/file.ts

@@ -0,0 +1,29 @@
+export const LANGCHAIN_SUPPORT_TEXT_LIST = [
+  'txt',
+  'md',
+  'markdown',
+  'md',
+  'mdx',
+  'toml',
+  'yaml',
+  'json',
+
+  'sh',
+  'patch',
+  // js
+  'js',
+  'jsx',
+  'mjs',
+  'tsx',
+  'ts',
+  'less',
+  'css',
+  // rust
+  'rs',
+  // python
+  'py',
+  'java',
+  // database
+  'sql',
+  'db',
+];

package/src/libs/langchain/index.ts

@@ -0,0 +1 @@
+export * from './loaders';

package/src/libs/langchain/loaders/code/__tests__/index.test.ts

@@ -0,0 +1,34 @@
+// @vitest-environment node
+import * as fs from 'node:fs';
+import { join } from 'node:path';
+
+import { CodeLoader } from '../index';
+import longResult from './long.json';
+
+describe('CodeLoader', () => {
+  it('split simple code', async () => {
+    const jsCode = `function helloWorld() {
+  console.log("Hello, World!");
+}
+// Call the function
+helloWorld();`;
+
+    const result = await CodeLoader(jsCode, 'js');
+
+    expect(result).toEqual([
+      {
+        pageContent:
+          'function helloWorld() {\n  console.log("Hello, World!");\n}\n// Call the function\nhelloWorld();',
+        metadata: { loc: { lines: { from: 1, to: 5 } } },
+      },
+    ]);
+  });
+
+  it('split long', async () => {
+    const code = fs.readFileSync(join(__dirname, `./long.txt`), 'utf-8');
+
+    const result = await CodeLoader(code, 'js');
+
+    expect(result).toEqual(longResult);
+  });
+});