From 860e52c3fabbbf1019359443314f9d8ab82c7be0 Mon Sep 17 00:00:00 2001 From: Mike Lin Date: Sun, 22 Jun 2025 17:39:38 +0800 Subject: [PATCH] Enhance Memory class: Remove 'model_use' and 'time' in messages payload when user select free model from oepn router which result 500 error --- sources/memory.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/sources/memory.py b/sources/memory.py index d5df812..1b67e9e 100644 --- a/sources/memory.py +++ b/sources/memory.py @@ -7,10 +7,14 @@ import json from typing import List, Tuple, Type, Dict import torch from transformers import AutoTokenizer, AutoModelForSeq2SeqLM +import configparser from sources.utility import timer_decorator, pretty_print, animate_thinking from sources.logger import Logger +config = configparser.ConfigParser() +config.read('config.ini') + class Memory(): """ Memory is a class for managing the conversation memory @@ -162,7 +166,10 @@ class Memory(): if self.memory[curr_idx-1]['content'] == content: pretty_print("Warning: same message have been pushed twice to memory", color="error") time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") - self.memory.append({'role': role, 'content': content, 'time': time_str, 'model_used': self.model_provider}) + if config["MAIN"]["provider_name"] == "openrouter": + self.memory.append({'role': role, 'content': content}) + else: + self.memory.append({'role': role, 'content': content, 'time': time_str, 'model_used': self.model_provider}) return curr_idx-1 def clear(self) -> None: