cryptorugmunch
8d25702eca
Squashed from chore/license-relicense. Full message preserved in the original branch commitbb77eb5. See ADR-0002 for the decision rationale. Refs: ADR-0002, commitbb77eb5
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Pry — AI Agent Plugin.
|
|
OpenAPI spec server, MCP server wrapper, and GPT Action helper."""
|
|
|
|
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Rug Munch Media LLC
|
|
#
|
|
# Part of Pry — https://git.rugmunch.io/RugMunchMedia/pryscraper
|
|
# Licensed under MIT. See LICENSE.
|
|
|
|
import json
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Any, cast
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
OPENAPI_SPEC_PATH = Path(__file__).parent / "openapi.json"
|
|
|
|
|
|
def get_openapi_spec() -> dict[str, Any]:
|
|
"""Get the OpenAPI spec for AI agent integration."""
|
|
if OPENAPI_SPEC_PATH.exists():
|
|
try:
|
|
spec = cast(dict[str, Any], json.loads(OPENAPI_SPEC_PATH.read_text()))
|
|
return spec
|
|
except json.JSONDecodeError as e:
|
|
logger.error("openapi_spec_parse_failed", extra={"error": str(e)})
|
|
return {"error": "OpenAPI spec not found"}
|
|
|
|
|
|
def get_gpt_action_manifest() -> dict[str, Any]:
|
|
"""Get the GPT Action manifest for ChatGPT integration.
|
|
|
|
This manifest configures ChatGPT to use Pry as a custom GPT Action.
|
|
"""
|
|
return {
|
|
"schema_version": "v1",
|
|
"name_for_human": "Pry Web Scraper",
|
|
"name_for_model": "pry_scraper",
|
|
"description_for_human": "Scrape, crawl, and extract data from any website. Get clean markdown, structured JSON, and monitor changes.",
|
|
"description_for_model": "Plugin for scraping websites. Use it to extract content from URLs, crawl sites, extract structured data with CSS selectors, check legal compliance, monitor changes, and more.",
|
|
"auth": {"type": "none"},
|
|
"api": {"type": "openapi", "url": "/openapi.json"},
|
|
"contact_email": "support@pry.dev",
|
|
"legal_info_url": "https://pry.dev/terms",
|
|
}
|
|
|
|
|
|
def get_mcp_server_config() -> dict[str, Any]:
|
|
"""Get the MCP server config for Claude/Cursor integration.
|
|
|
|
This can be added to the AI tool's MCP configuration.
|
|
"""
|
|
return {
|
|
"pry_scraper": {
|
|
"type": "local",
|
|
"command": ["python3", "-m", "mcp_server"],
|
|
"description": "Scrape, crawl, and extract data from any website",
|
|
"enabled": True,
|
|
}
|
|
}
|