Files
true-recall-base/scripts/search_q.sh

88 lines
2.7 KiB
Bash
Raw Normal View History

#!/bin/bash
# search_q.sh - Search memories with chronological sorting
# Usage: ./search_q.sh "search query"
# Returns: Results sorted by timestamp (newest first)
set -e
QDRANT_URL="${QDRANT_URL:-http://localhost:6333}"
COLLECTION="${QDRANT_COLLECTION:-memories_tr}"
LIMIT="${SEARCH_LIMIT:-10}"
if [ -z "$1" ]; then
echo "Usage: ./search_q.sh 'your search query'"
echo ""
echo "Environment variables:"
echo " QDRANT_URL - Qdrant endpoint (default: http://localhost:6333)"
echo " SEARCH_LIMIT - Number of results (default: 10)"
exit 1
fi
QUERY="$1"
echo "=========================================="
echo "Searching: '$QUERY'"
echo "=========================================="
echo ""
# Search with scroll to get all results, then sort by timestamp
# Using scroll API to handle large result sets
SCROLL_ID="null"
ALL_RESULTS="[]"
while true; do
if [ "$SCROLL_ID" = "null" ]; then
RESPONSE=$(curl -s -X POST "$QDRANT_URL/collections/$COLLECTION/points/scroll" \
-H "Content-Type: application/json" \
-d "{
\"limit\": $LIMIT,
\"with_payload\": true,
\"filter\": {
\"must\": [
{
\"key\": \"content\",
\"match\": {
\"text\": \"$QUERY\"
}
}
]
}
}") 2>/dev/null || echo '{"result": {"points": []}}'
else
break # For text search, we get results in first call
fi
# Extract results
POINTS=$(echo "$RESPONSE" | jq -r '.result.points // []')
if [ "$POINTS" = "[]" ] || [ "$POINTS" = "null" ]; then
break
fi
ALL_RESULTS="$POINTS"
break
done
# Sort by timestamp (newest first) and format output
echo "$ALL_RESULTS" | jq -r '
sort_by(.payload.timestamp) | reverse |
.[] |
"━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n" +
"📅 " + (.payload.timestamp | split("T") | join(" ")) + "\n" +
"👤 " + .payload.role + " | User: " + .payload.user_id + "\n" +
"📝 " + (.payload.content | if length > 250 then .[0:250] + "..." else . end) + "\n"
' 2>/dev/null | tee /tmp/search_results.txt
# Count results
RESULT_COUNT=$(cat /tmp/search_results.txt | grep -c "━━━━━━━━" 2>/dev/null || echo "0")
echo ""
echo "=========================================="
if [ "$RESULT_COUNT" -gt 0 ]; then
echo "Found $RESULT_COUNT result(s). Most recent shown first."
else
echo "No results found for '$QUERY'"
fi
echo "=========================================="