Merge pull request #614 from bjwswang/pypi

fix: set model in QAProviderZhiPuAIOnline
kubeagi · Jan 23, 2024 · 83ebf9e · 83ebf9e
2 parents 93be6a6 + d5775c4
commit 83ebf9e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/controllers/base/llm_controller.go b/controllers/base/llm_controller.go
@@ -137,7 +137,7 @@ func (r *LLMReconciler) SetupWithManager(mgr ctrl.Manager) error {
 				// Avoid to handle the event that it's not spec update or delete
 				oldLLM := ue.ObjectOld.(*arcadiav1alpha1.LLM)
 				newLLM := ue.ObjectNew.(*arcadiav1alpha1.LLM)
-				return !reflect.DeepEqual(oldLLM.Spec, newLLM.Spec) || oldLLM.DeletionTimestamp != nil
+				return !reflect.DeepEqual(oldLLM.Spec, newLLM.Spec) || newLLM.DeletionTimestamp != nil
 			},
 		})).
 		Watches(&source.Kind{Type: &arcadiav1alpha1.Worker{}},

diff --git a/pypi/data-processing/src/llm_api_service/qa_provider_zhi_pu_ai_online.py b/pypi/data-processing/src/llm_api_service/qa_provider_zhi_pu_ai_online.py
@@ -81,7 +81,7 @@ def generate_qa_list(
                     break
 
                 response = zhipuai.model_api.invoke(
-                    model="chatglm_6b",
+                    model=model,
                     prompt=[{"role": "user", "content": content}],
                     top_p=float(top_p),
                     temperature=float(temperature),