Spaces:
Paused
Paused
File size: 705 Bytes
8d1819a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from python.helpers.memory import Memory
from python.helpers.tool import Tool, Response
DEFAULT_THRESHOLD = 0.7
DEFAULT_LIMIT = 10
class MemoryLoad(Tool):
async def execute(self, query="", threshold=DEFAULT_THRESHOLD, limit=DEFAULT_LIMIT, filter="", **kwargs):
db = await Memory.get(self.agent)
docs = await db.search_similarity_threshold(query=query, limit=limit, threshold=threshold, filter=filter)
if len(docs) == 0:
result = self.agent.read_prompt("fw.memories_not_found.md", query=query)
else:
text = "\n\n".join(Memory.format_docs_plain(docs))
result = str(text)
return Response(message=result, break_loop=False)
|