drjones/alpaca-llm-bot-v1
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Maximize Qdrant utilization: richer memory payloads, filtered retrieval, and qdrant health in metrics API

Browse Source
This commit is contained in:
drjones
2026-03-01 10:30:56 -08:00
parent fee413a852
commit cd8f8430f9
3 changed files with 50 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
services.py
View File

@@ -34,8 +34,15 @@ def qdrant_add_memory(symbol: str, text: str, payload: dict):
if not vec:
return False
qdrant_ensure_collection(len(vec))
enriched = {
"symbol": symbol,
"text": text[:2000],
"memory_type": payload.get("memory_type", "decision"),
"created_at": int(time.time()),
**payload,
}
body = {
"points": [{"id": str(uuid.uuid4()), "vector": vec, "payload": {"symbol": symbol, "text": text[:2000], **payload}}]
"points": [{"id": str(uuid.uuid4()), "vector": vec, "payload": enriched}]
}
try:
r = requests.put(f"{settings.qdrant_url}/collections/{settings.qdrant_collection}/points", json=body, timeout=15)
@@ -44,11 +51,21 @@ def qdrant_add_memory(symbol: str, text: str, payload: dict):
return False
def qdrant_similar(symbol: str, query_text: str, limit: int = 5):
def qdrant_similar(symbol: str, query_text: str, limit: int = 8):
vec = _embed(query_text)
if not vec:
return []
body = {"vector": vec, "limit": limit, "with_payload": True}
body = {
"vector": vec,
"limit": limit,
"with_payload": True,
"filter": {
"should": [
{"key": "symbol", "match": {"value": symbol}},
{"key": "memory_type", "match": {"value": "macro"}}
]
}
}
try:
r = requests.post(f"{settings.qdrant_url}/collections/{settings.qdrant_collection}/points/search", json=body, timeout=15)
if not r.ok:
@@ -56,8 +73,14 @@ def qdrant_similar(symbol: str, query_text: str, limit: int = 5):
out = []
for p in r.json().get("result", []):
pl = p.get("payload", {})
if pl.get("symbol") in (symbol, None):
out.append({"score": p.get("score", 0), "text": pl.get("text", ""), "status": pl.get("status", "")})
out.append({
"score": p.get("score", 0),
"text": pl.get("text", ""),
"status": pl.get("status", ""),
"action": pl.get("action", ""),
"confidence": pl.get("confidence", None),
"outcome_score": pl.get("outcome_score", None),
})
return out
except Exception:
return []
@@ -233,6 +256,22 @@ def trilium_log(title: str, body: str):
return False
def qdrant_memory_health():
try:
r = requests.get(f"{settings.qdrant_url}/collections/{settings.qdrant_collection}", timeout=10)
if not r.ok:
return {"ok": False, "status": r.status_code}
j = r.json().get("result", {})
return {
"ok": True,
"collection": settings.qdrant_collection,
"points_count": j.get("points_count"),
"indexed_vectors_count": j.get("indexed_vectors_count"),
}
except Exception as e:
return {"ok": False, "error": str(e)[:160]}
def n8n_emit(event: dict):
if not settings.n8n_webhook:
return False