feat(web): Agent add ai prompt

web/src/views/ApplicationConfig/components/AiPromptModal.tsx

@@ -1,43 +1,136 @@
-import { forwardRef, useImperativeHandle, useState } from 'react';
-import { Row, Col, Space, Button } from 'antd';
+import { forwardRef, useImperativeHandle, useState, useRef } from 'react';
+import { Button, Form, Input, App, Row, Col } from 'antd';
 import { useTranslation } from 'react-i18next';
+import clsx from 'clsx'
+import copy from 'copy-to-clipboard';
 
-import type { AiPromptModalRef } from '../types'
-// import { request } from '@/utils/request'
+import { updatePromptMessages, createPromptSessions } from '@/api/prompt'
+import { getModelListUrl } from '@/api/models'
+import type { AiPromptModalRef, AiPromptVariableModalRef, AiPromptForm } from '../types'
 import RbModal from '@/components/RbModal'
-import Markdown from '@/components/Markdown';
+import type { Model } from '@/views/ModelManagement/types'
+import ChatContent from '@/components/Chat/ChatContent'
+import Empty from '@/components/Empty'
+import ChatSendIcon from '@/assets/images/application/chatSend.svg'
+import ConversationEmptyIcon from '@/assets/images/conversation/conversationEmpty.svg'
+import type { ChatItem } from '@/components/Chat/types' 
+import CustomSelect from '@/components/CustomSelect'
+import AiPromptVariableModal from './AiPromptVariableModal'
 
 interface AiPromptModalProps {
-  refresh: () => void;
+  refresh: (value: string) => void;
+  defaultModel: Model | null;
 }
 
 const AiPromptModal = forwardRef<AiPromptModalRef, AiPromptModalProps>(({
-  // refresh
+  refresh,
+  defaultModel,
 }, ref) => {
   const { t } = useTranslation();
+  const { message } = App.useApp()
   const [visible, setVisible] = useState(false);
   const [loading, setLoading] = useState(false)
-  const [content, setContent] = useState('');
+  const [form] = Form.useForm<AiPromptForm>()
+  const [chatList, setChatList] = useState<ChatItem[]>([])
+  const [variables, setVariables] = useState<string[]>([])
+  const [promptSession, setPromptSession] = useState<string | null>(null)
+  const aiPromptVariableModalRef = useRef<AiPromptVariableModalRef>(null)
+  const currentPromptRef = useRef<any>(null)
 
+  const values = Form.useWatch([], form)
 
   // 封装取消方法，添加关闭弹窗逻辑
   const handleClose = () => {
     setVisible(false);
     setLoading(false)
+    setChatList([])
+    setVariables([])
+    form.setFieldsValue({
+      message: undefined,
+      current_prompt: undefined,
+    })
   };
 
   const handleOpen = () => {
-    setVisible(true);
+    createPromptSessions()
+      .then(res => {
+        const response = res as { id: string }
+        setPromptSession(response.id)
+
+        if (!values.model_id && defaultModel?.id) {
+          form.setFieldValue('model_id', defaultModel?.id)
+        }
+        setVisible(true);
+      })
   };
-  // 封装保存方法，添加提交逻辑
-  // const handleSave = () => {
-  // }
+  const handleSend = () => {
+    if (!promptSession) return
+    if (!values.model_id) {
+      message.warning(t('common.selectPlaceholder', { title: t('application.model') }))
+      return
+    }
+    if (!values.message) {
+      message.warning(t('application.promptChatPlaceholder'))
+      return
+    }
+    const messageContent = values.message
+    setLoading(true)
+    setChatList(prev => {
+      return [...prev, { role: 'user', content: messageContent}]
+    })
+    form.setFieldsValue({ message: undefined })
+    updatePromptMessages(promptSession, values)
+      .then(res => {
+        const response = res as { prompt: string; desc: string; variables: string[] }
+        form.setFieldsValue({ current_prompt: response.prompt })
+        setChatList(prev => {
+          return [...prev, { role: 'assistant', content: response.desc }]
+        })
+        setVariables(response.variables)
+      })
+      .finally(() => {
+        setLoading(false)
+      })
+  }
+  const handleCopy = () => {
+    if (!values.current_prompt || values?.current_prompt?.trim() === '') return
+    copy(values.current_prompt)
+    message.success(t('common.copySuccess'))
+  }
+  const handleAdd = () => {
+    aiPromptVariableModalRef.current?.handleOpen()
+  }
+  const handleVariableApply = (value: string) => {
+    const textArea = currentPromptRef.current?.resizableTextArea?.textArea
+    if (textArea) {
+      const cursorPosition = textArea.selectionStart
+      const currentValue = values.current_prompt || ''
+      const newValue = currentValue.slice(0, cursorPosition) + value + currentValue.slice(cursorPosition)
+      form.setFieldValue('current_prompt', newValue)
+      
+      // 设置新的光标位置
+      setTimeout(() => {
+        textArea.focus()
+        textArea.setSelectionRange(cursorPosition + value.length, cursorPosition + value.length)
+      }, 0)
+    } else {
+      form.setFieldValue('current_prompt', (values.current_prompt || '') + value)
+    }
+  }
+  const handleApply = () => {
+    if (!values.current_prompt) {
+      return
+    }
+    refresh(values.current_prompt)
+    handleClose()
+  }
 
   // 暴露给父组件的方法
   useImperativeHandle(ref, () => ({
     handleOpen,
   }));
 
+  console.log(values)
   return (
     <RbModal
       title={t('application.AIPromptAssistant')}
@@ -46,34 +139,73 @@ const AiPromptModal = forwardRef<AiPromptModalRef, AiPromptModalProps>(({
       footer={null}
       width={1000}
     >
-      <Row className="rb:rounded-[12px] rb:border rb:border-[#DFE4ED]">
-        <Col span={12} className="rb:border-r rb:border-[#DFE4ED]">
-          <div className="rb:p-[12px_17px] rb:border-b rb:border-[#DFE4ED]">{t('application.generatedPrompt')}</div>
-          <div className="rb:h-[200px] rb:p-[16px]">
-            <div className="rb:bg-[#F0F3F8] rb:h-full rb:w-full">
-              <Markdown
-                content={content}
+      <Form form={form}>
+        <div className="rb:grid rb:grid-cols-2 rb:border-t rb:border-t-[#EBEBEB]">
+          <div className="rb:border-r rb:border-r-[#EBEBEB] rb:pr-6 rb:pt-3">
+            <Form.Item
+              label={t('application.model')}
+              name="model_id"
+              rules={[{ required: true, message: t('common.pleaseSelect') }]}
+            >
+              <CustomSelect
+                url={getModelListUrl}
+                params={{ type: 'llm,chat', pagesize: 100 }}
+                valueKey="id"
+                labelKey="name"
+                hasAll={false}
+                style={{ width: '100%' }}
               />
+            </Form.Item>
+
+            <ChatContent
+              classNames="rb:h-100.5 rb:px-[16px] rb:py-[20px] rb:bg-[#FBFDFF] rb:border rb:border-[#DFE4ED] rb:rounded-[8px]"
+              contentClassNames="rb:max-w-[260px]!"
+              empty={<Empty url={ConversationEmptyIcon} title={t('application.promptChatEmpty')} isNeedSubTitle={false} size={[240, 200]} className="rb:h-full" />}
+              data={chatList || []}
+              streamLoading={false}
+              labelPosition="top"
+              labelFormat={(item) => item.role === 'user' ? t('application.you') : t('application.ai')}
+            />
+
+            <div className="rb:flex rb:items-center rb:gap-2.5 rb:py-4">
+              <Form.Item name="message" className="rb:mb-0!" style={{ width: 'calc(100% - 54px)' }}>
+                <Input
+                  className="rb:h-11 rb:shadow-[0px_2px_8px_0px_rgba(33,35,50,0.1)]"
+                  placeholder={t('application.promptChatPlaceholder')}
+                  onPressEnter={handleSend}
+                />
+              </Form.Item>
+              <img src={ChatSendIcon} className={clsx("rb:w-11 rb:h-11 rb:cursor-pointer", {
+                'rb:opacity-50': loading,
+              })} onClick={handleSend} />
             </div>
           </div>
-        </Col>
-        <Col span={12}>
-          <div className="rb:p-[12px_17px] rb:border-b rb:border-[#DFE4ED]">{t('application.conversationOptimizationPrompt')}</div>
-          <div className="rb:h-[200px] rb:p-[16px]">
-            <div className="rb:bg-[#F0F3F8] rb:h-full rb:w-full">
-              
+
+          <div className="rb:pl-6 rb:pt-3">
+            <Row>
+              <Col span={12}>
+                <Form.Item label={t('application.conversationOptimizationPrompt')}></Form.Item>
+              </Col>
+              <Col span={12} className="rb:text-right">
+                <Button onClick={handleAdd}>+ {t('application.addVariable')}</Button>
+              </Col>
+            </Row>
+            <Form.Item name="current_prompt">
+              <Input.TextArea ref={currentPromptRef} className="rb:bg-[#FBFDFF]! rb:h-100.5!" />
+            </Form.Item>
+            <div className="rb:grid rb:grid-cols-2 rb:gap-4 rb:mt-6">
+              <Button block disabled={!values.current_prompt} onClick={handleCopy}>{t('common.copy')}</Button>
+              <Button type="primary" block disabled={!values.current_prompt} onClick={handleApply}>{t('application.apply')}</Button>
             </div>
           </div>
-        </Col>
-        <Col span={12} className="rb:border-r rb:border-[#DFE4ED]">
-          <Space>
-            <Button>{t('common.copy')}</Button>
-            <Button type="primary">{t('common.apply')}</Button>
-          </Space>
-        </Col>
-        <Col span={12}>
-        </Col>
-      </Row>
+        </div>
+      </Form>
+
+      <AiPromptVariableModal
+        ref={aiPromptVariableModalRef}
+        variables={variables}
+        refresh={handleVariableApply}
+      />
     </RbModal>
   );
 });

web/src/views/ApplicationConfig/components/AiPromptVariableModal.tsx

@@ -0,0 +1,107 @@
+import { forwardRef, useEffect, useImperativeHandle, useState } from 'react';
+import { Form, Input, App, Select, AutoComplete, type AutoCompleteProps } from 'antd';
+import { useTranslation } from 'react-i18next';
+
+import type { Application } from '@/views/ApplicationManagement/types'
+import type { AiPromptVariableModalRef } from '../types'
+import { createApiKey  } from '@/api/apiKey';
+import RbModal from '@/components/RbModal'
+
+const FormItem = Form.Item;
+
+interface AiPromptVariableModalProps {
+  refresh: (value: string) => void;
+  variables: string[];
+}
+
+const AiPromptVariableModal = forwardRef<AiPromptVariableModalRef, AiPromptVariableModalProps>(({
+  refresh,
+  variables
+}, ref) => {
+  const { t } = useTranslation();
+  const { message } = App.useApp();
+  const [visible, setVisible] = useState(false);
+  const [form] = Form.useForm();
+  const [loading, setLoading] = useState(false)
+  const [options, setOptions] = useState<AutoCompleteProps['options']>([])
+
+  useEffect(() => {
+    setOptions(variables.map(key => ({
+      value: key,
+      label: `{{${key}}}`
+    })))
+  }, [variables])
+  const handleSearch = (value: string) => {
+    const filterKeys = variables?.filter(key => key.includes(value))
+
+    if (filterKeys.length) {
+      setOptions(filterKeys.map(key => ({
+        value: key,
+        label: `{{${key}}}`
+      })))
+    } else {
+      setOptions([{
+        value: value,
+        label: `{{${value}}}`
+      }])
+    }
+  }
+
+  // 封装取消方法，添加关闭弹窗逻辑
+  const handleClose = () => {
+    setVisible(false);
+    form.resetFields();
+    setLoading(false)
+  };
+
+  const handleOpen = () => {
+    setVisible(true);
+      form.resetFields();
+  };
+  // 封装保存方法，添加提交逻辑
+  const handleSave = () => {
+    const variableName = form.getFieldValue('variableName')
+
+    if (!variableName) return
+
+    refresh(`{{${variableName}}}`)
+    handleClose()
+  }
+
+  // 暴露给父组件的方法
+  useImperativeHandle(ref, () => ({
+    handleOpen,
+    handleClose
+  }));
+
+  return (
+    <RbModal
+      title={t('application.addVariable')}
+      open={visible}
+      onCancel={handleClose}
+      confirmLoading={loading}
+      onOk={handleSave}
+      okText={t('application.apply')}
+    >
+      <Form
+        form={form}
+        layout="vertical"
+        scrollToFirstError={{ behavior: 'instant', block: 'end', focus: true }}
+      > 
+        <FormItem
+          name="variableName"
+          label={t('application.defineVariableName')}
+          extra={t('application.defineVariableNameExtra')}
+        >
+          <AutoComplete
+            placeholder={t('application.defineVariableNamePlaceholder')}
+            onSearch={handleSearch}
+            options={options}
+          />
+        </FormItem>
+      </Form>
+    </RbModal>
+  );
+});
+
+export default AiPromptVariableModal;

web/src/views/ApplicationConfig/components/Chat.tsx

@@ -361,7 +361,7 @@ const Chat: FC<ChatProps> = ({ chatList, data, updateChatList, handleSave, sourc
                 data={chat.list || []}
                 streamLoading={compareLoading}
                 labelPosition="top"
-                labelFormat={(item) => item.role === 'user' ? 'You' : chat.label}
+                labelFormat={(item) => item.role === 'user' ? t('application.you') : chat.label}
                 errorDesc={t('application.ReplyException')}
               />