Spaces:
Running
Running
File size: 1,686 Bytes
de28957 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | from __future__ import annotations
from app.db.postgres import get_readonly_connection
async def get_article_detail(article_id: int) -> dict:
"""
Resolves a single article's full title/snippet/metadata for citation
display. Used by the agent post-processing step to enrich raw
article_id references into full source objects.
Returns: {"rows": [...], "source_refs": [{"type": "article", "id": ...}]}
(kept in the same shape as sql_tool/graph_tool for consistency, even
though this always returns 0 or 1 row.)
"""
query = """
SELECT id, title, url, published_at, content
FROM news_articles
WHERE id = %s
"""
async with get_readonly_connection() as conn:
cur = await conn.execute(query, (article_id,))
row = await cur.fetchone()
if row is None:
return {"rows": [], "source_refs": []}
columns = [desc[0] for desc in cur.description]
article = dict(zip(columns, row))
# Truncate content into a short snippet — don't return full article text
# for a citation lookup, keep responses lean
snippet = (article.get("content") or "")[:280]
if len(article.get("content") or "") > 280:
snippet += "..."
return {
"rows": [{
"id": article["id"],
"title": article["title"],
"url": article["url"],
"published_at": (
article["published_at"].isoformat()
if article.get("published_at") else None
),
"snippet": snippet,
}],
"source_refs": [{"type": "article", "id": article["id"]}],
}
|