Merge pull request #421 from Abirdcfly/main

fix: application graph cant update llm name
kubeagi · Dec 22, 2023 · ae47513 · ae47513
2 parents ce2a09b + 9321822
commit ae47513
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/apiserver/pkg/application/application.go b/apiserver/pkg/application/application.go
@@ -426,6 +426,7 @@ func UpdateApplicationConfig(ctx context.Context, c dynamic.Interface, input gen
 			qachain.Spec.MaxLength = pointer.IntDeref(input.MaxLength, qachain.Spec.MaxLength)
 			qachain.Spec.Temperature = pointer.Float64Deref(input.Temperature, qachain.Spec.Temperature)
 			qachain.Spec.Memory.ConversionWindowSize = pointer.IntDeref(input.ConversionWindowSize, qachain.Spec.Memory.ConversionWindowSize)
+			qachain.Spec.Input.LLM.Name = input.Llm
 		}, qachain); err != nil {
 			return nil, err
 		}
@@ -481,6 +482,7 @@ func UpdateApplicationConfig(ctx context.Context, c dynamic.Interface, input gen
 			llmchain.Spec.MaxLength = pointer.IntDeref(input.MaxLength, llmchain.Spec.MaxLength)
 			llmchain.Spec.Temperature = pointer.Float64Deref(input.Temperature, llmchain.Spec.Temperature)
 			llmchain.Spec.Memory.ConversionWindowSize = pointer.IntDeref(input.ConversionWindowSize, llmchain.Spec.Memory.ConversionWindowSize)
+			llmchain.Spec.Input.LLM.Name = input.Llm
 		}, llmchain); err != nil {
 			return nil, err
 		}