feat: Add dynamic options support for block configurations and LLM model selection

Author: lss233

framework/llm/llm_manager.py

@@ -145,12 +145,11 @@ def get_llm(self, model_id: str) -> Optional[LLMBackendAdapter]:
             return None
         # TODO: 后续考虑支持更多的选择策略
         return random.choice(backends)
-
-    def get_llm_id_by_ability(self, ability: LLMAbility) -> str:
+    
+    def get_supported_models(self, ability: LLMAbility) -> List[str]:
         """
-        根据指定的能力获取严格符合要求的 LLM 适配器列表。
-        :param ability: 指定的能力。
-        :return: 符合要求的 LLM 适配器列表。
+        获取所有支持的模型
+        :return: 支持的模型列表
         """
         adapter_types = self.backend_registry.search_adapter_by_ability(ability)
         matched_backends = set()
@@ -162,5 +161,16 @@ def get_llm_id_by_ability(self, ability: LLMAbility) -> str:
                         break
 
         if not matched_backends:
+            return []
+        return list(matched_backends)
+
+    def get_llm_id_by_ability(self, ability: LLMAbility) -> str:
+        """
+        根据指定的能力获取严格符合要求的 LLM 适配器列表。
+        :param ability: 指定的能力。
+        :return: 符合要求的 LLM 适配器列表。
+        """
+        supported_models = self.get_supported_models(ability)
+        if not supported_models:
             return None
-        return random.choice(list(matched_backends))
+        return random.choice(supported_models)

framework/web/api/block/routes.py

@@ -18,17 +18,21 @@ async def list_block_types():
     for block_type in registry.get_all_types():
         inputs, outputs, configs = registry.extract_block_info(block_type)
         type_name = registry.get_block_type_name(block_type)
