From 7ca5a005ebda581499a6ee3021da62c8ea3069fd Mon Sep 17 00:00:00 2001
From: Ceceliachenen <chenanyu.cay@alibaba-inc.com>
Date: Mon, 14 Oct 2024 10:03:01 +0800
Subject: [PATCH] fix llm settings (#243)

Co-authored-by: Yue Fei <luxun.fy@alibaba-inc.com>
---
 src/pai_rag/app/web/view_model.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/pai_rag/app/web/view_model.py b/src/pai_rag/app/web/view_model.py
index 5ded968e..312caeb2 100644
--- a/src/pai_rag/app/web/view_model.py
+++ b/src/pai_rag/app/web/view_model.py
@@ -207,11 +207,11 @@ def from_app_config(config):
         )
         if view_model.llm.lower() == "paieas":
             view_model.llm_eas_model_name = config["llm"].get(
-                "name", view_model.llm_eas_model_name
+                "model_name", view_model.llm_eas_model_name
             )
         else:
             view_model.llm_api_model_name = config["llm"].get(
-                "name", view_model.llm_api_model_name
+                "model_name", view_model.llm_api_model_name
             )
 
         view_model.use_mllm = config["llm"]["multi_modal"].get(
@@ -233,14 +233,14 @@ def from_app_config(config):
                 "view_model.mllm_eas_model_name",
                 view_model.mllm_eas_model_name,
                 "2",
-                config["llm"]["multi_modal"]["name"],
+                config["llm"]["multi_modal"]["model_name"],
             )
             view_model.mllm_eas_model_name = config["llm"]["multi_modal"].get(
-                "name", view_model.mllm_eas_model_name
+                "model_name", view_model.mllm_eas_model_name
             )
         else:
             view_model.mllm_api_model_name = config["llm"]["multi_modal"].get(
-                "name", view_model.mllm_api_model_name
+                "model_name", view_model.mllm_api_model_name
             )
 
         view_model.use_oss = config["oss_store"].get("enable", view_model.use_oss)
@@ -446,9 +446,9 @@ def to_app_config(self):
         config["llm"]["api_key"] = self.llm_api_key
         config["llm"]["temperature"] = self.llm_temperature
         if self.llm.lower() == "paieas":
-            config["llm"]["name"] = self.llm_eas_model_name
+            config["llm"]["model_name"] = self.llm_eas_model_name
         else:
-            config["llm"]["name"] = self.llm_api_model_name
+            config["llm"]["model_name"] = self.llm_api_model_name
 
         config["llm"]["multi_modal"]["enable"] = self.use_mllm
         config["llm"]["multi_modal"]["source"] = self.mllm
@@ -456,9 +456,9 @@ def to_app_config(self):
         config["llm"]["multi_modal"]["token"] = self.mllm_eas_token
         config["llm"]["multi_modal"]["api_key"] = self.mllm_api_key
         if self.mllm.lower() == "paieas":
-            config["llm"]["multi_modal"]["name"] = self.mllm_eas_model_name
+            config["llm"]["multi_modal"]["model_name"] = self.mllm_eas_model_name
         else:
-            config["llm"]["multi_modal"]["name"] = self.mllm_api_model_name
+            config["llm"]["multi_modal"]["model_name"] = self.mllm_api_model_name
 
         config["oss_store"]["enable"] = self.use_oss
         if os.getenv("OSS_ACCESS_KEY_ID") is None and self.oss_ak: