tcsenpai/DualMind
1
0
Fork 0
mirror of https://github.com/tcsenpai/DualMind.git synced 2025-06-06 10:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

corrected repetitions

Browse Source
This commit is contained in:
tcsenpai 2024-10-06 21:16:44 +02:00
parent 71482432bd
commit 27344f599c
1 changed files with 35 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
ai_conversation.py
View File

@ -49,9 +49,19 @@ class AIConversation:
other_messages.append({"role": "assistant", "content": current_message})
#print(colored(f"Conversation with {name} ({self.current_model})", "blue"))
response = self.client.chat(model=self.current_model, messages=messages)
response = self.client.chat(
model=self.current_model,
messages=messages,
options={
"temperature": 0.7, # Adjust this value to control randomness
"repeat_penalty": 1.2, # Penalize repetition
}
)
response_content = response['message']['content']
# Post-process to remove repetition
response_content = self.remove_repetition(response_content)
model_name = f"{self.current_model.upper()} ({name}):"
formatted_response = f"{model_name}\n{response_content}\n"
@ -62,6 +72,8 @@ class AIConversation:
other_messages.append({"role": "user", "content": response_content})
current_message = response_content
# Switching the AI
self.current_model = self.model_2 if active_ai == 1 else self.model_1
active_ai = 1 if active_ai == 0 else 0
@ -101,3 +113,25 @@ class AIConversation:
f.write(log_content)
print(f"Conversation log saved to {filename}")
def remove_repetition(self, text):
# Split the text into sentences
split_tokens = [".", "!", "?"]
sentences = []
current_sentence = ""
for char in text:
current_sentence += char
if char in split_tokens:
sentences.append(current_sentence.strip())
current_sentence = ""
if current_sentence: # Add any remaining text
sentences.append(current_sentence.strip())
# Remove duplicates while preserving order
unique_sentences = []
for sentence in sentences:
if sentence not in unique_sentences:
unique_sentences.append(sentence)
# Join the sentences back together
return ' '.join(unique_sentences)