wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Run ruff

Browse Source
This commit is contained in:
Deshraj Yadav
2025-03-25 23:15:35 -07:00
parent e694971423
commit 1b8d326b49
1 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/pipecat/services/mem0.py
View File

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