promptreview-proxy 0.1.0__tar.gz

promptreview_proxy-0.1.0/PKG-INFO

@@ -0,0 +1,53 @@
+Metadata-Version: 2.4
+Name: promptreview-proxy
+Version: 0.1.0
+Summary: Local HTTP proxy forwarding engine for PromptReview VS Code extension with API subdomain analytics.
+Author-email: PromptReview Team <support@prompt-guard-ai.com>
+License: MIT
+Keywords: llm,proxy,fastapi,vscode,promptreview,ai
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Framework :: FastAPI
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.110.0
+Requires-Dist: uvicorn>=0.28.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: pydantic>=2.6.0
+
+# PromptReview Proxy (`promptreview-proxy`)
+
+`promptreview-proxy` is a lightweight, local HTTP proxy forwarding server for the PromptReview VS Code Extension.
+
+## Features
+- **Local HTTP Proxying**: Intercepts outbound LLM traffic from VS Code (`http.proxy`).
+- **Cloud Analytics Relay**: Asynchronously logs prompt metadata to your API subdomain (`https://api.prompt-guard-ai.com/v1/prompts`).
+- **Dynamic Provider Routing**: Forwards requests to target LLM providers (OpenAI, Anthropic, Groq, Ollama, etc.) and streams responses back to VS Code.
+
+## Quick Start
+
+### Installation
+```bash
+pip install promptreview-proxy
+```
+
+### Usage
+Run the proxy server from command line:
+```bash
+promptreview-proxy
+```
+
+Or configure environment variables:
+```bash
+export PROXY_PORT=6767
+export API_BASE_URL="https://api.prompt-guard-ai.com"
+promptreview-proxy
+```

promptreview_proxy-0.1.0/README.md

@@ -0,0 +1,28 @@
+# PromptReview Proxy (`promptreview-proxy`)
+
+`promptreview-proxy` is a lightweight, local HTTP proxy forwarding server for the PromptReview VS Code Extension.
+
+## Features
+- **Local HTTP Proxying**: Intercepts outbound LLM traffic from VS Code (`http.proxy`).
+- **Cloud Analytics Relay**: Asynchronously logs prompt metadata to your API subdomain (`https://api.prompt-guard-ai.com/v1/prompts`).
+- **Dynamic Provider Routing**: Forwards requests to target LLM providers (OpenAI, Anthropic, Groq, Ollama, etc.) and streams responses back to VS Code.
+
+## Quick Start
+
+### Installation
+```bash
+pip install promptreview-proxy
+```
+
+### Usage
+Run the proxy server from command line:
+```bash
+promptreview-proxy
+```
+
+Or configure environment variables:
+```bash
+export PROXY_PORT=6767
+export API_BASE_URL="https://api.prompt-guard-ai.com"
+promptreview-proxy
+```

promptreview_proxy-0.1.0/promptreview_proxy/__init__.py

@@ -0,0 +1,3 @@
+"""PromptReview Proxy Package"""
+
+__version__ = "0.1.0"

promptreview_proxy-0.1.0/promptreview_proxy/server.py

