refactor: remove constants and put in config file
removed all constants from all files and put them in src/control_backend/core/config.py also removed some old mock agents that we don't use anymore ref: N25B-236
This commit is contained in:
@@ -4,10 +4,16 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
|
||||
class ZMQSettings(BaseModel):
|
||||
internal_comm_address: str = "tcp://localhost:5560"
|
||||
ri_command_address: str = "tcp://localhost:0000"
|
||||
ri_communication_address: str = "tcp://*:5555"
|
||||
vad_agent_address: str = "tcp://localhost:5558"
|
||||
|
||||
|
||||
class AgentSettings(BaseModel):
|
||||
# connection settings
|
||||
host: str = "localhost"
|
||||
|
||||
# agent names
|
||||
bdi_core_agent_name: str = "bdi_core"
|
||||
belief_collector_agent_name: str = "belief_collector"
|
||||
text_belief_extractor_agent_name: str = "text_belief_extractor"
|
||||
@@ -15,14 +21,47 @@ class AgentSettings(BaseModel):
|
||||
llm_agent_name: str = "llm_agent"
|
||||
test_agent_name: str = "test_agent"
|
||||
transcription_agent_name: str = "transcription_agent"
|
||||
|
||||
ri_communication_agent_name: str = "ri_communication_agent"
|
||||
ri_command_agent_name: str = "ri_command_agent"
|
||||
|
||||
# default SPADE port
|
||||
default_spade_port: int = 5222
|
||||
|
||||
|
||||
class BehaviourSettings(BaseModel):
|
||||
default_rcv_timeout: float = 0.1
|
||||
llm_response_rcv_timeout: float = 1.0
|
||||
ping_sleep_s: float = 1.0
|
||||
comm_setup_max_retries: int = 5
|
||||
socket_poller_timeout_ms: int = 100
|
||||
|
||||
# VAD settings
|
||||
vad_prob_threshold: float = 0.5
|
||||
vad_initial_since_speech: int = 100
|
||||
vad_non_speech_patience_chunks: int = 3
|
||||
|
||||
# transcription behaviour
|
||||
transcription_max_concurrent_tasks: int = 3
|
||||
transcription_words_per_minute: int = 300
|
||||
transcription_words_per_token: float = 0.75 # (3 words = 4 tokens)
|
||||
|
||||
|
||||
class LLMSettings(BaseModel):
|
||||
local_llm_url: str = "http://localhost:1234/v1/chat/completions"
|
||||
local_llm_model: str = "openai/gpt-oss-20b"
|
||||
request_timeout_s: int = 120
|
||||
|
||||
|
||||
class VADSettings(BaseModel):
|
||||
repo_or_dir: str = "snakers4/silero-vad"
|
||||
model_name: str = "silero_vad"
|
||||
sample_rate_hz: int = 16000
|
||||
|
||||
|
||||
class SpeechModelSettings(BaseModel):
|
||||
# model identifiers for speech recognition
|
||||
mlx_model_name: str = "mlx-community/whisper-small.en-mlx"
|
||||
openai_model_name: str = "small.en"
|
||||
|
||||
|
||||
class Settings(BaseSettings):
|
||||
@@ -34,6 +73,12 @@ class Settings(BaseSettings):
|
||||
|
||||
agent_settings: AgentSettings = AgentSettings()
|
||||
|
||||
behaviour_settings: BehaviourSettings = BehaviourSettings()
|
||||
|
||||
vad_settings: VADSettings = VADSettings()
|
||||
|
||||
speech_model_settings: SpeechModelSettings = SpeechModelSettings()
|
||||
|
||||
llm_settings: LLMSettings = LLMSettings()
|
||||
|
||||
model_config = SettingsConfigDict(env_file=".env")
|
||||
|
||||
Reference in New Issue
Block a user