Revert gateway client endpoints

gateway_client.py

@@ -12,64 +12,46 @@ 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."""
-    params = {
-        "persona_name": user_id,
-        "query": query,
-        "model_type": model_type,
-        "skip_rewrite": False,
-    }
+    print("entered ingest_and_rewrite")
+
     resp = requests.post(
         f"{EXAMPLE_SERVER_PORT}/memory/store-and-search",
-        params=payload,
-        timeout=30,
+        params={"user_id": user_id, "query": query},
+        timeout=1000,
     )
     resp.raise_for_status()
-    data = resp.json()
-    return data.get("rewritten_query", query)
+
+    return resp.text
 
 
 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=payload,
-        timeout=10,
+        params={"user_id": user_id, "query": msg},
+        timeout=5,
     )
 
 
 def create_persona_query(user_id: str, query: str) -> str:
     """Create a persona-aware query by searching memory context via memory server."""
-    params = {
-        "persona_name": user_id,
-        "query": query,
-        "top_k": 5,
-    }
-    resp = requests.post(
-        f"{EXAMPLE_SERVER_PORT}/memory/search",
-        params=payload,
-        timeout=30,
+    resp = requests.get(
+        f"{EXAMPLE_SERVER_PORT}/memory",
+        params={
+            "query": query,
+            "user_id": user_id,
+            "timestamp": datetime.now().isoformat(),
+        },
+        timeout=1000,
     )
     resp.raise_for_status()
-    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}"
+
+    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}"
 
 
 def add_new_session_message(user_id: str, msg: str) -> None: