e2ba91cbea
- Add result count to summary - Increase content preview to 250 chars - Add user_id to result display - Improve 'no results' messaging - Better result counting with tee
88 lines
2.7 KiB
Bash
Executable File
88 lines
2.7 KiB
Bash
Executable File
#!/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 "=========================================="
|