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:
co-authored by
Shahil Kadia
parent
8b38da9ab8
commit
a214ec37bc
@@ -45,11 +45,11 @@ grok_client = OpenAI(
|
|||||||
|
|
||||||
def recommend_movie_with_memory(user_id: str, user_query: str):
|
def recommend_movie_with_memory(user_id: str, user_query: str):
|
||||||
# Retrieve prior memory about movies
|
# 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
|
prompt = user_query
|
||||||
if past_memories:
|
if past_memories["results"]:
|
||||||
prompt += f"\nPreviously, the user mentioned: {past_memories}"
|
prompt += f"\nPreviously, the user mentioned: {[m['memory'] for m in past_memories['results']]}"
|
||||||
|
|
||||||
# Generate movie recommendation using Grok 3
|
# Generate movie recommendation using Grok 3
|
||||||
response = grok_client.chat.completions.create(model="grok-3-beta", messages=[{"role": "user", "content": prompt}])
|
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
|
Relevant vector memories found or message if none found
|
||||||
"""
|
"""
|
||||||
try:
|
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:
|
if isinstance(results, dict) and 'results' in results:
|
||||||
memory_list = results['results']
|
memory_list = results['results']
|
||||||
@@ -245,7 +245,7 @@ def search_graph_memory_tool(query: str, user_id: str = "user") -> str:
|
|||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
graph_query = f"relationships connections {query}"
|
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:
|
if isinstance(results, dict) and 'results' in results:
|
||||||
memory_list = results['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
|
All memories for the user or message if none found
|
||||||
"""
|
"""
|
||||||
try:
|
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:
|
if isinstance(all_memories, dict) and 'results' in all_memories:
|
||||||
memory_list = all_memories['results']
|
memory_list = all_memories['results']
|
||||||
|
|||||||
@@ -107,16 +107,16 @@ def main():
|
|||||||
|
|
||||||
for query in search_queries:
|
for query in search_queries:
|
||||||
print(f"\nQuery: {query}")
|
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(f" - {memory_item['memory']}")
|
||||||
|
|
||||||
print("\n--> Getting all memories for user...")
|
print("\n--> Getting all memories for user...")
|
||||||
all_memories = memory.get_all(user_id="user_123")
|
all_memories = memory.get_all(filters={"user_id": "user_123"})
|
||||||
print(f"Total memories stored: {len(all_memories)}")
|
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(f" - {memory_item['memory']}")
|
||||||
|
|
||||||
print("\n--> vLLM integration demo completed successfully!")
|
print("\n--> vLLM integration demo completed successfully!")
|
||||||
|
|||||||
Reference in New Issue
Block a user