ctih1/goober
1
0
Fork 0
forked from gooberinc/goober
Code Pull requests Activity

cleaned up bot.py seperating it into different files, added formatting ignore rules for key_compiler, modified prestart checks

Browse source
This commit is contained in:
ctih1 2025-07-23 14:51:57 +03:00
parent f186e079da
commit 92dbc06b26
9 changed files with 311 additions and 264 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/markovmemory.py
View file

@ -45,16 +45,19 @@ def save_memory(memory):
json.dump(memory, f, indent=4)
def train_markov_model(memory, additional_data=None):
def train_markov_model(memory, additional_data=None) -> markovify.NewlineText | None:
if not memory:
return None
filtered_memory = [line for line in memory if isinstance(line, str)]
if additional_data:
filtered_memory.extend(
line for line in additional_data if isinstance(line, str)
)
if not filtered_memory:
return None
text = "\n".join(filtered_memory)
model = markovify.NewlineText(text, state_size=2)
return model