chore: import upstream snapshot with attribution
Validate YAML Workflows / Validate YAML Configuration Files (push) Has been cancelled
Validate YAML Workflows / Validate YAML Configuration Files (push) Has been cancelled
This commit is contained in:
+26
@@ -0,0 +1,26 @@
|
||||
"""Register built-in thinking modes."""
|
||||
|
||||
from entity.configs.node.thinking import ReflectionThinkingConfig, ThinkingConfig
|
||||
from runtime.node.agent.thinking.thinking_manager import ThinkingManagerBase
|
||||
from runtime.node.agent.thinking.self_reflection import SelfReflectionThinkingManager
|
||||
from runtime.node.agent.thinking.registry import (
|
||||
register_thinking_mode,
|
||||
get_thinking_registration,
|
||||
)
|
||||
|
||||
register_thinking_mode(
|
||||
"reflection",
|
||||
config_cls=ReflectionThinkingConfig,
|
||||
manager_cls=SelfReflectionThinkingManager,
|
||||
summary="LLM reflects on its output and refine its output",
|
||||
)
|
||||
|
||||
|
||||
class ThinkingManagerFactory:
|
||||
@staticmethod
|
||||
def get_thinking_manager(config: ThinkingConfig) -> ThinkingManagerBase:
|
||||
registration = get_thinking_registration(config.type)
|
||||
typed_config = config.as_config(registration.config_cls)
|
||||
if not typed_config:
|
||||
raise ValueError(f"Invalid thinking config for type '{config.type}'")
|
||||
return registration.manager_cls(typed_config)
|
||||
Reference in New Issue
Block a user