mirror of https://github.com/Mai-with-u/MaiBot.git
20 lines
784 B
Python
20 lines
784 B
Python
from src.common.logger import get_logger
|
|
from src.chat.heart_flow.frequency_control import frequency_control_manager
|
|
from src.config.config import global_config
|
|
|
|
logger = get_logger("frequency_api")
|
|
|
|
|
|
def get_current_talk_value(chat_id: str) -> float:
|
|
return frequency_control_manager.get_or_create_frequency_control(
|
|
chat_id
|
|
).get_talk_frequency_adjust() * global_config.chat.get_talk_value(chat_id)
|
|
|
|
|
|
def set_talk_frequency_adjust(chat_id: str, talk_frequency_adjust: float) -> None:
|
|
frequency_control_manager.get_or_create_frequency_control(chat_id).set_talk_frequency_adjust(talk_frequency_adjust)
|
|
|
|
|
|
def get_talk_frequency_adjust(chat_id: str) -> float:
|
|
return frequency_control_manager.get_or_create_frequency_control(chat_id).get_talk_frequency_adjust()
|