Update gateway_client.py

gateway_client.py

@@ -6,60 +6,74 @@ import requests
 # Backend server URL - can be set via environment variable
 # For Hugging Face Spaces: Set MEMORY_SERVER_URL in Space settings (Repository secrets)
 # For local development: Set MEMORY_SERVER_URL in your .env file
-EXAMPLE_SERVER_PORT = os.getenv("MEMORY_SERVER_URL")
+MEMMACHINE_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")
+PROMPT = """You are a helpful AI assistant. Use the provided context and profile information to answer the user's question accurately and helpfully.
 
-    resp = requests.post(
-        f"{EXAMPLE_SERVER_PORT}/memory/store-and-search",
-        params={"user_id": user_id, "query": query},
-        timeout=1000,
-    )
-    resp.raise_for_status()
+<CURRENT_DATE>
+{current_date}
+</CURRENT_DATE>
 
-    return resp.text
+Instructions:
+- Use the PROFILE and CONTEXT data provided to answer the user's question
+- Be conversational and helpful in your responses
+- If you don't have enough information to answer completely, say so and suggest what additional information might be helpful
+- If the context contains relevant information, use it to provide a comprehensive answer
+- If no relevant context is available, let the user know and offer to help in other ways
+- Be concise but thorough in your responses
+- Use markdown formatting when appropriate to make your response clear and readable
 
+Data Guidelines:
+- Don't invent information that isn't in the provided context
+- If information is missing or unclear, acknowledge this
+- Prioritize the most recent and relevant information when available
+- If there are conflicting pieces of information, mention this and explain the differences
 
-def add_session_message(user_id: str, msg: str) -> None:
-    """Add a raw message into memory via memory server."""
+Response Format:
+- Start with a direct answer to the user's question
+- Provide supporting details from the context when available
+- Use bullet points or numbered lists when appropriate
+- End with any relevant follow-up questions or suggestions"""
+
+def ingest_and_rewrite(user_id: str, query: str) -> str:
+    """Pass a raw user message through the memory server and get context-aware response."""
+    print("entered ingest_and_rewrite")
+    headers = {
+        "user-id": user_id,
+        "group-id": user_id,  
+        "session-id": user_id,
+        "agent-id": "agent",  
+    }
+    
     requests.post(
-        f"{EXAMPLE_SERVER_PORT}/memory",
-        params={"user_id": user_id, "query": msg},
+        f"{MEMMACHINE_PORT}/v1/memories",
+        json={"producer": user_id, "produced_for": "agent", "episode_content": query},
+        headers=headers,
         timeout=5,
     )
-
-
-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(),
-        },
+    
+    resp = requests.post(
+        f"{MEMMACHINE_PORT}/v1/memories/search",
+        headers=headers,
+        json={"query": query},
         timeout=1000,
     )
     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}"
-
-
-def add_new_session_message(user_id: str, msg: str) -> None:
-    """Alias for add_session_message for backward compatibility."""
-    add_session_message(user_id, msg)
+    return PROMPT + "\n\n" + resp.text + "\n\n" + "User Query: " + query
 
 
 def delete_profile(user_id: str) -> bool:
-    """Delete all memory for the given user_id via the CRM server."""
-    # NOT IMPLEMENTED
-    return False
+    """Delete the session for the given user_id"""
+    headers = {
+        "user-id": user_id,
+        "group-id": user_id,  
+        "session-id": user_id,
+        "agent-id": "agent",  
+    }
+    requests.delete(f"{MEMMACHINE_PORT}/v1/memories", headers=headers, json={})
+    return True
+Shared in
+