@@ -0,0 +1,198 @@
+import asyncio
+import json
+import os
+import sys
+from typing import Any, AsyncGenerator
+
+import httpx
+from fastapi import FastAPI, Header, HTTPException, Request, Response
+from fastapi.responses import JSONResponse, StreamingResponse
+from pydantic import BaseModel, Field
+
+
+class ProxyConfig:
+    def __init__(self) -> None:
+        self.host = os.getenv("PROXY_HOST", "127.0.0.1")
+        self.port = int(os.getenv("PROXY_PORT", "6767"))
+
+        self.api_base_url = os.getenv("API_BASE_URL", "https://api.prompt-guard-ai.com").rstrip("/")
+        self.api_token = os.getenv("API_TOKEN", "")
+
+        self.llm_provider_url = os.getenv("LLM_PROVIDER_URL", "https://api.openai.com/v1/chat/completions")
+        self.llm_provider_api_key = os.getenv("LLM_PROVIDER_API_KEY", "")
+
+
+config = ProxyConfig()
+app = FastAPI(title="PromptReview Local Proxy", version="0.1.0")
+
+
+class ProxyChatRequest(BaseModel):
+    company: str = Field(..., description="Organization name or email (C1)")
+    project: str = Field(default="default-project", description="Project name (P1)")
+    user: str = Field(..., description="User identity or email (U1)")
+    prompt: str = Field(..., min_length=1, description="Prompt string")
+    model: str | None = Field(default="gpt-4o", description="Target LLM model")
+    messages: list[dict[str, Any]] | None = Field(default=None, description="Optional messages payload")
+    metadata: dict[str, Any] | None = Field(default_factory=dict)
+
+
+async def send_event_to_api_subdomain(payload: dict[str, Any]) -> None:
+    """Asynchronously dispatch prompt event to API subdomain for cloud storage."""
+    target_url = f"{config.api_base_url}/v1/prompts"
+    headers = {"Content-Type": "application/json"}
+    if config.api_token:
+        headers["Authorization"] = f"Bearer {config.api_token}"
+
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            res = await client.post(target_url, json=payload, headers=headers)
+            if res.status_code >= 400:
+                print(f"[promptreview-proxy] API subdomain warning ({res.status_code}): {res.text}", file=sys.stderr)
+    except Exception as exc:
+        print(f"[promptreview-proxy] Failed sending event to API subdomain: {exc}", file=sys.stderr)
+
+
+@app.get("/health")
+async def health() -> dict[str, str]:
+    return {"status": "ok", "service": "promptreview-proxy"}
+
+
+@app.post("/v1/proxy/chat")
+async def proxy_chat(
+    request: ProxyChatRequest,
+    x_company: str | None = Header(default=None),
+    x_project: str | None = Header(default=None),
+    x_user: str | None = Header(default=None),
+    authorization: str | None = Header(default=None),
+) -> dict[str, Any]:
+    company = request.company or x_company or "UnknownCompany"
+    project = request.project or x_project or "default-project"
+    user = request.user or x_user or "UnknownUser"
+
+    api_payload = {
+        "company": company,
+        "project": project,
+        "user": user,
+        "prompt": request.prompt,
+        "model": request.model,
+        "messages": request.messages or [{"role": "user", "content": request.prompt}],
+        "metadata": request.metadata or {},
+    }
+
+    asyncio.create_task(send_event_to_api_subdomain(api_payload))
+
+    llm_payload = {
+        "model": request.model,
+        "messages": request.messages or [{"role": "user", "content": request.prompt}],
+    }
+
+    headers = {"Content-Type": "application/json"}
+    if authorization:
+        headers["Authorization"] = authorization
+    elif config.llm_provider_api_key:
+        headers["Authorization"] = f"Bearer {config.llm_provider_api_key}"
+
+    try:
+        async with httpx.AsyncClient(timeout=60.0) as client:
+            llm_res = await client.post(config.llm_provider_url, json=llm_payload, headers=headers)
+            llm_res.raise_for_status()
+            return {
+                "ok": True,
+                "company": company,
+                "project": project,
+                "user": user,
+                "provider_response": llm_res.json(),
+            }
+    except Exception as exc:
+        raise HTTPException(status_code=502, detail=f"LLM Provider error: {str(exc)}")
+
+
+@app.api_route("/v1/chat/completions", methods=["POST", "OPTIONS"])
+@app.api_route("/{full_path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD"])
+async def catch_all_proxy(request: Request, full_path: str = "") -> Response:
+    if request.method == "OPTIONS":
+        return Response(status_code=204)
+
+    body_bytes = await request.body()
+    body_json: dict[str, Any] = {}
+    if body_bytes:
+        try:
+            body_json = json.loads(body_bytes.decode("utf-8"))
+        except Exception:
+            body_json = {}
+
+    headers = dict(request.headers)
+    company = headers.get("x-company") or headers.get("x-organization-name") or "C1"
+    project = headers.get("x-project") or "P1"
+    user = headers.get("x-user") or headers.get("x-user-email") or "U1"
+
+    messages = body_json.get("messages", [])
+    prompt = ""
+    if messages and isinstance(messages, list):
+        last_msg = messages[-1]
+        if isinstance(last_msg, dict):
+            prompt = str(last_msg.get("content", ""))
+    elif "prompt" in body_json:
+        prompt = str(body_json.get("prompt", ""))
+
+    if prompt:
+        api_payload = {
+            "company": company,
+            "project": project,
+            "user": user,
+            "prompt": prompt,
+            "model": body_json.get("model"),
+            "messages": messages,
+            "metadata": {"path": full_path},
+        }
+        asyncio.create_task(send_event_to_api_subdomain(api_payload))
+
+    target_url = headers.get("x-target-llm-url") or config.llm_provider_url
+    if full_path and not target_url.endswith(full_path):
+        if not target_url.endswith("/"):
+            target_url = f"{target_url}/{full_path}"
+        else:
+            target_url = f"{target_url}{full_path}"
+
+    forward_headers = {
+        k: v for k, v in headers.items() if k.lower() not in ("host", "content-length", "x-target-llm-url", "x-company", "x-project", "x-user")
+    }
+
+    try:
+        client = httpx.AsyncClient(timeout=120.0)
+        req = client.build_request(
+            method=request.method,
+            url=target_url,
+            headers=forward_headers,
+            content=body_bytes,
+        )
+        res = await client.send(req, stream=True)
+
+        async def stream_response() -> AsyncGenerator[bytes, None]:
+            try:
+                async for chunk in res.aiter_bytes():
+                    yield chunk
+            finally:
+                await res.aclose()
+                await client.aclose()
+
+        return StreamingResponse(
+            stream_response(),
+            status_code=res.status_code,
+            headers={k: v for k, v in res.headers.items() if k.lower() not in ("content-length", "transfer-encoding")},
+        )
+    except Exception as exc:
+        return JSONResponse(
+            status_code=502,
+            content={"error": f"Local proxy forwarding failed: {str(exc)}"},
+        )
+
+
+def main() -> None:
+    import uvicorn
+
+    uvicorn.run(app, host=config.host, port=config.port)
+
+
+if __name__ == "__main__":
+    main()

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/PKG-INFO