-        types.append(
-            BlockType(
-                type_name=type_name,
+
+        for config in configs.values():
+            if config.has_options:
+                config.options = config.options_provider(g.container, block_type)
+
+        block_type_info = BlockType(
+            type_name=type_name,
                 name=block_type.name,
                 label=registry.get_localized_name(type_name),
                 description=getattr(block_type, "description", ""),
                 inputs=inputs.values(),
                 outputs=outputs.values(),
                 configs=configs.values(),
-            )
         )
+        types.append(block_type_info)
 
     return BlockTypeList(types=types).model_dump()
 
@@ -69,6 +73,8 @@ async def get_block_type(type_name: str):
 
     configs = []
     for name, info in block_type.get_configs().items():
+        if info.has_options:
+            info.options = info.options_provider(g.container, block_type)
         configs.append(
             BlockConfig(
                 name=name,

framework/workflow/core/block/param.py

@@ -1,13 +1,25 @@
-from typing import Optional
+from typing import Callable, List, Optional, TypeVar
 
+from framework.ioc.container import DependencyContainer
+from framework.workflow.core.block import Block
+
+T = TypeVar("T")
+OptionsProvider = Callable[[DependencyContainer, Block], List[T]]
 
 class ParamMeta:
-    def __init__(self, label: Optional[str] = None, description: Optional[str] = None):
+    def __init__(self, label: Optional[str] = None, description: Optional[str] = None, options_provider: Optional[OptionsProvider[T]] = None):
         self.label = label
         self.description = description
+        self.options_provider = options_provider
 
     def __repr__(self):
-        return f"ParamMeta(label={self.label}, description={self.description})"
+        return f"ParamMeta(label={self.label}, description={self.description}, options_provider={self.options_provider})"
 
     def __str__(self):
         return self.__repr__()
+    
+    def get_options(self, block: Block) -> List[T]:
+        if self.options_provider:
+            return self.options_provider(block)
+        return []
+    

framework/workflow/core/block/registry.py

@@ -16,7 +16,8 @@ def extract_block_param(param: Parameter, type_system: TypeSystem) -> BlockConfi
     param_type = param.annotation
     label = param.name
     description = None
-
+    has_options = False
+    options_provider = None
     if get_origin(param_type) is Annotated:
         args = get_args(param_type)
         if len(args) > 0:
@@ -26,6 +27,8 @@ def extract_block_param(param: Parameter, type_system: TypeSystem) -> BlockConfi
             if isinstance(metadata, ParamMeta):
                 label = metadata.label
                 description = metadata.description
+                has_options = metadata.options_provider is not None
+                options_provider = metadata.options_provider
 
             # 递归调用 extract_block_param 处理实际类型
             block_config = extract_block_param(
@@ -54,6 +57,9 @@ def extract_block_param(param: Parameter, type_system: TypeSystem) -> BlockConfi
         required=required,
         default=default,
         label=label,
+        has_options=has_options,
+        options=[],
+        options_provider=options_provider,
     )
 
 
@@ -145,7 +151,7 @@ def extract_block_info(
         for name, input_info in getattr(block_type, "inputs", {}).items():
             type_name = self._type_system.get_type_name(input_info.data_type)
             self._type_system.register_type(type_name, input_info.data_type)
-            
+
             inputs[name] = BlockInput(
                 name=name,
                 label=input_info.label,
@@ -158,7 +164,7 @@ def extract_block_info(
         for name, output_info in getattr(block_type, "outputs", {}).items():
             type_name = self._type_system.get_type_name(output_info.data_type)
             self._type_system.register_type(type_name, output_info.data_type)
-            
+
             outputs[name] = BlockOutput(
                 name=name,
                 label=output_info.label,

framework/workflow/core/block/schema.py

@@ -1,6 +1,8 @@
-from typing import Any, Optional
+from typing import Any, List, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
+
+from framework.workflow.core.block.param import OptionsProvider
 
 
 class BlockInput(BaseModel):
@@ -32,3 +34,6 @@ class BlockConfig(BaseModel):
     required: bool = True
     default: Optional[Any] = None
     label: Optional[str] = None
+    has_options: bool = False
+    options: Optional[List[Any]] = None
+    options_provider: Optional[OptionsProvider] = Field(exclude=True)

framework/workflow/implementations/blocks/llm/chat.py

@@ -14,6 +14,10 @@
 from framework.workflow.core.execution.executor import WorkflowExecutor
 
 
+def model_name_options_provider(container: DependencyContainer, block: Block) -> List[str]:
+    llm_manager: LLMManager = container.resolve(LLMManager)
+    return llm_manager.get_supported_models(LLMAbility.TextChat)
+
 class ChatMessageConstructor(Block):
     name = "chat_message_constructor"
     inputs = {
@@ -124,7 +128,8 @@ class ChatCompletion(Block):
     def __init__(
         self,
         model_name: Annotated[
-            Optional[str], ParamMeta(label="模型 ID", description="要使用的模型 ID")
+            Optional[str],
+            ParamMeta(label="模型 ID", description="要使用的模型 ID", options_provider=model_name_options_provider),
         ] = None,
     ):
         self.model_name = model_name

framework/workflow/implementations/blocks/memory/chat_memory.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Any, Dict, Optional
+from typing import Annotated, Any, Dict, List, Optional
 
 from framework.im.message import IMMessage
 from framework.im.sender import ChatSender
@@ -10,6 +10,10 @@
 from framework.workflow.core.block import Block, Input, Output, ParamMeta
 
 
+def scope_type_options_provider(container: DependencyContainer, block: Block) -> List[str]:
+    return ["global", "user", "group"]
+
+
 class ChatMemoryQuery(Block):
     name = "chat_memory_query"
     inputs = {
@@ -23,7 +27,12 @@ class ChatMemoryQuery(Block):
     def __init__(
         self,
         scope_type: Annotated[
-            Optional[str], ParamMeta(label="级别", description="要查询记忆的级别")
+            Optional[str],
+            ParamMeta(
+                label="级别",
+                description="要查询记忆的级别",
+                options_provider=scope_type_options_provider,
+            ),
         ],
     ):
         self.scope_type = scope_type
@@ -63,7 +72,12 @@ class ChatMemoryStore(Block):
     def __init__(
         self,
         scope_type: Annotated[
-            Optional[str], ParamMeta(label="级别", description="要查询记忆的级别")
+            Optional[str],
+            ParamMeta(
+                label="级别",
+                description="要查询记忆的级别",
+                options_provider=scope_type_options_provider,
+            ),
         ],
     ):
         self.scope_type = scope_type