entari-plugin-hyw 4.0.0rc17__py3-none-any.whl → 4.0.0rc19__py3-none-any.whl

entari_plugin_hyw/search_cache.py

@@ -1,253 +0,0 @@
-"""
-Search Result Cache
-
-Caches search results in memory for 10 minutes to support
-deep query operations on search results.
-"""
-
-import time
-import base64
-import io
-from dataclasses import dataclass, field
-from typing import Dict, List, Any, Optional
-
-from loguru import logger
-
-
-@dataclass
-class CacheEntry:
-    """A cached search result entry."""
-    results: List[Dict[str, Any]]
-    query: str
-    timestamp: float = field(default_factory=time.time)
-
-
-@dataclass
-class ScreenshotCacheEntry:
-    """A cached full screenshot."""
-    screenshot_b64: str
-    url: str
-    timestamp: float = field(default_factory=time.time)
-
-
-class SearchResultCache:
-    """
-    In-memory cache for search results with TTL-based expiration.
-
-    Cleanup is lazy - performed at the end of each request.
-    """
-
-    def __init__(self, ttl_seconds: float = 600.0):  # 10 minutes default
-        self._cache: Dict[str, CacheEntry] = {}
-        self._screenshot_cache: Dict[str, ScreenshotCacheEntry] = {}  # screenshot_id -> full screenshot
-        self._screenshot_counter: int = 0
-        self.ttl_seconds = ttl_seconds
-
-    def store(self, message_id: str, results: List[Dict[str, Any]], query: str):
-        """
-        Store search results associated with a message ID.
-
-        Args:
-            message_id: The sent message ID that contains the search results image
-            results: List of search result dicts with url, title, content, etc.
-            query: The original search query
-        """
-        self._cache[message_id] = CacheEntry(
-            results=results,
-            query=query,
-            timestamp=time.time()
-        )
-
-    def get(self, message_id: str) -> Optional[CacheEntry]:
-        """
-        Get cached search results for a message ID.
-
-        Returns None if not found or expired.
-        """
-        entry = self._cache.get(message_id)
-        if entry is None:
-            return None
-
-        # Check expiration
-        if time.time() - entry.timestamp > self.ttl_seconds:
-            del self._cache[message_id]
-            return None
-
-        return entry
-
-    def store_screenshot(self, screenshot_b64: str, url: str) -> str:
-        """
-        Store a full screenshot and return its cache ID.
-
-        Args:
-            screenshot_b64: Base64 encoded full screenshot
-            url: The URL that was screenshotted
-
-        Returns:
-            A short cache ID for referencing this screenshot
-        """
-        self._screenshot_counter += 1
-        cache_id = f"ss{self._screenshot_counter:04x}"
-        self._screenshot_cache[cache_id] = ScreenshotCacheEntry(
-            screenshot_b64=screenshot_b64,
-            url=url,
-            timestamp=time.time()
-        )
-        return cache_id
-
-    def get_screenshot(self, cache_id: str) -> Optional[str]:
-        """
-        Get a cached full screenshot by ID.
-
-        Returns:
-            Base64 encoded screenshot or None if not found/expired
-        """
-        entry = self._screenshot_cache.get(cache_id)
-        if entry is None:
-            return None
-
-        if time.time() - entry.timestamp > self.ttl_seconds:
-            del self._screenshot_cache[cache_id]
-            return None
-
-        return entry.screenshot_b64
-
-    def cleanup(self):
-        """
-        Remove all expired entries.
-
-        Called lazily at the end of each request.
-        """
-        now = time.time()
-        expired_keys = [
-            k for k, v in self._cache.items()
-            if now - v.timestamp > self.ttl_seconds
-        ]
-        for k in expired_keys:
-            del self._cache[k]
-
-        # Also cleanup screenshot cache
-        expired_ss = [
-            k for k, v in self._screenshot_cache.items()
-            if now - v.timestamp > self.ttl_seconds
-        ]
-        for k in expired_ss:
-            del self._screenshot_cache[k]
-
-    def __len__(self) -> int:
-        return len(self._cache)
-
-
-def crop_to_square_thumbnail(screenshot_b64: str, max_size: int = 400) -> Optional[str]:
-    """
-    Crop a screenshot to a 1:1 square from the top and resize.
-
-    Args:
-        screenshot_b64: Base64 encoded image
-        max_size: Maximum dimension of the output square
-
-    Returns:
-        Base64 encoded cropped/resized image, or None on error
-    """
-    try:
-        from PIL import Image
-
-        # Decode base64
-        img_data = base64.b64decode(screenshot_b64)
-        img = Image.open(io.BytesIO(img_data))
-
-        width, height = img.size
-
-        # Crop to square from top
-        square_size = min(width, height)
-        # Crop from top-left, taking full width if width < height
-        crop_box = (0, 0, square_size, square_size)
-        cropped = img.crop(crop_box)
-
-        # Resize if larger than max_size
-        if square_size > max_size:
-            cropped = cropped.resize((max_size, max_size), Image.Resampling.LANCZOS)
-
-        # Encode back to base64
-        buffer = io.BytesIO()
-        cropped.save(buffer, format='JPEG', quality=85)
-        return base64.b64encode(buffer.getvalue()).decode('utf-8')
-
-    except Exception as e:
-        logger.warning(f"Failed to crop screenshot: {e}")
-        return None
-
-
-def parse_single_index(text: str) -> Optional[int]:
-    """
-    Parse a single index from text like "1" or "3".
-    
-    Args:
-        text: The text to parse
-        
-    Returns:
-        0-based index or None if not a valid single index
-    """
-    if not text:
-        return None
-    text = text.strip()
-    if text.isdigit():
-        idx = int(text)
-        if 1 <= idx <= 10:  # 1-based, max 10 results
-            return idx - 1  # Convert to 0-based
-    return None
-
-
-def parse_multi_indices(text: str, max_count: int = 3) -> Optional[List[int]]:
-    """
-    Parse multiple indices from text like "1-2", "1,2,3", "1、2、5".
-    
-    Supports:
-        - Range: "1-2", "2-4"
-        - Comma separated: "1,2,3", "1, 2, 3"
-        - Chinese comma: "1、2、5"
-        - Space separated: "1 2 3"
-        
-    Args:
-        text: The text to parse
-        max_count: Maximum number of indices allowed (default 3), returns None if exceeded
-        
-    Returns:
-        List of 0-based indices, or None if empty/invalid/exceeds max_count
-    """
-    import re
-    
-    if not text:
-        return None
-    
-    text = text.strip()
-    if not text:
-        return None
-    
-    indices = set()
-    
-    # Check for range pattern: "1-3"
-    range_match = re.match(r'^(\d+)\s*[-–]\s*(\d+)$', text)
-    if range_match:
-        start, end = int(range_match.group(1)), int(range_match.group(2))
-        if 1 <= start <= 10 and 1 <= end <= 10 and start <= end:
-            indices.update(range(start - 1, end))  # 0-based
-            if len(indices) > max_count:
-                return None  # Exceeds max
-            return sorted(indices)
-        return None
-    
-    # Split by comma, Chinese comma, or space
-    parts = re.split(r'[,、\s]+', text)
-    for part in parts:
-        part = part.strip()
-        if part.isdigit():
-            idx = int(part)
-            if 1 <= idx <= 10:
-                indices.add(idx - 1)  # 0-based
-    
-    if indices:
-        if len(indices) > max_count:
-            return None  # Exceeds max
-        return sorted(indices)
-    return None

