Sync all code from test-playground - complete codebase match

app.py

@@ -7,7 +7,7 @@ from urllib.parse import urlencode
 import requests
 import streamlit as st
 
-from gateway_client import delete_profile, ingest_and_rewrite, get_memories, ingest_memories
+from gateway_client import delete_profile, ingest_and_rewrite, ingest_memories
 from llm import chat, set_model
 from model_config import MODEL_CHOICES, MODEL_TO_PROVIDER, MODEL_DISPLAY_NAMES
 
@@ -896,4 +896,4 @@ for turn in st.session_state.history:
                     )
         # Mark as no longer new
         if is_new:
-            turn["is_new"] = False
+            turn["is_new"] = False

gateway_client.py

@@ -64,6 +64,72 @@ def ingest_and_rewrite(user_id: str, query: str) -> str:
     return PROMPT + "\n\n" + resp.text + "\n\n" + "User Query: " + query
 
 
+def get_memories(user_id: str) -> dict:
+    """Fetch all memories for a given user_id"""
+    headers = {
+        "user-id": user_id,
+        "group-id": user_id,  
+        "session-id": user_id,
+        "agent-id": "agent",
+    }
+    
+    # Add API key if configured
+    if BACKEND_API_KEY:
+        headers["x-api-key"] = BACKEND_API_KEY
+    
+    try:
+        resp = requests.get(
+            f"{MEMMACHINE_PORT}/v1/memories",
+            headers=headers,
+            timeout=10,
+        )
+        resp.raise_for_status()
+        return resp.json()
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching memories: {e}")
+        return {}
+
+
+def ingest_memories(user_id: str, memories_text: str) -> bool:
+    """Ingest imported memories into MemMachine system.
+    
+    Args:
+        user_id: The user identifier
+        memories_text: Text containing memories/conversations to ingest
+        
+    Returns:
+        True if successful, False otherwise
+    """
+    headers = {
+        "user-id": user_id,
+        "group-id": user_id,  
+        "session-id": user_id,
+        "agent-id": "agent",
+    }
+    
+    # Add API key if configured
+    if BACKEND_API_KEY:
+        headers["x-api-key"] = BACKEND_API_KEY
+    
+    try:
+        # Ingest the memories as an episode
+        resp = requests.post(
+            f"{MEMMACHINE_PORT}/v1/memories",
+            json={
+                "producer": user_id,
+                "produced_for": "agent",
+                "episode_content": memories_text
+            },
+            headers=headers,
+            timeout=10,
+        )
+        resp.raise_for_status()
+        return True
+    except requests.exceptions.RequestException as e:
+        print(f"Error ingesting memories: {e}")
+        return False
+
+
 def delete_profile(user_id: str) -> bool:
     """Delete the session for the given user_id"""
     headers = {

styles.css

@@ -68,4 +68,4 @@ div[data-testid="column"] {
   
   div[data-testid="stSuccess"] > div {
     color: #667eea !important;
-  }
+  }