wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-14 18:54:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

style: format

Browse Source
This commit is contained in:
Henry Li
2026-01-14 09:21:19 +08:00
parent 421488a991
commit 2aeaf7c965
5 changed files with 9 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/src/models/factory.py
View File

@@ -4,9 +4,7 @@ from src.config import get_app_config
from src.reflection import resolve_class
def create_chat_model(
name: str | None = None, thinking_enabled: bool = False, **kwargs
) -> BaseChatModel:
def create_chat_model(name: str | None = None, thinking_enabled: bool = False, **kwargs) -> BaseChatModel:
"""Create a chat model instance from the config.
Args:
@@ -35,9 +33,7 @@ def create_chat_model(
)
if thinking_enabled and model_config.when_thinking_enabled is not None:
if not model_config.supports_thinking:
raise ValueError(
f"Model {name} does not support thinking. Set `supports_thinking` to true in the `config.yaml` to enable thinking."
) from None
raise ValueError(f"Model {name} does not support thinking. Set `supports_thinking` to true in the `config.yaml` to enable thinking.") from None
model_settings_from_config.update(model_config.when_thinking_enabled)
model_instance = model_class(**kwargs, **model_settings_from_config)
return model_instance