ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-29 01:40:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: template

Browse Source
This commit is contained in:
archer
2023-07-23 16:08:00 +08:00
parent 5e0b147048
commit b7d18e38d1
7 changed files with 923 additions and 1338 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
client/src/service/moduleDispatch/agent/classifyQuestion.ts
View File

@@ -51,7 +51,7 @@ export const dispatchClassifyQuestion = async (props: Record<string, any>): Prom
// function body
const agentFunction = {
name: agentFunName,
description: '判断用户问题的类型，并返回指定值',
description: '判断用户问题的类型属于哪方面，返回对应的枚举字段',
parameters: {
type: 'object',
properties: {
@@ -81,13 +81,9 @@ export const dispatchClassifyQuestion = async (props: Record<string, any>): Prom
const arg = JSON.parse(response.data.choices?.[0]?.message?.function_call?.arguments || '');
if (!arg.type) {
throw new Error('');
}
const tokens = response.data.usage?.total_tokens || 0;
const result = agents.find((item) => item.key === arg.type) || agents[0];
const result = agents.find((item) => item.key === arg?.type) || agents[0];
return {
[result.key]: 1,