xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-09 15:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: model testing

Browse Source
This commit is contained in:
Junyan Qin
2025-08-07 10:01:04 +08:00
parent 0f239ace17
commit 02dbe80d2f
6 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
web/src/app/home/models/component/embedding-form/EmbeddingForm.tsx
View File

@@ -298,6 +298,18 @@ export default function EmbeddingForm({
function testEmbeddingModelInForm() {
setModelTesting(true);
const extraArgsObj: Record<string, string | number | boolean> = {};
form
.getValues('extra_args')
?.forEach((arg: { key: string; type: string; value: string }) => {
if (arg.type === 'number') {
extraArgsObj[arg.key] = Number(arg.value);
} else if (arg.type === 'boolean') {
extraArgsObj[arg.key] = arg.value === 'true';
} else {
extraArgsObj[arg.key] = arg.value;
}
});
httpClient
.testEmbeddingModel('_', {
uuid: '',
@@ -309,6 +321,7 @@ export default function EmbeddingForm({
timeout: 120,
},
api_keys: [form.getValues('api_key')],
extra_args: extraArgsObj,
})
.then((res) => {
console.log(res);

14
web/src/app/home/models/component/llm-form/LLMForm.tsx
View File

@@ -312,6 +312,18 @@ export default function LLMForm({
function testLLMModelInForm() {
setModelTesting(true);
const extraArgsObj: Record<string, string | number | boolean> = {};
form
.getValues('extra_args')
?.forEach((arg: { key: string; type: string; value: string }) => {
if (arg.type === 'number') {
extraArgsObj[arg.key] = Number(arg.value);
} else if (arg.type === 'boolean') {
extraArgsObj[arg.key] = arg.value === 'true';
} else {
extraArgsObj[arg.key] = arg.value;
}
});
httpClient
.testLLMModel('_', {
uuid: '',
@@ -324,7 +336,7 @@ export default function LLMForm({
},
api_keys: [form.getValues('api_key')],
abilities: form.getValues('abilities'),
extra_args: form.getValues('extra_args'),
extra_args: extraArgsObj,
})
.then((res) => {
console.log(res);

2
web/src/app/home/pipelines/components/debug-dialog/DebugDialog.tsx
View File

@@ -257,7 +257,7 @@ export default function DebugDialog({
sessionType,
messageChain,
selectedPipelineId,
120000,
180000,
);
setMessages((prevMessages) => [...prevMessages, response.message]);