feat: Add component AkShare #1739

web/src/locales/en.ts

@@ -892,6 +892,9 @@ The above is the content you need to summarize.`,
         lccp: 'Financing',
         foreign_exchange: 'Foreign currency',
       },
+      akShare: 'AkShare',
+      akShareDescription:
+        'This component can be used to obtain news information for the corresponding stock from the Eastmoney website.',
     },
     footer: {
       profile: 'All rights reserved @ React',

web/src/locales/zh-traditional.ts

@@ -846,6 +846,8 @@ export default {
         lccp: '理財',
         foreign_exchange: '股票',
       },
+      akShare: 'AkShare',
+      akShareDescription: '此組件可用於從東方財富網取得對應股票的新聞資訊。',
     },
     footer: {
       profile: '“保留所有權利 @ react”',

web/src/locales/zh.ts

@@ -864,6 +864,8 @@ export default {
         lccp: '理财',
         foreign_exchange: '股票',
       },
+      akShare: 'AkShare',
+      akShareDescription: '该组件可用于从东方财富网站获取相应股票的新闻信息。',
     },
     footer: {
       profile: 'All rights reserved @ React',

web/src/pages/flow/akshare-form/index.tsx

@@ -0,0 +1,20 @@
+import TopNItem from '@/components/top-n-item';
+import { Form } from 'antd';
+import { IOperatorForm } from '../interface';
+
+const AkShareForm = ({ onValuesChange, form }: IOperatorForm) => {
+  return (
+    <Form
+      name="basic"
+      labelCol={{ span: 6 }}
+      wrapperCol={{ span: 18 }}
+      autoComplete="off"
+      form={form}
+      onValuesChange={onValuesChange}
+    >
+      <TopNItem initialValue={10} max={99}></TopNItem>
+    </Form>
+  );
+};
+
+export default AkShareForm;

web/src/pages/flow/constant.tsx

@@ -1,3 +1,4 @@
+import { ReactComponent as AkShareIcon } from '@/assets/svg/akshare.svg';
 import { ReactComponent as ArXivIcon } from '@/assets/svg/arxiv.svg';
 import { ReactComponent as baiduFanyiIcon } from '@/assets/svg/baidu-fanyi.svg';
 import { ReactComponent as BaiduIcon } from '@/assets/svg/baidu.svg';
@@ -61,6 +62,7 @@ export enum Operator {
   ExeSQL = 'ExeSQL',
   Switch = 'Switch',
   WenCai = 'WenCai',
+  AkShare = 'AkShare',
 }
 
 export const operatorIconMap = {
@@ -88,6 +90,7 @@ export const operatorIconMap = {
   [Operator.ExeSQL]: ExeSqlIcon,
   [Operator.Switch]: SwitchIcon,
   [Operator.WenCai]: WenCaiIcon,
+  [Operator.AkShare]: AkShareIcon,
 };
 
 export const operatorMap: Record<
@@ -198,6 +201,7 @@ export const operatorMap: Record<
   [Operator.ExeSQL]: { backgroundColor: '#b9efe8' },
   [Operator.Switch]: { backgroundColor: '#dbaff6' },
   [Operator.WenCai]: { backgroundColor: '#faac5b' },
+  [Operator.AkShare]: { backgroundColor: '#8085f5' },
 };
 
 export const componentMenuList = [
@@ -270,6 +274,9 @@ export const componentMenuList = [
   {
     name: Operator.WenCai,
   },
+  {
+    name: Operator.AkShare,
+  },
 ];
 
 export const initialRetrievalValues = {
@@ -411,6 +418,8 @@ export const initialSwitchValues = { conditions: [] };
 
 export const initialWenCaiValues = { top_n: 20, query_type: 'stock' };
 
+export const initialAkShareValues = { top_n: 10 };
+
 export const CategorizeAnchorPointPositions = [
   { top: 1, right: 34 },
   { top: 8, right: 18 },
@@ -482,6 +491,7 @@ export const RestrictedUpstreamMap = {
   [Operator.ExeSQL]: [Operator.Begin],
   [Operator.Switch]: [Operator.Begin],
   [Operator.WenCai]: [Operator.Begin],
+  [Operator.AkShare]: [Operator.Begin],
 };
 
 export const NodeMap = {
@@ -509,6 +519,7 @@ export const NodeMap = {
   [Operator.ExeSQL]: 'ragNode',
   [Operator.Switch]: 'categorizeNode',
   [Operator.WenCai]: 'ragNode',
+  [Operator.AkShare]: 'ragNode',
 };
 
 export const LanguageOptions = [

web/src/pages/flow/flow-drawer/index.tsx

@@ -3,6 +3,7 @@ import { IModalProps } from '@/interfaces/common';
 import { Drawer, Flex, Form, Input } from 'antd';
 import { useEffect } from 'react';
 import { Node } from 'reactflow';
+import AkShareForm from '../akshare-form';
 import AnswerForm from '../answer-form';
 import ArXivForm from '../arxiv-form';
 import BaiduFanyiForm from '../baidu-fanyi-form';
@@ -62,6 +63,7 @@ const FormMap = {
   [Operator.ExeSQL]: ExeSQLForm,
   [Operator.Switch]: SwitchForm,
   [Operator.WenCai]: WenCaiForm,
+  [Operator.AkShare]: AkShareForm,
 };
 
 const EmptyContent = () => <div>empty</div>;

web/src/pages/flow/hooks.ts

@@ -31,6 +31,7 @@ import {
   Operator,
   RestrictedUpstreamMap,
   SwitchElseTo,
+  initialAkShareValues,
   initialArXivValues,
   initialBaiduFanyiValues,
   initialBaiduValues,
@@ -115,6 +116,7 @@ export const useInitializeOperatorParams = () => {
       [Operator.ExeSQL]: initialExeSqlValues,
       [Operator.Switch]: initialSwitchValues,
       [Operator.WenCai]: initialWenCaiValues,
+      [Operator.AkShare]: initialAkShareValues,
     };
   }, [llmId]);
 

web/src/pages/flow/wencai-form/index.tsx

@@ -24,7 +24,7 @@ const WenCaiForm = ({ onValuesChange, form }: IOperatorForm) => {
       form={form}
       onValuesChange={onValuesChange}
     >
-      <TopNItem initialValue={10} max={99}></TopNItem>
+      <TopNItem initialValue={20} max={99}></TopNItem>
       <Form.Item label={t('queryType')} name={'query_type'}>
         <Select options={wenCaiQueryTypeOptions}></Select>
       </Form.Item>