c762564d40
- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
125 lines
4.1 KiB
Python
125 lines
4.1 KiB
Python
"""
|
|
RAG optimization endpoints - included via APIRouter for clean git history.
|
|
Confidence scoring, Solidity chunking, deployer reputation, cross-chain,
|
|
multi-modal, investigation narratives, graph RAG, streaming, email.
|
|
"""
|
|
|
|
import httpx
|
|
from fastapi import APIRouter, Request
|
|
|
|
router = APIRouter(prefix="/api/v1", tags=["rag-optimization"])
|
|
|
|
|
|
@router.post("/rag/confidence")
|
|
async def rag_confidence_score(request: Request, data: dict):
|
|
from app.confidence import score_confidence
|
|
|
|
return score_confidence(
|
|
data.get("results", []),
|
|
query=data.get("query", ""),
|
|
entity_matches=data.get("entity_matches", []),
|
|
)
|
|
|
|
|
|
@router.post("/rag/chunk-solidity")
|
|
async def rag_chunk_solidity(request: Request, data: dict):
|
|
source = data.get("source", "")
|
|
if not source:
|
|
return {"error": "source is required"}
|
|
from app.solidity_chunker import chunk_solidity_ast
|
|
|
|
chunks = chunk_solidity_ast(source)
|
|
return {"chunks": chunks, "count": len(chunks)}
|
|
|
|
|
|
@router.post("/scanner/deployer-risk")
|
|
async def scanner_deployer_risk(request: Request, data: dict):
|
|
from app.deployer_reputation import score_deployer_risk
|
|
|
|
return score_deployer_risk(
|
|
address=data.get("address", ""),
|
|
deployment_count=data.get("deployment_count", 0),
|
|
mixer_funded=data.get("mixer_funded", False),
|
|
source_verified=data.get("source_verified", False),
|
|
code_similarity=data.get("code_similarity", 0.0),
|
|
rapid_deploy=data.get("rapid_deploy", False),
|
|
funded_by_scammer=data.get("funded_by_scammer", False),
|
|
)
|
|
|
|
|
|
@router.post("/rag/cross-chain")
|
|
async def rag_cross_chain(request: Request, data: dict):
|
|
from app.cross_chain import resolve_cross_chain_identity
|
|
|
|
return resolve_cross_chain_identity(
|
|
address=data.get("address", ""),
|
|
chain=data.get("chain", "ethereum"),
|
|
funding_sources=data.get("funding_sources", []),
|
|
label_hints=data.get("label_hints", []),
|
|
)
|
|
|
|
|
|
@router.post("/rag/image-analysis")
|
|
async def rag_image_analysis(request: Request, data: dict):
|
|
from app.multimodal_rag import describe_image
|
|
|
|
return {
|
|
"task": data.get("task", "describe"),
|
|
"description": await describe_image(image_url=data.get("image_url", ""), task=data.get("task", "describe")),
|
|
}
|
|
|
|
|
|
@router.post("/rag/logo-check")
|
|
async def rag_logo_check(request: Request, data: dict):
|
|
url = data.get("image_url", "")
|
|
if not url:
|
|
return {"error": "image_url is required"}
|
|
r = await httpx.AsyncClient(timeout=15).get(url)
|
|
from app.multimodal_rag import compute_image_hash, get_logo_db
|
|
|
|
h = compute_image_hash(r.content)
|
|
return {**get_logo_db().check_theft(h), "computed_hash": h}
|
|
|
|
|
|
@router.post("/rag/investigate")
|
|
async def rag_investigate_endpoint(request: Request, data: dict):
|
|
q = data.get("query", "")
|
|
if not q:
|
|
return {"error": "query is required"}
|
|
from app.investigation_narratives import investigate
|
|
|
|
return await investigate(query=q, investigation_type=data.get("type", "auto"), max_hops=data.get("max_hops", 5))
|
|
|
|
|
|
@router.get("/rag/graph-communities")
|
|
async def rag_graph_communities(request: Request):
|
|
from app.graph_rag import graph_rag_search
|
|
|
|
return await graph_rag_search(query="")
|
|
|
|
|
|
@router.post("/rag/watch")
|
|
async def rag_create_watch(request: Request, data: dict):
|
|
from app.streaming_rag import create_watch
|
|
|
|
return await create_watch(token_address=data.get("token_address", ""), chain=data.get("chain", "ethereum"))
|
|
|
|
|
|
@router.get("/rag/watches")
|
|
async def rag_list_watches(request: Request):
|
|
from app.streaming_rag import list_active_watches
|
|
|
|
return await list_active_watches()
|
|
|
|
|
|
@router.get("/email/accounts")
|
|
async def email_accounts(request: Request):
|
|
return {
|
|
"accounts": [
|
|
{"id": "gmail", "email": "cryptorugmuncher@gmail.com", "type": "gmail"},
|
|
{"id": "rmi", "email": "admin@rugmunch.io", "type": "gmail"},
|
|
{"id": "rmi-local", "email": "admin@rugmunch.io", "type": "local"},
|
|
{"id": "biz", "email": "biz@rugmunch.io", "type": "local"},
|
|
],
|
|
"webmail_url": "https://webmail.rugmunch.io/",
|
|
}
|