diff --git a/swarms/structs/conversation.py b/swarms/structs/conversation.py index 3e47cae1..6e11ebdd 100644 --- a/swarms/structs/conversation.py +++ b/swarms/structs/conversation.py @@ -184,7 +184,6 @@ class Conversation: system_prompt: Optional[str] = None, time_enabled: bool = False, autosave: bool = False, # Changed default to False - save_enabled: bool = False, # New parameter to control if saving is enabled save_filepath: str = None, load_filepath: str = None, # New parameter to specify which file to load from context_length: int = 8192, @@ -223,7 +222,6 @@ class Conversation: self.system_prompt = system_prompt self.time_enabled = time_enabled self.autosave = autosave - self.save_enabled = save_enabled self.conversations_dir = conversations_dir self.tokenizer_model_name = tokenizer_model_name self.message_id_on = message_id_on @@ -1021,13 +1019,6 @@ class Conversation: ) return - # Don't save if saving is disabled (你的PR代码) - if not self.save_enabled: - logger.warning( - "An attempt to save the conversation failed: save_enabled is False." - "Please set save_enabled=True when creating a Conversation object to enable saving." - ) - return # Get the full data including metadata and conversation history data = self.get_init_params() @@ -1284,7 +1275,7 @@ class Conversation: Returns: None """ - from swarms.utils.litellm_tokenizer import count_tokens + total_tokens = 0 truncated_history = [] @@ -1350,7 +1341,7 @@ class Conversation: Returns: str: Truncated text with token count not exceeding target_tokens """ - from swarms.utils.litellm_tokenizer import count_tokens + # If text is empty or target tokens is 0, return empty string if not text or target_tokens <= 0: @@ -1390,7 +1381,7 @@ class Conversation: return truncated_at_sentence return best_text - + def clear(self): """Clear the conversation history.""" if self.backend_instance: @@ -1796,4 +1787,4 @@ class Conversation: # # # conversation.add("assistant", "I am doing well, thanks.") # # # # print(conversation.to_json()) # # print(type(conversation.to_dict())) -# # print(conversation.to_yaml()) +# # print(conversation.to_yaml()) \ No newline at end of file