Run ruff
This commit is contained in:
@@ -39,9 +39,7 @@ class Mem0MemoryService(FrameProcessor):
|
||||
search_limit: int = Field(default=10, ge=1)
|
||||
search_threshold: float = Field(default=0.1, ge=0.0, le=1.0)
|
||||
api_version: str = Field(default="v2")
|
||||
system_prompt: str = Field(
|
||||
default="Based on previous conversations, I recall: \n\n"
|
||||
)
|
||||
system_prompt: str = Field(default="Based on previous conversations, I recall: \n\n")
|
||||
add_as_system_message: bool = Field(default=True)
|
||||
position: int = Field(default=1)
|
||||
|
||||
@@ -82,7 +80,11 @@ class Mem0MemoryService(FrameProcessor):
|
||||
"""
|
||||
try:
|
||||
logger.debug(f"Storing {len(messages)} messages in Mem0")
|
||||
params = {"messages": messages, "metadata": {"platform": "pipecat"}, "output_format": "v1.1"}
|
||||
params = {
|
||||
"messages": messages,
|
||||
"metadata": {"platform": "pipecat"},
|
||||
"output_format": "v1.1",
|
||||
}
|
||||
for id in ["user_id", "agent_id", "run_id"]:
|
||||
if getattr(self, id):
|
||||
params[id] = getattr(self, id)
|
||||
@@ -102,7 +104,11 @@ class Mem0MemoryService(FrameProcessor):
|
||||
"""
|
||||
try:
|
||||
logger.debug(f"Retrieving memories for query: {query}")
|
||||
id_pairs = [("user_id", self.user_id), ("agent_id", self.agent_id), ("run_id", self.run_id)]
|
||||
id_pairs = [
|
||||
("user_id", self.user_id),
|
||||
("agent_id", self.agent_id),
|
||||
("run_id", self.run_id),
|
||||
]
|
||||
clauses = [{name: value} for name, value in id_pairs if value is not None]
|
||||
filters = {"AND": clauses} if clauses else {}
|
||||
results = self.memory_client.search(
|
||||
|
||||
Reference in New Issue
Block a user