Merge pull request #174 from SuanmoSuanyangTechnology/feature/ui_zy
Feature/UI zy
This commit is contained in:
@@ -2064,6 +2064,7 @@ Memory Bear: After the rebellion, regional warlordism intensified for several re
|
|||||||
stability: 'Stability',
|
stability: 'Stability',
|
||||||
resilience: 'Resilience',
|
resilience: 'Resilience',
|
||||||
suggestions: 'Personalized Suggestions',
|
suggestions: 'Personalized Suggestions',
|
||||||
|
suggestionLoading: 'Your personalized suggestions are being generated',
|
||||||
},
|
},
|
||||||
reflectionEngine: {
|
reflectionEngine: {
|
||||||
reflectionEngineConfig: 'Reflection Engine Configuration',
|
reflectionEngineConfig: 'Reflection Engine Configuration',
|
||||||
|
|||||||
@@ -2158,6 +2158,7 @@ export const zh = {
|
|||||||
stability: '稳定性',
|
stability: '稳定性',
|
||||||
resilience: '恢复力',
|
resilience: '恢复力',
|
||||||
suggestions: '个性化建议',
|
suggestions: '个性化建议',
|
||||||
|
suggestionLoading: '您的个性化建议正在生成中',
|
||||||
},
|
},
|
||||||
reflectionEngine: {
|
reflectionEngine: {
|
||||||
reflectionEngineConfig: '反思引擎配置',
|
reflectionEngineConfig: '反思引擎配置',
|
||||||
|
|||||||
@@ -52,6 +52,8 @@ service.interceptors.request.use(
|
|||||||
config.headers.Authorization = `Bearer ${token}`;
|
config.headers.Authorization = `Bearer ${token}`;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
const language = localStorage.getItem('language')
|
||||||
|
config.headers['X-Language-Type'] = language || 'en';
|
||||||
config.headers.Cookie = undefined
|
config.headers.Cookie = undefined
|
||||||
return config;
|
return config;
|
||||||
},
|
},
|
||||||
@@ -146,7 +148,7 @@ service.interceptors.response.use(
|
|||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
if (!msg && Array.isArray(error.response?.data?.detail)) {
|
if (!msg && Array.isArray(error.response?.data?.detail)) {
|
||||||
msg = error.response?.data?.detail?.map(item => item.msg).join(';')
|
msg = error.response?.data?.detail?.map((item: { msg: string }) => item.msg).join(';')
|
||||||
} else {
|
} else {
|
||||||
msg = msg || i18n.t('common.unknownError');
|
msg = msg || i18n.t('common.unknownError');
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -128,7 +128,11 @@ const Agent = forwardRef<AgentRef>((_props, ref) => {
|
|||||||
let allTools = Array.isArray(response.tools) ? response.tools : []
|
let allTools = Array.isArray(response.tools) ? response.tools : []
|
||||||
form.setFieldsValue({
|
form.setFieldsValue({
|
||||||
...response,
|
...response,
|
||||||
tools: allTools
|
tools: allTools,
|
||||||
|
memory: {
|
||||||
|
...response.memory,
|
||||||
|
memory_content: response.memory?.memory_content ? Number(response.memory?.memory_content) : undefined
|
||||||
|
}
|
||||||
})
|
})
|
||||||
setData({
|
setData({
|
||||||
...response,
|
...response,
|
||||||
|
|||||||
@@ -57,7 +57,8 @@ const MemoryInsight = forwardRef<MemoryInsightRef>((_props, ref) => {
|
|||||||
: Object.keys(data).length > 0
|
: Object.keys(data).length > 0
|
||||||
? <Space size={16} direction="vertical" className="rb:w-full">
|
? <Space size={16} direction="vertical" className="rb:w-full">
|
||||||
{['memory_insight', 'key_findings', 'behavior_pattern', 'growth_trajectory'].map(key => {
|
{['memory_insight', 'key_findings', 'behavior_pattern', 'growth_trajectory'].map(key => {
|
||||||
if (data[key as keyof Data]) {
|
const value = data[key as keyof Data];
|
||||||
|
if (Array.isArray(value) && value.length > 0 || (!Array.isArray(value) && value)) {
|
||||||
return (
|
return (
|
||||||
<div key={key} className="rb:bg-[#F6F8FC] rb:border rb:border-[#DFE4ED] rb:rounded-lg rb:py-3 rb:text-[#5B6167] rb:leading-5">
|
<div key={key} className="rb:bg-[#F6F8FC] rb:border rb:border-[#DFE4ED] rb:rounded-lg rb:py-3 rb:text-[#5B6167] rb:leading-5">
|
||||||
<div className={clsx(`rb:relative rb:before:content-[''] rb:before:block rb:before:h-4 rb:before:absolute rb:before:top-0.5 rb:before:left-0 rb:before:w-1 rb:pl-4 rb:mb-2 rb:font-medium rb:leading-5`, {
|
<div className={clsx(`rb:relative rb:before:content-[''] rb:before:block rb:before:h-4 rb:before:absolute rb:before:top-0.5 rb:before:left-0 rb:before:w-1 rb:pl-4 rb:mb-2 rb:font-medium rb:leading-5`, {
|
||||||
|
|||||||
@@ -35,13 +35,13 @@ const PageHeader: FC<ConfigHeaderProps> = ({
|
|||||||
{operation}
|
{operation}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<Space size={12}>
|
<div className="rb:flex rb:items-center rb:gap-3">
|
||||||
<Button type="primary" ghost className="rb:group rb:h-6! rb:px-2!" onClick={goBack}>
|
<Button type="primary" ghost className="rb:h-6! rb:px-2! rb:leading-5.5!" onClick={goBack}>
|
||||||
<img src={logoutIcon} className="rb:w-4 rb:h-4" />
|
<img src={logoutIcon} className="rb:w-4 rb:h-4" />
|
||||||
{t('common.return')}
|
{t('common.return')}
|
||||||
</Button>
|
</Button>
|
||||||
{extra}
|
{extra}
|
||||||
</Space>
|
</div>
|
||||||
</Header>
|
</Header>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -266,7 +266,7 @@ const RelationshipNetwork:FC = () => {
|
|||||||
size={[197.81, 150]}
|
size={[197.81, 150]}
|
||||||
/>
|
/>
|
||||||
: <>
|
: <>
|
||||||
<div className="rb:bg-[#F6F8FC] rb:border-t rb:border-b rb:border-[#DFE4ED] rb:font-medium rb:py-2 rb:px-4 rb:h-10">{selectedNode.name}</div>
|
{selectedNode.name && <div className="rb:bg-[#F6F8FC] rb:border-t rb:border-b rb:border-[#DFE4ED] rb:font-medium rb:py-2 rb:px-4 rb:h-10">{selectedNode.name}</div>}
|
||||||
<div className="rb:p-4">
|
<div className="rb:p-4">
|
||||||
<>
|
<>
|
||||||
<div className="rb:font-medium rb:leading-5">{t('userMemory.memoryContent')}</div>
|
<div className="rb:font-medium rb:leading-5">{t('userMemory.memoryContent')}</div>
|
||||||
@@ -297,7 +297,8 @@ const RelationshipNetwork:FC = () => {
|
|||||||
{selectedNode.label === 'Statement' && <>
|
{selectedNode.label === 'Statement' && <>
|
||||||
{(['emotion_keywords', 'emotion_type', 'emotion_subject', 'importance_score'] as const).map(key => {
|
{(['emotion_keywords', 'emotion_type', 'emotion_subject', 'importance_score'] as const).map(key => {
|
||||||
const statementProps = selectedNode.properties as StatementNodeProperties;
|
const statementProps = selectedNode.properties as StatementNodeProperties;
|
||||||
if ((key === 'emotion_keywords' && statementProps[key]?.length > 0) || statementProps[key]) {
|
if ((key === 'emotion_keywords' && statementProps[key]?.length > 0) || typeof statementProps[key] === 'string') {
|
||||||
|
console.log('statementProps[key]', statementProps[key])
|
||||||
return (
|
return (
|
||||||
<div className="rb:mt-4" key={key}>
|
<div className="rb:mt-4" key={key}>
|
||||||
{t(`userMemory.Statement_${key}`)}
|
{t(`userMemory.Statement_${key}`)}
|
||||||
@@ -321,7 +322,10 @@ const RelationshipNetwork:FC = () => {
|
|||||||
<div className="rb:mt-4" key={key}>
|
<div className="rb:mt-4" key={key}>
|
||||||
{t(`userMemory.ExtractedEntity_${key}`)}
|
{t(`userMemory.ExtractedEntity_${key}`)}
|
||||||
<div className="rb:text-[#5B6167] rb:font-regular rb:leading-5 rb:mt-1 rb:pb-4 rb:border-b rb:border-[#DFE4ED]">
|
<div className="rb:text-[#5B6167] rb:font-regular rb:leading-5 rb:mt-1 rb:pb-4 rb:border-b rb:border-[#DFE4ED]">
|
||||||
{entityProps[key]}
|
{Array.isArray(entityProps[key]) && entityProps[key].length > 0
|
||||||
|
? entityProps[key].map((vo, index) => <div key={index}>- {vo}</div>)
|
||||||
|
: entityProps[key]
|
||||||
|
}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -21,6 +21,7 @@ interface Suggestions {
|
|||||||
const Suggestions = forwardRef<{ handleRefresh: () => void; }>((_props, ref) => {
|
const Suggestions = forwardRef<{ handleRefresh: () => void; }>((_props, ref) => {
|
||||||
const { t } = useTranslation()
|
const { t } = useTranslation()
|
||||||
const { id } = useParams()
|
const { id } = useParams()
|
||||||
|
const [loading, setLoading] = useState(false)
|
||||||
const [suggestions, setSuggestions] = useState<Suggestions | null>(null)
|
const [suggestions, setSuggestions] = useState<Suggestions | null>(null)
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -31,10 +32,14 @@ const Suggestions = forwardRef<{ handleRefresh: () => void; }>((_props, ref) =>
|
|||||||
if (!id) {
|
if (!id) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
setLoading(true)
|
||||||
getEmotionSuggestions(id)
|
getEmotionSuggestions(id)
|
||||||
.then((res) => {
|
.then((res) => {
|
||||||
setSuggestions(res as Suggestions)
|
setSuggestions(res as Suggestions)
|
||||||
})
|
})
|
||||||
|
.finally(() => {
|
||||||
|
setLoading(false)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
useImperativeHandle(ref, () => ({
|
useImperativeHandle(ref, () => ({
|
||||||
@@ -63,7 +68,7 @@ const Suggestions = forwardRef<{ handleRefresh: () => void; }>((_props, ref) =>
|
|||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
</>
|
</>
|
||||||
: <Empty size={88} className="rb:h-full" />
|
: <Empty size={88} subTitle={t(loading ? 'statementDetail.suggestionLoading' : 'empty.tableEmpty')} className="rb:h-full" />
|
||||||
}
|
}
|
||||||
</RbCard>
|
</RbCard>
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -20,14 +20,22 @@ const ImplicitDetail = forwardRef<{ handleRefresh: () => void; }>((_props, ref)
|
|||||||
const habitsRef = useRef<{ handleRefresh: () => void; }>(null)
|
const habitsRef = useRef<{ handleRefresh: () => void; }>(null)
|
||||||
|
|
||||||
const handleRefresh = () => {
|
const handleRefresh = () => {
|
||||||
if (!id) return
|
if (!id) {
|
||||||
generateProfile(id)
|
return Promise.resolve()
|
||||||
.then(() => {
|
}
|
||||||
preferencesRef.current?.handleRefresh()
|
return new Promise((resolve, reject) => {
|
||||||
portraitRef.current?.handleRefresh()
|
generateProfile(id)
|
||||||
interestAreasRef.current?.handleRefresh()
|
.then(() => {
|
||||||
habitsRef.current?.handleRefresh()
|
preferencesRef.current?.handleRefresh()
|
||||||
})
|
portraitRef.current?.handleRefresh()
|
||||||
|
interestAreasRef.current?.handleRefresh()
|
||||||
|
habitsRef.current?.handleRefresh()
|
||||||
|
resolve(true)
|
||||||
|
})
|
||||||
|
.catch((error) => {
|
||||||
|
reject(error)
|
||||||
|
})
|
||||||
|
})
|
||||||
}
|
}
|
||||||
useImperativeHandle(ref, () => ({
|
useImperativeHandle(ref, () => ({
|
||||||
handleRefresh
|
handleRefresh
|
||||||
|
|||||||
@@ -13,11 +13,20 @@ const StatementDetail = forwardRef((_props, ref) => {
|
|||||||
const { id } = useParams()
|
const { id } = useParams()
|
||||||
const suggestionsRef = useRef<{ handleRefresh: () => void; }>(null)
|
const suggestionsRef = useRef<{ handleRefresh: () => void; }>(null)
|
||||||
const handleRefresh = () => {
|
const handleRefresh = () => {
|
||||||
if (!id) return
|
if (!id) {
|
||||||
generateSuggestions(id)
|
return Promise.resolve()
|
||||||
.then(() => {
|
}
|
||||||
suggestionsRef.current?.handleRefresh()
|
|
||||||
})
|
return new Promise((resolve, reject) => {
|
||||||
|
generateSuggestions(id)
|
||||||
|
.then(() => {
|
||||||
|
suggestionsRef.current?.handleRefresh()
|
||||||
|
resolve(true)
|
||||||
|
})
|
||||||
|
.catch((error) => {
|
||||||
|
reject(error)
|
||||||
|
})
|
||||||
|
})
|
||||||
}
|
}
|
||||||
useImperativeHandle(ref, () => ({
|
useImperativeHandle(ref, () => ({
|
||||||
handleRefresh
|
handleRefresh
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ import { type FC, useEffect, useState, useMemo } from 'react'
|
|||||||
import clsx from 'clsx'
|
import clsx from 'clsx'
|
||||||
import { useTranslation } from 'react-i18next'
|
import { useTranslation } from 'react-i18next'
|
||||||
import { useParams } from 'react-router-dom'
|
import { useParams } from 'react-router-dom'
|
||||||
import { Row, Col, Select, Form, Space, Skeleton, Input, Button, Divider } from 'antd'
|
import { Row, Col, Skeleton, Button, Divider, Tooltip } from 'antd'
|
||||||
import RbCard from '@/components/RbCard/Card'
|
import RbCard from '@/components/RbCard/Card'
|
||||||
import {
|
import {
|
||||||
getConversations,
|
getConversations,
|
||||||
@@ -10,7 +10,6 @@ import {
|
|||||||
getConversationDetail,
|
getConversationDetail,
|
||||||
} from '@/api/memory'
|
} from '@/api/memory'
|
||||||
import { formatDateTime } from '@/utils/format'
|
import { formatDateTime } from '@/utils/format'
|
||||||
import Tag from '@/components/Tag'
|
|
||||||
import RbAlert from '@/components/RbAlert'
|
import RbAlert from '@/components/RbAlert'
|
||||||
import Empty from '@/components/Empty'
|
import Empty from '@/components/Empty'
|
||||||
import ChatContent from '@/components/Chat/ChatContent'
|
import ChatContent from '@/components/Chat/ChatContent'
|
||||||
@@ -33,7 +32,6 @@ interface Detail {
|
|||||||
const WorkingDetail: FC = () => {
|
const WorkingDetail: FC = () => {
|
||||||
const { t } = useTranslation()
|
const { t } = useTranslation()
|
||||||
const { id } = useParams()
|
const { id } = useParams()
|
||||||
const [form] = Form.useForm()
|
|
||||||
const [loading, setLoading] = useState<boolean>(false)
|
const [loading, setLoading] = useState<boolean>(false)
|
||||||
const [data, setData] = useState<Conversation[]>([])
|
const [data, setData] = useState<Conversation[]>([])
|
||||||
const [messagesLoading, setMessagesLoading] = useState<boolean>(false)
|
const [messagesLoading, setMessagesLoading] = useState<boolean>(false)
|
||||||
@@ -110,13 +108,15 @@ const WorkingDetail: FC = () => {
|
|||||||
<div className="rb:h-full! rb:border-r rb:border-[#EAECEE] rb:py-3 rb:px-4">
|
<div className="rb:h-full! rb:border-r rb:border-[#EAECEE] rb:py-3 rb:px-4">
|
||||||
{data.map(item => (
|
{data.map(item => (
|
||||||
<div key={item.id} className="rb:mb-3">
|
<div key={item.id} className="rb:mb-3">
|
||||||
<div className={clsx("rb:p-[8px_13px] rb:rounded-lg rb:leading-5 rb:cursor-pointer rb:hover:bg-[#F0F3F8]", {
|
<Tooltip title={item.title}>
|
||||||
'rb:bg-[#FFFFFF] rb:shadow-[0px_2px_4px_0px_rgba(0,0,0,0.15)] rb:font-medium rb:hover:bg-[#FFFFFF]!': item.id === selected?.id,
|
<div className={clsx("rb:p-[8px_13px] rb:rounded-lg rb:leading-5 rb:cursor-pointer rb:hover:bg-[#F0F3F8] rb:text-ellipsis rb:overflow-hidden rb:whitespace-nowrap", {
|
||||||
})}
|
'rb:bg-[#FFFFFF] rb:shadow-[0px_2px_4px_0px_rgba(0,0,0,0.15)] rb:font-medium rb:hover:bg-[#FFFFFF]!': item.id === selected?.id,
|
||||||
onClick={() => setSelected(item)}
|
})}
|
||||||
>
|
onClick={() => setSelected(item)}
|
||||||
{item.title}
|
>
|
||||||
</div>
|
{item.title}
|
||||||
|
</div>
|
||||||
|
</Tooltip>
|
||||||
</div>
|
</div>
|
||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ import { type FC, useEffect, useState, useMemo, useRef } from 'react'
|
|||||||
import { useParams, useNavigate } from 'react-router-dom'
|
import { useParams, useNavigate } from 'react-router-dom'
|
||||||
import { useTranslation } from 'react-i18next'
|
import { useTranslation } from 'react-i18next'
|
||||||
import { Dropdown, Button } from 'antd'
|
import { Dropdown, Button } from 'antd'
|
||||||
|
import { LoadingOutlined } from '@ant-design/icons';
|
||||||
|
|
||||||
import PageHeader from '../components/PageHeader'
|
import PageHeader from '../components/PageHeader'
|
||||||
import StatementDetail from './StatementDetail'
|
import StatementDetail from './StatementDetail'
|
||||||
@@ -46,18 +47,30 @@ const Detail: FC = () => {
|
|||||||
const onClick = ({ key }: { key: string }) => {
|
const onClick = ({ key }: { key: string }) => {
|
||||||
navigate(`/user-memory/detail/${id}/${key}`, { replace: true })
|
navigate(`/user-memory/detail/${id}/${key}`, { replace: true })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const [loading, setLoading] = useState(false)
|
||||||
const handleRefresh = () => {
|
const handleRefresh = () => {
|
||||||
|
setLoading(true)
|
||||||
|
let response: any = null
|
||||||
switch(type) {
|
switch(type) {
|
||||||
case 'FORGET_MEMORY':
|
case 'FORGET_MEMORY':
|
||||||
forgetDetailRef.current?.handleRefresh()
|
forgetDetailRef.current?.handleRefresh()
|
||||||
break;
|
break;
|
||||||
case 'EMOTIONAL_MEMORY':
|
case 'EMOTIONAL_MEMORY':
|
||||||
statementDetailRef.current?.handleRefresh()
|
response = statementDetailRef.current?.handleRefresh()
|
||||||
break
|
break
|
||||||
case 'IMPLICIT_MEMORY':
|
case 'IMPLICIT_MEMORY':
|
||||||
implicitDetailRef.current?.handleRefresh()
|
response = implicitDetailRef.current?.handleRefresh()
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (response instanceof Promise) {
|
||||||
|
response.finally(() => {
|
||||||
|
setLoading(false)
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
setLoading(false)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (type === 'GRAPH') {
|
if (type === 'GRAPH') {
|
||||||
@@ -80,8 +93,8 @@ const Detail: FC = () => {
|
|||||||
</Dropdown>
|
</Dropdown>
|
||||||
}
|
}
|
||||||
extra={['FORGET_MEMORY', 'EMOTIONAL_MEMORY', 'IMPLICIT_MEMORY'].includes(type as string) &&
|
extra={['FORGET_MEMORY', 'EMOTIONAL_MEMORY', 'IMPLICIT_MEMORY'].includes(type as string) &&
|
||||||
<Button type="primary" ghost className="rb:group rb:h-6! rb:px-2!" onClick={handleRefresh}>
|
<Button type="primary" ghost size="small" className="rb:h-6! rb:px-2! rb:leading-5.5!" loading={loading} onClick={handleRefresh}>
|
||||||
<img src={refreshIcon} className="rb:w-4 rb:h-4" />
|
{!loading && <img src={refreshIcon} className="rb:w-4 rb:h-4" /> }
|
||||||
{t('common.refresh')}
|
{t('common.refresh')}
|
||||||
</Button>}
|
</Button>}
|
||||||
/>
|
/>
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import { Form, Button, Select, Space, Divider, InputNumber, Radio, type SelectPr
|
|||||||
import type { Suggestion } from '../../Editor/plugin/AutocompletePlugin'
|
import type { Suggestion } from '../../Editor/plugin/AutocompletePlugin'
|
||||||
import VariableSelect from '../VariableSelect'
|
import VariableSelect from '../VariableSelect'
|
||||||
import Editor from '../../Editor'
|
import Editor from '../../Editor'
|
||||||
import { edgeAttrs } from '../../../constant'
|
import { edgeAttrs, portArgs } from '../../../constant'
|
||||||
|
|
||||||
interface CaseListProps {
|
interface CaseListProps {
|
||||||
value?: Array<{ logical_operator: 'and' | 'or'; expressions: { left: string; operator: string; right: string; input_type?: string; }[] }>;
|
value?: Array<{ logical_operator: 'and' | 'or'; expressions: { left: string; operator: string; right: string; input_type?: string; }[] }>;
|
||||||
@@ -92,6 +92,7 @@ const CaseList: FC<CaseListProps> = ({
|
|||||||
selectedNode.addPort({
|
selectedNode.addPort({
|
||||||
id: 'CASE1',
|
id: 'CASE1',
|
||||||
group: 'right',
|
group: 'right',
|
||||||
|
args: portArgs,
|
||||||
attrs: { text: { text: 'IF', fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: 'IF', fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -100,6 +101,7 @@ const CaseList: FC<CaseListProps> = ({
|
|||||||
selectedNode.addPort({
|
selectedNode.addPort({
|
||||||
id: `CASE${i + 1}`,
|
id: `CASE${i + 1}`,
|
||||||
group: 'right',
|
group: 'right',
|
||||||
|
args: portArgs,
|
||||||
attrs: { text: { text: 'ELIF', fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: 'ELIF', fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -108,6 +110,7 @@ const CaseList: FC<CaseListProps> = ({
|
|||||||
selectedNode.addPort({
|
selectedNode.addPort({
|
||||||
id: `CASE${caseCount + 1}`,
|
id: `CASE${caseCount + 1}`,
|
||||||
group: 'right',
|
group: 'right',
|
||||||
|
args: portArgs,
|
||||||
attrs: { text: { text: 'ELSE', fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: 'ELSE', fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import { Graph, Node } from '@antv/x6';
|
|||||||
|
|
||||||
import Editor from '../../Editor';
|
import Editor from '../../Editor';
|
||||||
import type { Suggestion } from '../../Editor/plugin/AutocompletePlugin'
|
import type { Suggestion } from '../../Editor/plugin/AutocompletePlugin'
|
||||||
import { edgeAttrs } from '../../../constant'
|
import { edgeAttrs, portArgs } from '../../../constant'
|
||||||
|
|
||||||
interface CategoryListProps {
|
interface CategoryListProps {
|
||||||
parentName: string;
|
parentName: string;
|
||||||
@@ -55,7 +55,7 @@ const CategoryList: FC<CategoryListProps> = ({ parentName, selectedNode, graphRe
|
|||||||
selectedNode.addPort({
|
selectedNode.addPort({
|
||||||
id: `CASE${i + 1}`,
|
id: `CASE${i + 1}`,
|
||||||
group: 'right',
|
group: 'right',
|
||||||
args: i === 0 ? { dy: 24 } : undefined,
|
args: portArgs,
|
||||||
attrs: { text: { text: `分类${i + 1}`, fontSize: 12, fill: '#5B6167' } }
|
attrs: { text: { text: `分类${i + 1}`, fontSize: 12, fill: '#5B6167' } }
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -190,6 +190,12 @@ export const useVariableList = (
|
|||||||
if (iv?.dataType.startsWith('array[')) itemType = iv.dataType.replace(/^array\[(.+)\]$/, '$1');
|
if (iv?.dataType.startsWith('array[')) itemType = iv.dataType.replace(/^array\[(.+)\]$/, '$1');
|
||||||
addVariable(list, keys, `${pid}_item`, 'item', itemType, `${pid}.item`, pd);
|
addVariable(list, keys, `${pid}_item`, 'item', itemType, `${pid}.item`, pd);
|
||||||
addVariable(list, keys, `${pid}_index`, 'index', 'number', `${pid}.index`, pd);
|
addVariable(list, keys, `${pid}_index`, 'index', 'number', `${pid}.index`, pd);
|
||||||
|
} else if (pd.type === 'iteration' && !pd.config.input.defaultValue) {
|
||||||
|
let itemType = 'object';
|
||||||
|
const iv = list.find(v => `{{${v.value}}}` === pd.config.input.defaultValue);
|
||||||
|
if (iv?.dataType.startsWith('array[')) itemType = iv.dataType.replace(/^array\[(.+)\]$/, '$1');
|
||||||
|
addVariable(list, keys, `${pid}_item`, 'item', 'string', `${pid}.item`, pd);
|
||||||
|
addVariable(list, keys, `${pid}_index`, 'index', 'number', `${pid}.index`, pd);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -291,7 +291,69 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
return filteredList;
|
return filteredList;
|
||||||
}
|
}
|
||||||
if (nodeType === 'iteration' && key === 'output') {
|
if (nodeType === 'iteration' && key === 'output') {
|
||||||
return variableList.filter(variable => variable.value.includes('sys.'));
|
let filteredList = variableList.filter(variable => variable.value.includes('sys.'));
|
||||||
|
// Add child node output variables for loop nodes
|
||||||
|
if (selectedNode) {
|
||||||
|
const graph = graphRef.current;
|
||||||
|
if (graph) {
|
||||||
|
const nodes = graph.getNodes();
|
||||||
|
const childNodes = nodes.filter(node => {
|
||||||
|
const nodeData = node.getData();
|
||||||
|
return nodeData?.cycle === selectedNode.id;
|
||||||
|
});
|
||||||
|
|
||||||
|
// Add output variables from child nodes
|
||||||
|
childNodes.forEach(childNode => {
|
||||||
|
const childData = childNode.getData();
|
||||||
|
const childNodeId = childData.id;
|
||||||
|
|
||||||
|
// Add child node output variables based on their type
|
||||||
|
switch (childData.type) {
|
||||||
|
case 'llm':
|
||||||
|
case 'jinja-render':
|
||||||
|
case 'tool':
|
||||||
|
const outputKey = `${childNodeId}_output`;
|
||||||
|
const existingOutput = filteredList.find(v => v.key === outputKey);
|
||||||
|
if (!existingOutput) {
|
||||||
|
filteredList.push({
|
||||||
|
key: outputKey,
|
||||||
|
label: 'output',
|
||||||
|
type: 'variable',
|
||||||
|
dataType: 'string',
|
||||||
|
value: `${childNodeId}.output`,
|
||||||
|
nodeData: childData,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'http-request':
|
||||||
|
const bodyKey = `${childNodeId}_body`;
|
||||||
|
const statusKey = `${childNodeId}_status_code`;
|
||||||
|
if (!filteredList.find(v => v.key === bodyKey)) {
|
||||||
|
filteredList.push({
|
||||||
|
key: bodyKey,
|
||||||
|
label: 'body',
|
||||||
|
type: 'variable',
|
||||||
|
dataType: 'string',
|
||||||
|
value: `${childNodeId}.body`,
|
||||||
|
nodeData: childData,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (!filteredList.find(v => v.key === statusKey)) {
|
||||||
|
filteredList.push({
|
||||||
|
key: statusKey,
|
||||||
|
label: 'status_code',
|
||||||
|
type: 'variable',
|
||||||
|
dataType: 'number',
|
||||||
|
value: `${childNodeId}.status_code`,
|
||||||
|
nodeData: childData,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return filteredList;
|
||||||
}
|
}
|
||||||
if (nodeType === 'iteration') {
|
if (nodeType === 'iteration') {
|
||||||
return variableList.filter(variable => variable.dataType.includes('array'));
|
return variableList.filter(variable => variable.dataType.includes('array'));
|
||||||
@@ -411,7 +473,7 @@ const Properties: FC<PropertiesProps> = ({
|
|||||||
/>
|
/>
|
||||||
: selectedNode?.data?.type === 'tool'
|
: selectedNode?.data?.type === 'tool'
|
||||||
? <ToolConfig options={variableList} />
|
? <ToolConfig options={variableList} />
|
||||||
: selectedNode?.data.type === 'jinja-render'
|
: selectedNode?.data?.type === 'jinja-render'
|
||||||
? <JinjaRender
|
? <JinjaRender
|
||||||
selectedNode={selectedNode}
|
selectedNode={selectedNode}
|
||||||
options={getFilteredVariableList(selectedNode?.data?.type, 'mapping')}
|
options={getFilteredVariableList(selectedNode?.data?.type, 'mapping')}
|
||||||
|
|||||||
@@ -564,6 +564,7 @@ const defaultPortItems = [
|
|||||||
// { group: 'bottom' },
|
// { group: 'bottom' },
|
||||||
{ group: 'left' }
|
{ group: 'left' }
|
||||||
];
|
];
|
||||||
|
export const portArgs = { dy: 18 }
|
||||||
export const graphNodeLibrary: Record<string, NodeConfig> = {
|
export const graphNodeLibrary: Record<string, NodeConfig> = {
|
||||||
iteration: {
|
iteration: {
|
||||||
width: 240,
|
width: 240,
|
||||||
@@ -591,8 +592,8 @@ export const graphNodeLibrary: Record<string, NodeConfig> = {
|
|||||||
groups: defaultPortGroups,
|
groups: defaultPortGroups,
|
||||||
items: [
|
items: [
|
||||||
{ group: 'left' },
|
{ group: 'left' },
|
||||||
{ group: 'right', id: 'CASE1', args: { dy: 24 }, attrs: { text: { text: 'IF', fontSize: 12, color: '#5B6167' }} },
|
{ group: 'right', id: 'CASE1', args: portArgs, attrs: { text: { text: 'IF', fontSize: 12, color: '#5B6167' }} },
|
||||||
{ group: 'right', id: 'CASE2', attrs: { text: { text: 'ELSE', fontSize: 12, color: '#5B6167' }} }
|
{ group: 'right', id: 'CASE2', args: portArgs, attrs: { text: { text: 'ELSE', fontSize: 12, color: '#5B6167' }} }
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -604,8 +605,8 @@ export const graphNodeLibrary: Record<string, NodeConfig> = {
|
|||||||
groups: defaultPortGroups,
|
groups: defaultPortGroups,
|
||||||
items: [
|
items: [
|
||||||
{ group: 'left' },
|
{ group: 'left' },
|
||||||
{ group: 'right', id: 'CASE1', args: { dy: 24 }, attrs: { text: { text: '分类1', fontSize: 12, color: '#5B6167' } } },
|
{ group: 'right', id: 'CASE1', args: portArgs, attrs: { text: { text: '分类1', fontSize: 12, color: '#5B6167' } } },
|
||||||
{ group: 'right', id: 'CASE2', attrs: { text: { text: '分类2', fontSize: 12, color: '#5B6167' } } }
|
{ group: 'right', id: 'CASE2', args: portArgs, attrs: { text: { text: '分类2', fontSize: 12, color: '#5B6167' } } }
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import { App } from 'antd'
|
|||||||
import { Graph, Node, MiniMap, Snapline, Clipboard, Keyboard, type Edge } from '@antv/x6';
|
import { Graph, Node, MiniMap, Snapline, Clipboard, Keyboard, type Edge } from '@antv/x6';
|
||||||
import { register } from '@antv/x6-react-shape';
|
import { register } from '@antv/x6-react-shape';
|
||||||
|
|
||||||
import { nodeRegisterLibrary, graphNodeLibrary, nodeLibrary, portMarkup, portAttrs, edgeAttrs, edge_color, edge_selected_color } from '../constant';
|
import { nodeRegisterLibrary, graphNodeLibrary, nodeLibrary, portMarkup, portAttrs, edgeAttrs, edge_color, edge_selected_color, portArgs } from '../constant';
|
||||||
import type { WorkflowConfig, NodeProperties, ChatVariable } from '../types';
|
import type { WorkflowConfig, NodeProperties, ChatVariable } from '../types';
|
||||||
import { getWorkflowConfig, saveWorkflowConfig } from '@/api/application'
|
import { getWorkflowConfig, saveWorkflowConfig } from '@/api/application'
|
||||||
import type { PortMetadata } from '@antv/x6/lib/model/port';
|
import type { PortMetadata } from '@antv/x6/lib/model/port';
|
||||||
@@ -132,7 +132,7 @@ export const useWorkflowGraph = ({
|
|||||||
|
|
||||||
const portItems: PortMetadata[] = [
|
const portItems: PortMetadata[] = [
|
||||||
{ group: 'left' },
|
{ group: 'left' },
|
||||||
{ group: 'right', id: 'CASE1', args: { dy: 24 }, attrs: { text: { text: 'IF', fontSize: 12, fill: '#5B6167' }} }
|
{ group: 'right', id: 'CASE1', args: portArgs, attrs: { text: { text: 'IF', fontSize: 12, fill: '#5B6167' }} }
|
||||||
];
|
];
|
||||||
|
|
||||||
// 添加 ELIF 端口
|
// 添加 ELIF 端口
|
||||||
@@ -140,6 +140,7 @@ export const useWorkflowGraph = ({
|
|||||||
portItems.push({
|
portItems.push({
|
||||||
group: 'right',
|
group: 'right',
|
||||||
id: `CASE${i + 1}`,
|
id: `CASE${i + 1}`,
|
||||||
|
args: portArgs,
|
||||||
attrs: { text: { text: 'ELIF', fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: 'ELIF', fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -148,6 +149,7 @@ export const useWorkflowGraph = ({
|
|||||||
portItems.push({
|
portItems.push({
|
||||||
group: 'right',
|
group: 'right',
|
||||||
id: `CASE${caseCount + 1}`,
|
id: `CASE${caseCount + 1}`,
|
||||||
|
args: portArgs,
|
||||||
attrs: { text: { text: 'ELSE', fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: 'ELSE', fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -173,12 +175,12 @@ export const useWorkflowGraph = ({
|
|||||||
];
|
];
|
||||||
|
|
||||||
// 添加分类端口
|
// 添加分类端口
|
||||||
config.categories.forEach((category: any, index: number) => {
|
config.categories.forEach((_category: any, index: number) => {
|
||||||
portItems.push({
|
portItems.push({
|
||||||
group: 'right',
|
group: 'right',
|
||||||
id: `CASE${index + 1}`,
|
id: `CASE${index + 1}`,
|
||||||
args: index === 0 ? { dy: 24 } : undefined,
|
args: portArgs,
|
||||||
attrs: { text: { text: category.class_name || `分类${index + 1}`, fontSize: 12, fill: '#5B6167' }}
|
attrs: { text: { text: `分类${index + 1}`, fontSize: 12, fill: '#5B6167' }}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user