mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-08-01 02:46:02 +00:00
2453718acd
* fix(title): avoid default LLM call before stream end * fix(title): keep default fallback local * fix(title): harden fallback and replay e2e * docs(title): align fallback title behavior
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
"""Configuration for automatic thread title generation."""
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class TitleConfig(BaseModel):
|
|
"""Configuration for automatic thread title generation."""
|
|
|
|
enabled: bool = Field(
|
|
default=True,
|
|
description="Whether to enable automatic title generation",
|
|
)
|
|
max_words: int = Field(
|
|
default=6,
|
|
ge=1,
|
|
le=20,
|
|
description="Maximum number of words in the generated title",
|
|
)
|
|
max_chars: int = Field(
|
|
default=60,
|
|
ge=10,
|
|
le=200,
|
|
description="Maximum number of characters in the generated title",
|
|
)
|
|
model_name: str | None = Field(
|
|
default=None,
|
|
description="Model name to use for LLM title generation (None = use local fallback title)",
|
|
)
|
|
prompt_template: str = Field(
|
|
default=("Generate a concise title (max {max_words} words) for this conversation.\nUser: {user_msg}\nAssistant: {assistant_msg}\n\nReturn ONLY the title, no quotes, no explanation."),
|
|
description="Prompt template for LLM title generation when model_name is set",
|
|
)
|
|
|
|
|
|
# Global configuration instance
|
|
_title_config: TitleConfig = TitleConfig()
|
|
|
|
|
|
def get_title_config() -> TitleConfig:
|
|
"""Get the current title configuration."""
|
|
return _title_config
|
|
|
|
|
|
def set_title_config(config: TitleConfig) -> None:
|
|
"""Set the title configuration."""
|
|
global _title_config
|
|
_title_config = config
|
|
|
|
|
|
def load_title_config_from_dict(config_dict: dict) -> None:
|
|
"""Load title configuration from a dictionary."""
|
|
global _title_config
|
|
_title_config = TitleConfig(**config_dict)
|
|
|
|
|
|
def reset_title_config() -> None:
|
|
"""Restore the title configuration to its pristine ``TitleConfig()`` default.
|
|
|
|
Public API so that tests do not have to reach into the private
|
|
``_title_config`` module attribute. ``AppConfig.from_file()`` calls
|
|
:func:`load_title_config_from_dict`, which permanently mutates the
|
|
singleton; tests that need a clean slate between cases should call
|
|
this between tests.
|
|
"""
|
|
global _title_config
|
|
_title_config = TitleConfig()
|