Feat : integration of more code interpreter

This commit is contained in:
martin legrand
2025-03-02 11:10:17 +01:00
parent 5bc2c26757
commit bba98786f5
3 changed files with 14 additions and 11 deletions
+1 -10
View File
@@ -5,6 +5,7 @@ import datetime
import uuid
import os
import json
from sources.utility import timer_decorator
class Memory():
"""
@@ -101,16 +102,6 @@ class Memory():
)
summary = self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
return summary
def timer_decorator(func):
from time import time
def wrapper(*args, **kwargs):
start_time = time()
result = func(*args, **kwargs)
end_time = time()
print(f"{func.__name__} took {end_time - start_time:.2f} seconds to execute")
return result
return wrapper
@timer_decorator
def compress(self) -> str: