3de3c8cb36
- matrix-gateway: POST /internal/matrix/presence/online endpoint - usePresenceHeartbeat hook with activity tracking - Auto away after 5 min inactivity - Offline on page close/visibility change - Integrated in MatrixChatRoom component
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import httpx
|
|
import os
|
|
from typing import List, Dict, Any
|
|
|
|
AGENT_MEMORY_URL = os.getenv("AGENT_MEMORY_URL", "http://agent-memory:7008")
|
|
|
|
async def query_memory(
|
|
agent_id: str,
|
|
microdao_id: str,
|
|
query: str,
|
|
k: int = 5
|
|
) -> List[Dict[str, Any]]:
|
|
"""
|
|
Query agent memory for relevant context
|
|
|
|
Returns list of memory fragments relevant to the query
|
|
Falls back to empty list if Agent Memory service is not available
|
|
"""
|
|
try:
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
response = await client.post(
|
|
f"{AGENT_MEMORY_URL}/internal/agent-memory/query",
|
|
headers={
|
|
"X-Internal-Secret": os.getenv("MEMORY_ORCHESTRATOR_SECRET", "dev-secret-token"),
|
|
"Content-Type": "application/json"
|
|
},
|
|
json={
|
|
"agent_id": agent_id,
|
|
"microdao_id": microdao_id,
|
|
"query": query,
|
|
"limit": k
|
|
}
|
|
)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
results = data.get("results", [])
|
|
print(f"✅ Retrieved {len(results)} memory fragments")
|
|
return results
|
|
except httpx.ConnectError:
|
|
print(f"⚠️ Agent Memory service not available (Phase 2 - OK)")
|
|
return []
|
|
except httpx.HTTPStatusError as e:
|
|
print(f"⚠️ Agent Memory HTTP error: {e.response.status_code}")
|
|
return []
|
|
except Exception as e:
|
|
print(f"⚠️ Agent Memory error: {e}")
|
|
return []
|
|
|
|
async def store_memory(
|
|
agent_id: str,
|
|
microdao_id: str,
|
|
channel_id: str,
|
|
content: Dict[str, Any]
|
|
) -> bool:
|
|
"""
|
|
Store interaction in agent memory
|
|
|
|
Returns True if successful, False otherwise
|
|
Optional in Phase 2 - memory writeback
|
|
"""
|
|
try:
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
response = await client.post(
|
|
f"{AGENT_MEMORY_URL}/internal/agent-memory/store",
|
|
headers={
|
|
"X-Internal-Secret": os.getenv("MEMORY_ORCHESTRATOR_SECRET", "dev-secret-token"),
|
|
"Content-Type": "application/json"
|
|
},
|
|
json={
|
|
"agent_id": agent_id,
|
|
"microdao_id": microdao_id,
|
|
"channel_id": channel_id,
|
|
"kind": "conversation",
|
|
"content": content
|
|
}
|
|
)
|
|
response.raise_for_status()
|
|
print(f"✅ Stored memory for {agent_id}")
|
|
return True
|
|
except httpx.ConnectError:
|
|
print(f"⚠️ Agent Memory service not available (Phase 2 - OK)")
|
|
return False
|
|
except Exception as e:
|
|
print(f"⚠️ Error storing memory: {e}")
|
|
return False
|
|
|