Skip to content

Commit 9b57b87

Browse files
committed
feat(rag): add use query transfer option, default useTopKey
1 parent 0771858 commit 9b57b87

File tree

1 file changed

+13
-3
lines changed
  • packages/vmind/__tests__/experiment/src/pages/ChartDialogueQA

1 file changed

+13
-3
lines changed

packages/vmind/__tests__/experiment/src/pages/ChartDialogueQA/qaPairRag.tsx

+13-3
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,13 @@ export function QARag() {
6969
useTopKey: boolean;
7070
topK: number;
7171
useSpec: boolean;
72+
useQueryTransfer: boolean;
7273
}>({
7374
topK: 5,
7475
useTopKey: true,
7576
useSpec: false,
76-
type: 'qa'
77+
type: 'qa',
78+
useQueryTransfer: true
7779
});
7880
const [llmTopKey, setLLmTopKey] = React.useState('');
7981
const [loading, setLoading] = React.useState(false);
@@ -332,9 +334,17 @@ export function QARag() {
332334
<Option value={3}>3</Option>
333335
<Option value={4}>4</Option>
334336
<Option value={5}>5</Option>
337+
<Option value={6}>6</Option>
338+
<Option value={7}>7</Option>
339+
<Option value={8}>8</Option>
340+
<Option value={9}>9</Option>
341+
<Option value={10}>10</Option>
335342
</Select>
336-
<Checkbox checked={ragOption.useTopKey} onChange={v => setRagOptions({ ...ragOption, useTopKey: v })}>
337-
Use Top Key(先用大模型找到topKey)
343+
<Checkbox
344+
checked={ragOption.useQueryTransfer}
345+
onChange={v => setRagOptions({ ...ragOption, useQueryTransfer: v })}
346+
>
347+
Use Query Transfer(问题黑话转译)
338348
</Checkbox>
339349
<Checkbox checked={ragOption.useSpec} onChange={v => setRagOptions({ ...ragOption, useSpec: v })}>
340350
Use Spec(传递关联的spec给大模型)

0 commit comments

Comments
 (0)