用搜索支撑提升RAG回答质量

import requests, os

API_KEY = os.environ["SCAVIO_API_KEY"]
H = {"x-api-key": API_KEY, "Content-Type": "application/json"}

FRESH_KEYWORDS = ["current", "latest", "today", "price", "cost", "news", "trending"]

def needs_fresh_data(query: str) -> bool:
    return any(kw in query.lower() for kw in FRESH_KEYWORDS)

def search_grounding(query: str) -> str:
    resp = requests.post(
        "https://api.scavio.dev/api/v1/search",
        headers=H,
        json={"query": query, "country_code": "us"},
        timeout=10,
    )
    results = resp.json().get("organic_results", [])[:5]
    return "\n".join(
        f"[{r['title']}]({r['link']}): {r['snippet']}" for r in results
    )

def rag_with_grounding(query: str, vector_context: str) -> str:
    context_parts = [f"Knowledge base:\n{vector_context}"]
    if needs_fresh_data(query):
        fresh = search_grounding(query)
        context_parts.append(f"Live web results (searched just now):\n{fresh}")
    return "\n\n".join(context_parts)

# Example: merge vector store + live search for grounded RAG
vector_ctx = "Tesla Model 3 starts at $38,990 as of Q1 2026..."
grounded = rag_with_grounding("current Tesla Model 3 price", vector_ctx)
print(grounded)

const API_KEY = process.env.SCAVIO_API_KEY;
const H = {"x-api-key": API_KEY, "Content-Type": "application/json"};

const FRESH_KEYWORDS = ["current", "latest", "today", "price", "cost", "news", "trending"];

function needsFreshData(query) {
  return FRESH_KEYWORDS.some(kw => query.toLowerCase().includes(kw));
}

async function searchGrounding(query) {
  const res = await fetch("https://api.scavio.dev/api/v1/search", {
    method: "POST",
    headers: H,
    body: JSON.stringify({ query, country_code: "us" }),
  });
  const results = (await res.json()).organic_results || [];
  return results.slice(0, 5)
    .map(r => `[${r.title}](${r.link}): ${r.snippet}`)
    .join("\n");
}

async function ragWithGrounding(query, vectorContext) {
  const parts = [`Knowledge base:\n${vectorContext}`];
  if (needsFreshData(query)) {
    const fresh = await searchGrounding(query);
    parts.push(`Live web results (searched just now):\n${fresh}`);
  }
  return parts.join("\n\n");
}

const vectorCtx = "Tesla Model 3 starts at $38,990 as of Q1 2026...";
const grounded = await ragWithGrounding("current Tesla Model 3 price", vectorCtx);
console.log(grounded);