Unfucked gemini's mess
This commit is contained in:
@@ -149,6 +149,9 @@ class ShortTermMemory:
|
||||
# Current conversation topic
|
||||
current_topic: str | None = None
|
||||
|
||||
# Conversation language
|
||||
language: str = "en"
|
||||
|
||||
# History message limit
|
||||
max_history: int = 20
|
||||
|
||||
@@ -206,12 +209,18 @@ class ShortTermMemory:
|
||||
self.current_topic = topic
|
||||
logger.debug(f"STM: Topic -> {topic}")
|
||||
|
||||
def set_language(self, language: str) -> None:
|
||||
"""Set the conversation language."""
|
||||
self.language = language
|
||||
logger.debug(f"STM: Language -> {language}")
|
||||
|
||||
def clear(self) -> None:
|
||||
"""Reset short-term memory."""
|
||||
self.conversation_history = []
|
||||
self.current_workflow = None
|
||||
self.extracted_entities = {}
|
||||
self.current_topic = None
|
||||
self.language = "en"
|
||||
logger.info("STM: Cleared")
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
@@ -221,6 +230,7 @@ class ShortTermMemory:
|
||||
"current_workflow": self.current_workflow,
|
||||
"extracted_entities": self.extracted_entities,
|
||||
"current_topic": self.current_topic,
|
||||
"language": self.language,
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user