@@ -0,0 +1,53 @@
+Metadata-Version: 2.4
+Name: promptreview-proxy
+Version: 0.1.0
+Summary: Local HTTP proxy forwarding engine for PromptReview VS Code extension with API subdomain analytics.
+Author-email: PromptReview Team <support@prompt-guard-ai.com>
+License: MIT
+Keywords: llm,proxy,fastapi,vscode,promptreview,ai
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Framework :: FastAPI
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.110.0
+Requires-Dist: uvicorn>=0.28.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: pydantic>=2.6.0
+
+# PromptReview Proxy (`promptreview-proxy`)
+
+`promptreview-proxy` is a lightweight, local HTTP proxy forwarding server for the PromptReview VS Code Extension.
+
+## Features
+- **Local HTTP Proxying**: Intercepts outbound LLM traffic from VS Code (`http.proxy`).
+- **Cloud Analytics Relay**: Asynchronously logs prompt metadata to your API subdomain (`https://api.prompt-guard-ai.com/v1/prompts`).
+- **Dynamic Provider Routing**: Forwards requests to target LLM providers (OpenAI, Anthropic, Groq, Ollama, etc.) and streams responses back to VS Code.
+
+## Quick Start
+
+### Installation
+```bash
+pip install promptreview-proxy
+```
+
+### Usage
+Run the proxy server from command line:
+```bash
+promptreview-proxy
+```
+
+Or configure environment variables:
+```bash
+export PROXY_PORT=6767
+export API_BASE_URL="https://api.prompt-guard-ai.com"
+promptreview-proxy
+```

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+promptreview_proxy/__init__.py
+promptreview_proxy/server.py
+promptreview_proxy.egg-info/PKG-INFO
+promptreview_proxy.egg-info/SOURCES.txt
+promptreview_proxy.egg-info/dependency_links.txt
+promptreview_proxy.egg-info/entry_points.txt
+promptreview_proxy.egg-info/requires.txt
+promptreview_proxy.egg-info/top_level.txt

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+promptreview-proxy = promptreview_proxy.server:main

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/requires.txt

@@ -0,0 +1,4 @@
+fastapi>=0.110.0
+uvicorn>=0.28.0
+httpx>=0.27.0
+pydantic>=2.6.0

promptreview_proxy-0.1.0/promptreview_proxy.egg-info/top_level.txt

@@ -0,0 +1 @@
+promptreview_proxy

promptreview_proxy-0.1.0/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["setuptools>=77", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "promptreview-proxy"
+version = "0.1.0"
+description = "Local HTTP proxy forwarding engine for PromptReview VS Code extension with API subdomain analytics."
+readme = "README.md"
+requires-python = ">=3.9"
+authors = [{ name = "PromptReview Team", email = "support@prompt-guard-ai.com" }]
+license = { text = "MIT" }
+keywords = ["llm", "proxy", "fastapi", "vscode", "promptreview", "ai"]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Environment :: Console",
+  "Framework :: FastAPI",
+  "Intended Audience :: Developers",
+  "Operating System :: OS Independent",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13"
+]
+dependencies = [
+  "fastapi>=0.110.0",
+  "uvicorn>=0.28.0",
+  "httpx>=0.27.0",
+  "pydantic>=2.6.0"
+]
+
+[project.scripts]
+promptreview-proxy = "promptreview_proxy.server:main"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["promptreview_proxy*"]

promptreview_proxy-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+