entari_plugin_hyw-4.0.0rc17.dist-info/METADATA

@@ -1,119 +0,0 @@
-Metadata-Version: 2.4
-Name: entari_plugin_hyw
-Version: 4.0.0rc17
-Summary: Use large language models to interpret chat messages
-Author-email: kumoSleeping <zjr2992@outlook.com>
-License: MIT
-Project-URL: Homepage, https://github.com/kumoSleeping/entari-plugin-hyw
-Project-URL: Repository, https://github.com/kumoSleeping/entari-plugin-hyw
-Project-URL: Issue Tracker, https://github.com/kumoSleeping/entari-plugin-hyw/issues
-Keywords: entari,llm,ai,bot,chat
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: <3.13,>=3.10
-Description-Content-Type: text/markdown
-Requires-Dist: arclet-entari[full]>=0.16.5
-Requires-Dist: openai
-Requires-Dist: httpx
-Requires-Dist: markdown>=3.10
-Requires-Dist: DrissionPage>=4.1.1.2
-Requires-Dist: trafilatura>=1.6.0
-Requires-Dist: json-repair>=0.55.0
-Requires-Dist: Pillow>=10.0.0
-Provides-Extra: dev
-Requires-Dist: entari-plugin-server>=0.5.0; extra == "dev"
-Requires-Dist: satori-python-adapter-onebot11>=0.2.5; extra == "dev"
-Requires-Dist: noneprompt; extra == "dev"
-Requires-Dist: rich; extra == "dev"
-
-# Entari Plugin HYW
-
-[![PyPI version](https://badge.fury.io/py/entari-plugin-hyw.svg)](https://badge.fury.io/py/entari-plugin-hyw)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![Python Versions](https://img.shields.io/pypi/pyversions/entari-plugin-hyw.svg)](https://pypi.org/project/entari-plugin-hyw/)
-
-**English** | [简体中文](docs/README_CN.md)
-
-**Entari Plugin HYW** is an advanced agentic chat plugin for the [Entari](https://github.com/entari-org/entari) framework. It leverages Large Language Models (LLMs) to provide intelligent, context-aware, and multi-modal responses within instant messaging environments (OneBot 11, Satori).
-
-The plugin implements a three-stage pipeline (**Vision**, **Instruct**, **Agent**) to autonomously decide when to search the web, crawl pages, or analyze images to answer user queries effectively.
-
-<p align="center">
-  <img src="docs/demo_mockup.svg" width="800" />
-</p>
-
-## Features
-
-- 📖 **Agentic Workflow**  
-  Autonomous decision-making process to search, browse, and reason.
-
-- 🎑 **Multi-Modal Support**  
-  Native support for image analysis using Vision Language Models (VLMs).
-
-- 🔍 **Web Search & Crawling**  
-  Integrated **DuckDuckGo** and **Crawl4AI** for real-time information retrieval.
-
-- 🎨 **Rich Rendering**  
-  Responses are rendered as images containing Markdown, syntax-highlighted code, LaTeX math, and citation badges.
-
-- 🔌 **Protocol Support**  
-  Deep integration with OneBot 11 and Satori protocols, handling reply context and JSON cards perfectly.
-
-## Installation
-
-```bash
-pip install entari-plugin-hyw
-```
-
-## Configuration
-
-Configure the plugin in your `entari.yml`.
-
-### Minimal Configuration
-
-```yaml
-plugins:
-  entari_plugin_hyw:
-    model_name: google/gemini-2.0-flash-exp
-    api_key: "your-or-api-key-here"
-    # Rendering Configuration
-    render_timeout_ms: 6000 # Browser wait timeout
-    render_image_timeout_ms: 3000 # Image load wait timeout
-```
-
-## Usage
-
-### Commands
-
-- **Text Query**
-  ```text
-  /q What's the latest news on Rust 1.83?
-  ```
-
-- **Image Analysis**
-  *(Send an image with command, or reply to an image)*
-  ```text
-  /q [Image] Explain this error.
-  ```
-- **Quote Query**
-  ```text
-  [quote: User Message] /q
-  ```
-
-- **Follow-up**
-  *Reply to the bot's message to continue the conversation.*
-
-## Documentation for AI/LLMs
-
-- [Instruction Guide (English)](docs/README_LLM_EN.md)
-- [指导手册 (简体中文)](docs/README_LLM_CN.md)
-
----
-
-## License
-
-This project is licensed under the MIT License.

entari_plugin_hyw-4.0.0rc17.dist-info/RECORD

@@ -1,52 +0,0 @@
-entari_plugin_hyw/__init__.py,sha256=unglikkK8GZ6wzJgmVZ5mMy4x9N1RegZWK9C2PSeNUI,35351
-entari_plugin_hyw/filters.py,sha256=sQnLaiqqZ2NkykcH4QgzFImP-JW3uVU7l6iuAAyUsJg,3080
-entari_plugin_hyw/history.py,sha256=0XJwbfvXH5T1EPt4G1J5wWMJsKi0FfmajY5cvw8CQWE,12065
-entari_plugin_hyw/misc.py,sha256=5IqF5Z2C_6Ufy5TI89uX5hX5fVYcXOTZIQUIu_tvf54,6855
-entari_plugin_hyw/search_cache.py,sha256=INRb3NERcPK812tNXn91yg3phRczSNiHU12B77HRzOw,7219
-hyw_core/__init__.py,sha256=Jlr9Ic-BLOPTnff6OctUCdjDMdK4nssTF_vHie4QKTo,1958
-hyw_core/agent.py,sha256=luKJR0MNrNMn3lzZ-3VuFFOr9OdIEPDJ3IA_XvUIYtQ,37843
-hyw_core/config.py,sha256=DHxwToUVLm1nT88gG05e3hVzSLxXMk9BjgjAnhGCADk,4918
-hyw_core/core.py,sha256=HaAAY2fpreW6p9anEB3LrAnUBtxQTavozBJPEAUrvwI,15684
-hyw_core/definitions.py,sha256=KgWgS7trErTqoxVjgtlLoJr0_e2HbY3XUGnY_MQ_70Q,7751
-hyw_core/image_cache.py,sha256=t8pr1kgH2ngK9IhrBAhzUqhBWERNztUywMzgCFZEtQk,9899
-hyw_core/pipeline.py,sha256=ZWwF0DHa29-65lUMU1_Fem3xQmxl7X_vgeni0ErOb8Q,22826
-hyw_core/search.py,sha256=cLx1rPpy8BH6WiOKudH_XcO-DhqEU-CHANRDlyEB3MY,7612
-hyw_core/browser_control/__init__.py,sha256=IeMErRC6fbq1PJWNK3klSbarSrUwOM4yyd_kJ6uWCPM,1406
-hyw_core/browser_control/landing.html,sha256=wgqldumdylz69T83pvOkrigT1Mdb9GY0_KU0ceLGwdY,4642
-hyw_core/browser_control/manager.py,sha256=-dHb0FamRsLfuU3jqX5cKaDo8DOOFV32zY912GuMdXU,6048
-hyw_core/browser_control/renderer.py,sha256=s-QNIU-NMVQGLd_drLmeERgHsTm6C9XYm78CObt2KXc,17409
-hyw_core/browser_control/service.py,sha256=mo2qkFJgi5DkzURVD8QrIj3xdd_jKwiJSoTVEfFAsY0,46626
-hyw_core/browser_control/assets/card-dist/index.html,sha256=MMMovCSWL8Q3xqGYNoMNoh6Rz_HITT8Q_vmv00suPE0,2213337
-hyw_core/browser_control/assets/card-dist/vite.svg,sha256=SnSK_UQ5GLsWWRyDTEAdrjPoeGGrXbrQgRw6O0qSFPs,1497
-hyw_core/browser_control/assets/card-dist/logos/anthropic.svg,sha256=ASsy1ypo3osNc3n-B0R81tk_dIFsVgg7qQORrd5T2kA,558
-hyw_core/browser_control/assets/card-dist/logos/cerebras.svg,sha256=bpmiiYTODwc06knTmPj3GQ7NNtosMog5lkggvB_Z-7M,44166
-hyw_core/browser_control/assets/card-dist/logos/deepseek.png,sha256=KWWAr9aeYMc6I07U_1qo7zcXO6e7-kfd9S2XjQumnf4,25338
-hyw_core/browser_control/assets/card-dist/logos/gemini.svg,sha256=H74CoVmx5opcCtr3Ay3M09dpqL9cd9Whkx-M6an3t7s,599
-hyw_core/browser_control/assets/card-dist/logos/google.svg,sha256=H74CoVmx5opcCtr3Ay3M09dpqL9cd9Whkx-M6an3t7s,599
-hyw_core/browser_control/assets/card-dist/logos/grok.png,sha256=uSulvvDVqoA4RUOW0ZAkdvBVM2rpyGJRZIbn5dEFspw,362
-hyw_core/browser_control/assets/card-dist/logos/huggingface.png,sha256=8eAudeftUDO11jf0coOscPeRkskCb7l9TNMx78q61mY,24564
-hyw_core/browser_control/assets/card-dist/logos/microsoft.svg,sha256=-am_6N3UEQYSzldDg-xrdGYjTWsagH-3v4Q_eia1ymE,684
-hyw_core/browser_control/assets/card-dist/logos/minimax.png,sha256=tWqVlMdFNPpP8zWWX9tvIsWXI9q76P7O3t3CEZO7NU0,1525
-hyw_core/browser_control/assets/card-dist/logos/mistral.png,sha256=0vv7jPmPKiBRYVYYJxVL_wIH_qa_ZssIdV3NDO5vbmk,869
-hyw_core/browser_control/assets/card-dist/logos/nvida.png,sha256=JMITdcyjR9Lz6Gub0n1_30d0ynvV1ZSCJRcjy23qgrA,1607
-hyw_core/browser_control/assets/card-dist/logos/openai.svg,sha256=LhVwCR4qaXj6qHm31qniQTCkJ-FX932JLSycUis5kao,1692
-hyw_core/browser_control/assets/card-dist/logos/openrouter.png,sha256=exxfjWGDWpYH-Vc8xJDbhNVeXFEVxnu6TMxYIBc1WmY,1665
-hyw_core/browser_control/assets/card-dist/logos/perplexity.svg,sha256=mHWZFoeWmDYXOIDzm9pj6_sRotaI8xNy5Lkeg5Vzu70,555
-hyw_core/browser_control/assets/card-dist/logos/qwen.png,sha256=eqLbnIPbjh2_PsODU_mmqjeD82xXj8fV_kN0fDrNaD0,38419
-hyw_core/browser_control/assets/card-dist/logos/xai.png,sha256=uSulvvDVqoA4RUOW0ZAkdvBVM2rpyGJRZIbn5dEFspw,362
-hyw_core/browser_control/assets/card-dist/logos/xiaomi.png,sha256=WHxlDFGU5FCjb-ure3ngdGG18-efYZUUfqA3_lqCUN0,4084
-hyw_core/browser_control/assets/card-dist/logos/zai.png,sha256=K-gnabdsjMLInppHA1Op7Nyt33iegrx1x-yNlvCZ0Tc,2351
-hyw_core/browser_control/engines/__init__.py,sha256=01-jOjvtQcqWIwwY56ql3j00oSHGE2XhDHjkIi1Ij3Q,284
-hyw_core/browser_control/engines/base.py,sha256=q5y4SM1G6xS7-6TQ-nZz9iTWw3XonjJn01fWzoTxr6c,414
-hyw_core/browser_control/engines/default.py,sha256=BlHCQI4-rN9cEzLLfqvRD4bvhyP2G2KUGlo92J4kFNw,6092
-hyw_core/browser_control/engines/duckduckgo.py,sha256=QVYj1I6Fw6--2-f3DYdHVlOmeLPzet9qG4HUT7ynrSE,6789
-hyw_core/crawling/__init__.py,sha256=W5Be02uQueAyADFhPXVbDAFpNUJ_W7KdMtMr923_N24,437
-hyw_core/crawling/completeness.py,sha256=OKdS8XlYYWDU1Vl1k-u7yEFqppukuJv-YQB0Px5KC2Q,16371
-hyw_core/crawling/models.py,sha256=pCKe0k9xT3taSAlTlh0PazcLV0xYsm8p3XIkLHGf-LM,2353
-hyw_core/stages/__init__.py,sha256=W89cWpq-HBLi2FprtJQSjQNLzpbhM8ZCkqPG61D_imE,521
-hyw_core/stages/base.py,sha256=EfnTkISXbBNxjARykqIhmMrVqw2tqZl7ozJbJEbRnhI,2806
-hyw_core/stages/summary.py,sha256=D0XwhqmtoyovSXUWoa-_RxeoKWEkdjqMyoJV-gWQwvQ,8342
-entari_plugin_hyw-4.0.0rc17.dist-info/METADATA,sha256=yJQI0rRb92QgqLRBnqS1VZ5zcn6j538KweZtiJh6nV4,3875
-entari_plugin_hyw-4.0.0rc17.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-entari_plugin_hyw-4.0.0rc17.dist-info/top_level.txt,sha256=ah76OrufRX0okOl4Fv8MO6PXiT0IaZ1oG0eDrdAPoNo,27
-entari_plugin_hyw-4.0.0rc17.dist-info/RECORD,,

entari_plugin_hyw-4.0.0rc17.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-entari_plugin_hyw
-hyw_core

hyw_core/__init__.py

@@ -1,94 +0,0 @@
-"""
-hyw-core - Core LLM Pipeline and Browser Automation
-
-This package provides:
-- HywCore: Main service class with unified query interface
-- HywCoreConfig: Configuration management with standalone/passthrough support
-- QueryRequest/QueryResponse: Request/response data classes
-- SearchService: Web search abstraction
-- ModularPipeline: LLM pipeline orchestration
-- browser_control: Browser automation subpackage
-
-Usage:
-    from hyw_core import HywCore, HywCoreConfig, QueryRequest
-    
-    # Standalone usage with YAML config
-    config = HywCoreConfig.from_yaml("config.yaml")
-    core = HywCore(config)
-    
-    response = await core.query(QueryRequest(
-        user_input="What is Python?",
-        images=[],
-        conversation_history=[]
-    ))
-    
-    # Passthrough from parent package
-    config = HywCoreConfig.from_dict({
-        "model_name": parent_config.model_name,
-        "api_key": parent_config.api_key,
-        ...
-    })
-"""
-
-__version__ = "4.0.0-rc8"
-
-# Core classes
-from .core import HywCore, QueryRequest, QueryResponse
-
-# Configuration
-from .config import HywCoreConfig, ModelConfig
-
-# Pipeline components
-from .pipeline import ModularPipeline
-from .search import SearchService
-
-# Stage components
-from .stages import (
-    BaseStage,
-    StageContext,
-    StageResult,
-
-    SummaryStage,
-)
-
-# Definitions
-from .definitions import (
-    SUMMARY_REPORT_SP,
-    get_refuse_answer_tool,
-)
-
-# Browser control is available as subpackage
-from . import browser_control
-
-__all__ = [
-    # Version
-    "__version__",
-    
-    # Core
-    "HywCore",
-    "QueryRequest", 
-    "QueryResponse",
-    
-    # Configuration
-    "HywCoreConfig",
-    "ModelConfig",
-    
-    # Pipeline
-    "ModularPipeline",
-    "SearchService",
-    
-    # Stages
-    "BaseStage",
-    "StageContext",
-    "StageResult",
-
-    "SummaryStage",
-    
-    # Definitions
-    "SUMMARY_REPORT_SP",
-    "get_refuse_answer_tool",
-
-    
-    # Subpackage
-    "browser_control",
-]