yangyin
2024-12-13 c6cf46f3e11e60f692e0b8d12188458ad1b6120e
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
<template>
    <NodeBasicLayout
        v-model:title="data.title"
        :type="NodeType.LLM"
        :description="VueFlowHelper.getConfigValue(data, 'description', '调用大模型回答用户问题或者处理任务。')"
    >
        <Handle :id="targetHandleId" type="target" :position="Position.Left" />
        <FieldLayout :title="VueFlowHelper.getConfigValue(modelSetting, 'name', '模型设置')">
            <div class="w-[340px] flex-items-center gap-x-2">
                <div class="flex-column gap-y-1.5">
                    <span class="text-gray-500">{{ VueFlowHelper.getParams(modelSetting, 'llm_model').label }}</span>
 
                    <el-select
                        v-model="VueFlowHelper.getParams(modelSetting, 'llm_model').value"
                        class="w-[150px]"
                        :placeholder="VueFlowHelper.getParams(modelSetting, 'llm_model').placeholder"
                    >
                        <el-option v-for="item in VueFlowConstant.LLM_MODEL_LIST" :key="item.value" :label="item.label" :value="item.value" />
                    </el-select>
                </div>
                <div class="flex-column gap-y-1.5">
                    <span class="text-gray-500">{{ VueFlowHelper.getParams(modelSetting, 'temperature').label }}</span>
                    <el-input-number
                        v-model="VueFlowHelper.getParams(modelSetting, 'temperature').value"
                        class="w-[180px]"
                        placeholder=""
                    ></el-input-number>
                </div>
            </div>
        </FieldLayout>
        <FieldLayout :title="prompt.name">
            <el-input v-model="VueFlowHelper.getParams(prompt, 'prompt').value" type="textarea" :rows="3" autocomplete="off"> </el-input>
        </FieldLayout>
        <Handle :id="sourceHandleId" type="source" :position="Position.Right" />
    </NodeBasicLayout>
</template>
 
<script lang="ts" setup>
import { Handle, Position, useNode, useVueFlow } from '@vue-flow/core';
import { ref } from 'vue';
 
import type { NodeProps } from '@vue-flow/core';
import { VueFlowConstant } from '../../VueFlowConstant';
import { VueFlowHelper } from '../../VueFlowHelper';
import { LLMNodeData, LLMNodeEvents } from './index';
import { deepClone } from '/@/utils/other';
import { NodeType } from '../../vueFlowEnum';
 
import FieldLayout from './components/FieldLayout.vue';
import NodeBasicLayout from './components/NodeBasicLayout.vue';
 
defineProps<NodeProps<LLMNodeData, LLMNodeEvents>>();
 
const node = useNode();
const sourceHandleId = ref(VueFlowHelper.getHandleId(node.node, 'source'));
const targetHandleId = ref(VueFlowHelper.getHandleId(node.node, 'target'));
 
const data = ref(node.node.data);
 
const modelSetting = ref(VueFlowHelper.getGroupParam(data.value, 0));
const prompt = ref(VueFlowHelper.getGroupParam(data.value, 1));
 
 
 
VueFlowHelper.getConfigValue(VueFlowHelper.getParams(prompt.value, 'prompt'), 'label', '');
</script>