Restore gateway client v1 endpoints

gateway_client.py

@@ -12,46 +12,64 @@ EXAMPLE_SERVER_PORT = os.getenv("MEMORY_SERVER_URL")
 
 def ingest_and_rewrite(user_id: str, query: str, model_type: str = "openai") -> str:
     """Pass a raw user message through the memory server and get context-aware response."""
-    print("entered ingest_and_rewrite")
-
+    params = {
+        "persona_name": user_id,
+        "query": query,
+        "model_type": model_type,
+        "skip_rewrite": False,
+    }
     resp = requests.post(
         f"{EXAMPLE_SERVER_PORT}/memory/store-and-search",
-        params={"user_id": user_id, "query": query},
-        timeout=1000,
+        params=payload,
+        timeout=30,
     )
     resp.raise_for_status()
-
-    return resp.text
+    data = resp.json()
+    return data.get("rewritten_query", query)
 
 
 def add_session_message(user_id: str, msg: str) -> None:
     """Add a raw message into memory via memory server."""
+    params = {
+        "persona_name": user_id,
+        "message": msg,
+        "source": "frontend",
+    }
     requests.post(
         f"{EXAMPLE_SERVER_PORT}/memory",
-        params={"user_id": user_id, "query": msg},
-        timeout=5,
+        params=payload,
+        timeout=10,
     )
 
 
 def create_persona_query(user_id: str, query: str) -> str:
     """Create a persona-aware query by searching memory context via memory server."""
-    resp = requests.get(
-        f"{EXAMPLE_SERVER_PORT}/memory",
-        params={
-            "query": query,
-            "user_id": user_id,
-            "timestamp": datetime.now().isoformat(),
-        },
-        timeout=1000,
+    params = {
+        "persona_name": user_id,
+        "query": query,
+        "top_k": 5,
+    }
+    resp = requests.post(
+        f"{EXAMPLE_SERVER_PORT}/memory/search",
+        params=payload,
+        timeout=30,
     )
     resp.raise_for_status()
-
-    search_results = resp.json()
-
-    if search_results.get("profile"):
-        return f"Based on your profile: {search_results['profile']}\n\nQuery: {query}"
-    else:
-        return f"Query: {query}"
+    search_results = resp.json() or {}
+    profile = search_results.get("profile")
+    memories = search_results.get("memories", [])
+
+    summary = ""
+    if profile:
+        summary += f"Profile: {profile}\n"
+    if memories:
+        summary += "Memories:\n"
+        for memory in memories:
+            summary += f"- {memory.get('content', '')}\n"
+
+    if summary:
+        return f"{summary}\nQuery: {query}"
+    return f"Query: {query}"
 
 
 def add_new_session_message(user_id: str, msg: str) -> None: