fix(examples): migrate OSS Memory search/get_all calls to v3 filters API (#7300)

Co-authored-by: Shahil Kadia <nexiouscaliver@users.noreply.github.com>
This commit is contained in:
Shahil kadia
2026-09-22 19:01:33 +05:30
committed by GitHub
parent 8b38da9ab8
commit a214ec37bc
3 changed files with 11 additions and 11 deletions
+3 -3
View File
@@ -45,11 +45,11 @@ grok_client = OpenAI(
def recommend_movie_with_memory(user_id: str, user_query: str):
# Retrieve prior memory about movies
past_memories = memory.search("movie preferences", user_id=user_id)
past_memories = memory.search("movie preferences", filters={"user_id": user_id})
prompt = user_query
if past_memories:
prompt += f"\nPreviously, the user mentioned: {past_memories}"
if past_memories["results"]:
prompt += f"\nPreviously, the user mentioned: {[m['memory'] for m in past_memories['results']]}"
# Generate movie recommendation using Grok 3
response = grok_client.chat.completions.create(model="grok-3-beta", messages=[{"role": "user", "content": prompt}])
@@ -198,7 +198,7 @@ def search_memory_tool(query: str, user_id: str = "user") -> str:
Relevant vector memories found or message if none found
"""
try:
results = m.search(query, user_id=user_id)
results = m.search(query, filters={"user_id": user_id})
if isinstance(results, dict) and 'results' in results:
memory_list = results['results']
@@ -245,7 +245,7 @@ def search_graph_memory_tool(query: str, user_id: str = "user") -> str:
"""
try:
graph_query = f"relationships connections {query}"
results = m.search(graph_query, user_id=user_id)
results = m.search(graph_query, filters={"user_id": user_id})
if isinstance(results, dict) and 'results' in results:
memory_list = results['results']
@@ -290,7 +290,7 @@ def get_all_memories_tool(user_id: str = "user") -> str:
All memories for the user or message if none found
"""
try:
all_memories = m.get_all(user_id=user_id)
all_memories = m.get_all(filters={"user_id": user_id})
if isinstance(all_memories, dict) and 'results' in all_memories:
memory_list = all_memories['results']
+5 -5
View File
@@ -107,16 +107,16 @@ def main():
for query in search_queries:
print(f"\nQuery: {query}")
memories = memory.search(query=query, user_id="user_123")
memories = memory.search(query=query, filters={"user_id": "user_123"})
for memory_item in memories:
for memory_item in memories["results"]:
print(f" - {memory_item['memory']}")
print("\n--> Getting all memories for user...")
all_memories = memory.get_all(user_id="user_123")
print(f"Total memories stored: {len(all_memories)}")
all_memories = memory.get_all(filters={"user_id": "user_123"})
print(f"Total memories stored: {len(all_memories['results'])}")
for memory_item in all_memories:
for memory_item in all_memories["results"]:
print(f" - {memory_item['memory']}")
print("\n--> vLLM integration demo completed successfully!")