fix:llm select add model_name

pilot/openapi/api_v1/api_v1.py

@@ -149,6 +149,7 @@ async def dialogue_list(user_id: str = None):
         conv_uid = item.get("conv_uid")
         summary = item.get("summary")
         chat_mode = item.get("chat_mode")
+        model_name = item.get("model_name", CFG.LLM_MODEL)
 
         messages = json.loads(item.get("messages"))
         last_round = max(messages, key=lambda x: x["chat_order"])
@@ -160,6 +161,7 @@ async def dialogue_list(user_id: str = None):
             conv_uid=conv_uid,
             user_input=summary,
             chat_mode=chat_mode,
+            model_name=model_name,
             select_param=select_param,
         )
         dialogues.append(conv_vo)
@@ -259,6 +261,7 @@ def get_hist_messages(conv_uid: str):
     history_messages: List[OnceConversation] = history_mem.get_messages()
     if history_messages:
         for once in history_messages:
+            print(f"once:{once}")
             model_name = once.get("model_name", CFG.LLM_MODEL)
             once_message_vos = [
                 message2Vo(element, once["chat_order"], model_name) for element in once["messages"]

pilot/scene/base_chat.py

@@ -59,6 +59,7 @@ class BaseChat(ABC):
 
         self.history_message: List[OnceConversation] = self.memory.messages()
         self.current_message: OnceConversation = OnceConversation(self.chat_mode.value())
+        self.current_message.model_name = self.llm_model
         if chat_param["select_param"]:
             if len(self.chat_mode.param_types()) > 0:
                 self.current_message.param_type = self.chat_mode.param_types()[0]

pilot/scene/message.py

@@ -105,7 +105,7 @@ def _conversation_to_dic(once: OnceConversation) -> dict:
 
     return {
         "chat_mode": once.chat_mode,
-        "model_name": once.model_name if once.model_name else "proxyllm",
+        "model_name": once.model_name,
         "chat_order": once.chat_order,
         "start_date": start_str,
         "cost": once.cost if once.cost else 0,