fix: Remove default completion params and omit empty params from model

config
This commit is contained in:
zhsama
2026-01-29 01:10:03 +08:00
parent 8326b9e3e5
commit 23ee9e618b
2 changed files with 33 additions and 44 deletions

View File

@ -53,15 +53,6 @@ export const GetCodeGeneratorResModal: FC<IGetCodeGeneratorResProps> = (
},
) => {
const { t } = useTranslation()
const defaultCompletionParams = {
temperature: 0.7,
max_tokens: 0,
top_p: 0,
echo: false,
stop: [],
presence_penalty: 0,
frequency_penalty: 0,
}
const localModel = localStorage.getItem('auto-gen-model')
? JSON.parse(localStorage.getItem('auto-gen-model') as string) as Model
: null
@ -69,7 +60,7 @@ export const GetCodeGeneratorResModal: FC<IGetCodeGeneratorResProps> = (
name: '',
provider: '',
mode: mode as unknown as ModelModeType.chat,
completion_params: defaultCompletionParams,
completion_params: {} as CompletionParams,
})
const {
defaultModel,
@ -134,12 +125,19 @@ export const GetCodeGeneratorResModal: FC<IGetCodeGeneratorResProps> = (
return
setLoadingTrue()
try {
const completionParams = model.completion_params
const modelConfig = Object.keys(completionParams).length > 0
? model
: (() => {
const { completion_params: _completion_params, ...rest } = model
return rest
})()
const { error, ...res } = await generateRule({
flow_id: flowId,
node_id: nodeId,
current: currentCode,
instruction,
model_config: model,
model_config: modelConfig,
ideal_output: ideaOutput,
language: languageMap[codeLanguages] || 'javascript',
})
@ -174,10 +172,7 @@ export const GetCodeGeneratorResModal: FC<IGetCodeGeneratorResProps> = (
if (localModel) {
setModel({
...localModel,
completion_params: {
...defaultCompletionParams,
...localModel.completion_params,
},
completion_params: (localModel.completion_params ?? {}) as CompletionParams,
})
}
else {

View File

@ -31,16 +31,6 @@ export type ContextGenerateChatMessage = ContextGenerateMessage & {
durationMs?: number
}
const defaultCompletionParams: CompletionParams = {
temperature: 0.7,
max_tokens: 4096,
top_p: 0.1,
echo: false,
stop: [],
presence_penalty: 0,
frequency_penalty: 0,
}
export const normalizeCodeLanguage = (value?: string) => {
if (value === CodeLanguage.javascript)
return CodeLanguage.javascript
@ -286,10 +276,7 @@ const useContextGenerate = ({
const parsed = JSON.parse(stored) as Model
return {
...parsed,
completion_params: {
...defaultCompletionParams,
...parsed.completion_params,
},
completion_params: (parsed.completion_params ?? {}) as CompletionParams,
}
})
@ -305,17 +292,32 @@ const useContextGenerate = ({
name: '',
provider: '',
mode: AppModeEnum.CHAT as unknown as ModelModeType.chat,
completion_params: defaultCompletionParams,
completion_params: {} as CompletionParams,
}
}
return {
name: defaultModel.model,
provider: defaultModel.provider.provider,
mode: AppModeEnum.CHAT as unknown as ModelModeType.chat,
completion_params: defaultCompletionParams,
completion_params: {} as CompletionParams,
}
}, [defaultModel, modelOverride])
const modelConfig = useMemo(() => {
const completionParams = model.completion_params
if (Object.keys(completionParams).length === 0) {
return {
provider: model.provider,
name: model.name,
}
}
return {
provider: model.provider,
name: model.name,
completion_params: completionParams,
}
}, [model.provider, model.name, model.completion_params])
const handleModelChange = useCallback((newValue: { modelId: string, provider: string, mode?: string, features?: string[] }) => {
const newModel = {
...model,
@ -365,7 +367,7 @@ const useContextGenerate = ({
const handleFetchSuggestedQuestions = useCallback(async () => {
if (!toolNodeId || !paramKey)
return
if (!model.name || !model.provider)
if (!modelConfig.name || !modelConfig.provider)
return
if (hasFetchedSuggestions || isFetchingSuggestions || !isInitView)
return
@ -377,9 +379,7 @@ const useContextGenerate = ({
const response = await fetchContextGenerateSuggestedQuestions({
language: promptLanguage,
model_config: {
provider: model.provider,
name: model.name,
completion_params: model.completion_params,
...modelConfig,
},
available_vars: availableVarsPayload,
parameter_info: parameterInfo,
@ -422,9 +422,7 @@ const useContextGenerate = ({
hasFetchedSuggestions,
isFetchingSuggestions,
isInitView,
model.completion_params,
model.name,
model.provider,
modelConfig,
paramKey,
parameterInfo,
promptLanguage,
@ -466,9 +464,7 @@ const useContextGenerate = ({
tool_call_id,
})),
model_config: {
provider: model.provider,
name: model.name,
completion_params: model.completion_params,
...modelConfig,
},
available_vars: availableVarsPayload,
parameter_info: parameterInfo,
@ -506,9 +502,7 @@ const useContextGenerate = ({
defaultAssistantMessage,
inputValue,
isGenerating,
model.completion_params,
model.name,
model.provider,
modelConfig,
paramKey,
parameterInfo,
promptMessages,