fix: llm model wrongly required when runner is not local-agent (#1386)

Author: RockChinQ

pkg/core/entities.py

@@ -133,7 +133,7 @@ class Conversation(pydantic.BaseModel):
 
     update_time: typing.Optional[datetime.datetime] = pydantic.Field(default_factory=datetime.datetime.now)
 
-    use_llm_model: requester.RuntimeLLMModel
+    use_llm_model: typing.Optional[requester.RuntimeLLMModel] = None
 
     use_funcs: typing.Optional[list[tools_entities.LLMFunction]]
 

pkg/pipeline/preproc/preproc.py

@@ -32,8 +32,18 @@ async def process(
         """处理"""
         session = await self.ap.sess_mgr.get_session(query)
 
+        # 非 local-agent 时，llm_model 为 None
+        llm_model = (
+            await self.ap.model_mgr.get_model_by_uuid(query.pipeline_config['ai']['local-agent']['model'])
+            if query.pipeline_config['ai']['runner'] == 'local-agent'
+            else None
+        )
+
         conversation = await self.ap.sess_mgr.get_conversation(
-            query, session, query.pipeline_config['ai']['local-agent']['prompt']
+            query,
+            session,
+            query.pipeline_config['ai']['local-agent']['prompt'],
+            llm_model,
         )
 
         # 设置query
@@ -43,16 +53,17 @@ async def process(
 
         query.use_llm_model = conversation.use_llm_model
 
-        query.use_funcs = (
-            conversation.use_funcs if query.use_llm_model.model_entity.abilities.__contains__('tool_call') else None
-        )
+        if query.pipeline_config['ai']['runner'] == 'local-agent':
+            query.use_funcs = (
+                conversation.use_funcs if query.use_llm_model.model_entity.abilities.__contains__('tool_call') else None
+            )
 
         query.variables = {
             'session_id': f'{query.session.launcher_type.value}_{query.session.launcher_id}',
             'conversation_id': conversation.uuid,
-            'msg_create_time': int(query.message_event.time)
-            if query.message_event.time
-            else int(datetime.datetime.now().timestamp()),
+            'msg_create_time': (
+                int(query.message_event.time) if query.message_event.time else int(datetime.datetime.now().timestamp())
+            ),
         }
 
         # Check if this model supports vision, if not, remove all images

pkg/provider/modelmgr/modelmgr.py

@@ -99,7 +99,7 @@ async def get_model_by_name(self, name: str) -> entities.LLMModelInfo:  # deprec
         for model in self.model_list:
             if model.name == name:
                 return model
-        raise ValueError(f'无法确定模型 {name} 的信息，请在元数据中配置')
+        raise ValueError(f'无法确定模型 {name} 的信息')
 
     async def get_model_by_uuid(self, uuid: str) -> entities.LLMModelInfo:
         """通过uuid获取模型"""

pkg/provider/session/sessionmgr.py

@@ -4,6 +4,7 @@
 
 from ...core import app, entities as core_entities
 from ...provider import entities as provider_entities
+from ...provider.modelmgr import entities as model_entities
 
 
 class SessionManager:
@@ -41,6 +42,7 @@ async def get_conversation(
         query: core_entities.Query,
         session: core_entities.Session,
         prompt_config: list[dict],
+        llm_model: model_entities.LLMModelInfo,
     ) -> core_entities.Conversation:
         """获取对话或创建对话"""
 
@@ -62,9 +64,7 @@ async def get_conversation(
             conversation = core_entities.Conversation(
                 prompt=prompt,
                 messages=[],
-                use_llm_model=await self.ap.model_mgr.get_model_by_uuid(
-                    query.pipeline_config['ai']['local-agent']['model']
-                ),
+                use_llm_model=llm_model,
                 use_funcs=await self.ap.tool_mgr.get_all_functions(
                     plugin_enabled=True,
                 ),