xtrm-tools 0.7.3 → 0.7.4

package/.xtrm/skills/default/last30days/scripts/lib/dedupe.py

@@ -0,0 +1,290 @@
+"""Near-duplicate detection for last30days skill."""
+
+import re
+from typing import List, Set, Tuple, Union
+
+from . import schema
+
+# Stopwords for token-based Jaccard (cross-source linking)
+STOPWORDS = frozenset({
+    'the', 'a', 'an', 'to', 'for', 'how', 'is', 'in', 'of', 'on',
+    'and', 'with', 'from', 'by', 'at', 'this', 'that', 'it', 'my',
+    'your', 'i', 'me', 'we', 'you', 'what', 'are', 'do', 'can',
+    'its', 'be', 'or', 'not', 'no', 'so', 'if', 'but', 'about',
+    'all', 'just', 'get', 'has', 'have', 'was', 'will', 'show', 'hn',
+})
+
+
+def normalize_text(text: str) -> str:
+    """Normalize text for comparison.
+
+    - Lowercase
+    - Remove punctuation
+    - Collapse whitespace
+    """
+    text = text.lower()
+    text = re.sub(r'[^\w\s]', ' ', text)
+    text = re.sub(r'\s+', ' ', text)
+    return text.strip()
+
+
+def get_ngrams(text: str, n: int = 3) -> Set[str]:
+    """Get character n-grams from text."""
+    text = normalize_text(text)
+    if len(text) < n:
+        return {text}
+    return {text[i:i+n] for i in range(len(text) - n + 1)}
+
+
+def jaccard_similarity(set1: Set[str], set2: Set[str]) -> float:
+    """Compute Jaccard similarity between two sets."""
+    if not set1 or not set2:
+        return 0.0
+    intersection = len(set1 & set2)
+    union = len(set1 | set2)
+    return intersection / union if union > 0 else 0.0
+
+
+AnyItem = Union[schema.RedditItem, schema.XItem, schema.YouTubeItem, schema.TikTokItem,
+                schema.InstagramItem, schema.HackerNewsItem, schema.PolymarketItem, schema.WebSearchItem]
+
+
+def get_item_text(item: AnyItem) -> str:
+    """Get comparable text from an item."""
+    if isinstance(item, schema.RedditItem):
+        return item.title
+    elif isinstance(item, schema.HackerNewsItem):
+        return item.title
+    elif isinstance(item, schema.YouTubeItem):
+        return f"{item.title} {item.channel_name}"
+    elif isinstance(item, schema.TikTokItem):
+        return f"{item.text} {item.author_name}"
+    elif isinstance(item, schema.InstagramItem):
+        return f"{item.text} {item.author_name}"
+    elif isinstance(item, schema.PolymarketItem):
+        return f"{item.title} {item.question}"
+    elif isinstance(item, schema.WebSearchItem):
+        return item.title
+    else:
+        return item.text
+
+
+def _get_cross_source_text(item: AnyItem) -> str:
+    """Get text for cross-source comparison.
+
+    Same as get_item_text() but truncates X posts to 100 chars
+    to level the playing field against short Reddit/HN titles.
+    Strips 'Show HN:' prefix from HN titles for fairer matching.
+    """
+    if isinstance(item, schema.XItem):
+        return item.text[:100]
+    if isinstance(item, schema.TikTokItem):
+        return item.text[:100]
+    if isinstance(item, schema.InstagramItem):
+        return item.text[:100]
+    if isinstance(item, schema.HackerNewsItem):
+        title = item.title
+        if title.startswith("Show HN:"):
+            title = title[8:].strip()
+        elif title.startswith("Ask HN:"):
+            title = title[7:].strip()
+        return title
+    if isinstance(item, schema.PolymarketItem):
+        return item.title
+    return get_item_text(item)
+
+
+def _tokenize_for_xref(text: str) -> Set[str]:
+    """Tokenize text for cross-source token Jaccard comparison."""
+    words = re.sub(r'[^\w\s]', ' ', text.lower()).split()
+    return {w for w in words if w not in STOPWORDS and len(w) > 1}
+
+
+def _token_jaccard(text_a: str, text_b: str) -> float:
+    """Token-level Jaccard similarity (word overlap)."""
+    tokens_a = _tokenize_for_xref(text_a)
+    tokens_b = _tokenize_for_xref(text_b)
+    if not tokens_a or not tokens_b:
+        return 0.0
+    intersection = len(tokens_a & tokens_b)
+    union = len(tokens_a | tokens_b)
+    return intersection / union if union else 0.0
+
+
+def _hybrid_similarity(text_a: str, text_b: str) -> float:
+    """Hybrid similarity: max of char-trigram Jaccard and token Jaccard."""
+    trigram_sim = jaccard_similarity(get_ngrams(text_a), get_ngrams(text_b))
+    token_sim = _token_jaccard(text_a, text_b)
+    return max(trigram_sim, token_sim)
+
+
+def find_duplicates(
+    items: List[Union[schema.RedditItem, schema.XItem]],
+    threshold: float = 0.7,
+) -> List[Tuple[int, int]]:
+    """Find near-duplicate pairs in items.
+
+    Args:
+        items: List of items to check
+        threshold: Similarity threshold (0-1)
+
+    Returns:
+        List of (i, j) index pairs where i < j and items are similar
+    """
+    duplicates = []
+
+    # Pre-compute n-grams
+    ngrams = [get_ngrams(get_item_text(item)) for item in items]
+
+    for i in range(len(items)):
+        for j in range(i + 1, len(items)):
+            similarity = jaccard_similarity(ngrams[i], ngrams[j])
+            if similarity >= threshold:
+                duplicates.append((i, j))
+
+    return duplicates
+
+
+def dedupe_items(
+    items: List[Union[schema.RedditItem, schema.XItem]],
+    threshold: float = 0.7,
+) -> List[Union[schema.RedditItem, schema.XItem]]:
+    """Remove near-duplicates, keeping highest-scored item.
+
+    Args:
+        items: List of items (should be pre-sorted by score descending)
+        threshold: Similarity threshold
+
+    Returns:
+        Deduplicated items
+    """
+    if len(items) <= 1:
+        return items
+
+    # Find duplicate pairs
+    dup_pairs = find_duplicates(items, threshold)
+
+    # Mark indices to remove (always remove the lower-scored one)
+    # Since items are pre-sorted by score, the second index is always lower
+    to_remove = set()
+    for i, j in dup_pairs:
+        # Keep the higher-scored one (lower index in sorted list)
+        if items[i].score >= items[j].score:
+            to_remove.add(j)
+        else:
+            to_remove.add(i)
+
+    # Return items not marked for removal
+    return [item for idx, item in enumerate(items) if idx not in to_remove]
+
+
+def dedupe_reddit(
+    items: List[schema.RedditItem],
+    threshold: float = 0.7,
+) -> List[schema.RedditItem]:
+    """Dedupe Reddit items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_x(
+    items: List[schema.XItem],
+    threshold: float = 0.7,
+) -> List[schema.XItem]:
+    """Dedupe X items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_youtube(
+    items: List[schema.YouTubeItem],
+    threshold: float = 0.7,
+) -> List[schema.YouTubeItem]:
+    """Dedupe YouTube items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_tiktok(
+    items: List[schema.TikTokItem],
+    threshold: float = 0.7,
+) -> List[schema.TikTokItem]:
+    """Dedupe TikTok items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_instagram(
+    items: List[schema.InstagramItem],
+    threshold: float = 0.7,
+) -> List[schema.InstagramItem]:
+    """Dedupe Instagram items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_hackernews(
+    items: List[schema.HackerNewsItem],
+    threshold: float = 0.7,
+) -> List[schema.HackerNewsItem]:
+    """Dedupe Hacker News items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_bluesky(
+    items: List[schema.BlueskyItem],
+    threshold: float = 0.7,
+) -> List[schema.BlueskyItem]:
+    """Dedupe Bluesky items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_truthsocial(
+    items: List[schema.TruthSocialItem],
+    threshold: float = 0.7,
+) -> List[schema.TruthSocialItem]:
+    """Dedupe Truth Social items."""
+    return dedupe_items(items, threshold)
+
+
+def dedupe_polymarket(
+    items: List[schema.PolymarketItem],
+    threshold: float = 0.7,
+) -> List[schema.PolymarketItem]:
+    """Dedupe Polymarket items."""
+    return dedupe_items(items, threshold)
+
+
+def cross_source_link(
+    *source_lists: List[AnyItem],
+    threshold: float = 0.40,
+) -> None:
+    """Annotate items with cross-source references.
+
+    Compares items across different source types using hybrid similarity
+    (max of char-trigram Jaccard and token Jaccard). When similarity exceeds
+    threshold, adds bidirectional cross_refs with the related item's ID.
+    Modifies items in-place.
+
+    Args:
+        *source_lists: Variable number of per-source item lists
+        threshold: Similarity threshold for cross-linking (default 0.40)
+    """
+    all_items = []
+    for source_list in source_lists:
+        all_items.extend(source_list)
+
+    if len(all_items) <= 1:
+        return
+
+    # Pre-compute cross-source text for each item
+    texts = [_get_cross_source_text(item) for item in all_items]
+
+    for i in range(len(all_items)):
+        for j in range(i + 1, len(all_items)):
+            # Skip same-source comparisons (handled by per-source dedupe)
+            if type(all_items[i]) is type(all_items[j]):
+                continue
+
+            similarity = _hybrid_similarity(texts[i], texts[j])
+            if similarity >= threshold:
+                # Bidirectional cross-reference
+                if all_items[j].id not in all_items[i].cross_refs:
+                    all_items[i].cross_refs.append(all_items[j].id)
+                if all_items[i].id not in all_items[j].cross_refs:
+                    all_items[j].cross_refs.append(all_items[i].id)

package/.xtrm/skills/default/last30days/scripts/lib/entity_extract.py

@@ -0,0 +1,127 @@
+"""Entity extraction from Phase 1 search results for supplemental searches."""
+
+import re
+from collections import Counter
+from typing import Any, Dict, List
+
+# Handles that appear too frequently to be useful for targeted search.
+# These are generic/platform accounts, not topic-specific voices.
+GENERIC_HANDLES = {
+    "elonmusk", "openai", "google", "microsoft", "apple", "meta",
+    "github", "youtube", "x", "twitter", "reddit", "wikipedia",
+    "nytimes", "washingtonpost", "cnn", "bbc", "reuters",
+    "verified", "jack", "sundarpichai",
+}
+
+
+def extract_entities(
+    reddit_items: List[Dict[str, Any]],
+    x_items: List[Dict[str, Any]],
+    max_handles: int = 5,
+    max_hashtags: int = 3,
+    max_subreddits: int = 5,
+) -> Dict[str, List[str]]:
+    """Extract key entities from Phase 1 results for supplemental searches.
+
+    Parses X results for @handles and #hashtags, Reddit results for subreddit
+    names and cross-referenced communities.
+
+    Args:
+        reddit_items: Raw Reddit item dicts from Phase 1
+        x_items: Raw X item dicts from Phase 1
+        max_handles: Maximum handles to return
+        max_hashtags: Maximum hashtags to return
+        max_subreddits: Maximum subreddits to return
+
+    Returns:
+        Dict with keys: x_handles, x_hashtags, reddit_subreddits
+    """
+    handles = _extract_x_handles(x_items)
+    hashtags = _extract_x_hashtags(x_items)
+    subreddits = _extract_subreddits(reddit_items)
+
+    return {
+        "x_handles": handles[:max_handles],
+        "x_hashtags": hashtags[:max_hashtags],
+        "reddit_subreddits": subreddits[:max_subreddits],
+    }
+
+
+def _extract_x_handles(x_items: List[Dict[str, Any]]) -> List[str]:
+    """Extract and rank @handles from X results.
+
+    Sources handles from:
+    1. author_handle field (who posted)
+    2. @mentions in post text (who they're talking about/to)
+
+    Returns handles ranked by frequency, filtered for generic accounts.
+    """
+    handle_counts = Counter()
+
+    for item in x_items:
+        # Author handle
+        author = item.get("author_handle", "").strip().lstrip("@").lower()
+        if author and author not in GENERIC_HANDLES:
+            handle_counts[author] += 1
+
+        # @mentions in text
+        text = item.get("text", "")
+        mentions = re.findall(r'@(\w{1,15})', text)
+        for mention in mentions:
+            mention_lower = mention.lower()
+            if mention_lower not in GENERIC_HANDLES:
+                handle_counts[mention_lower] += 1
+
+    # Return all handles ranked by frequency
+    return [h for h, _ in handle_counts.most_common()]
+
+
+def _extract_x_hashtags(x_items: List[Dict[str, Any]]) -> List[str]:
+    """Extract and rank #hashtags from X results.
+
+    Returns hashtags ranked by frequency.
+    """
+    hashtag_counts = Counter()
+
+    for item in x_items:
+        text = item.get("text", "")
+        tags = re.findall(r'#(\w{2,30})', text)
+        for tag in tags:
+            hashtag_counts[tag.lower()] += 1
+
+    # Return all hashtags ranked by frequency
+    return [f"#{t}" for t, _ in hashtag_counts.most_common()]
+
+
+def _extract_subreddits(reddit_items: List[Dict[str, Any]]) -> List[str]:
+    """Extract and rank subreddits from Reddit results.
+
+    Sources from:
+    1. subreddit field on each result
+    2. Cross-references in comment text (e.g., "check out r/localLLaMA")
+
+    Returns subreddits ranked by frequency.
+    """
+    sub_counts = Counter()
+
+    for item in reddit_items:
+        # Primary subreddit
+        sub = item.get("subreddit", "").strip().lstrip("r/")
+        if sub:
+            sub_counts[sub] += 1
+
+        # Cross-references in comment insights
+        for insight in item.get("comment_insights", []):
+            cross_refs = re.findall(r'r/(\w{2,30})', insight)
+            for ref in cross_refs:
+                sub_counts[ref] += 1
+
+        # Cross-references in top comments
+        for comment in item.get("top_comments", []):
+            excerpt = comment.get("excerpt", "")
+            cross_refs = re.findall(r'r/(\w{2,30})', excerpt)
+            for ref in cross_refs:
+                sub_counts[ref] += 1
+
+    # Return subreddits ranked by frequency
+    return [sub for sub, _ in sub_counts.most_common()]