adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(citation): add download_url to citations when allow_download is enabled

Browse Source
This commit is contained in:
Timebomb2018
2026-04-24 14:44:27 +08:00
parent 89f2f9a045
commit 2c2551e15c
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
api/app/services/workflow_service.py
View File

@@ -773,9 +773,16 @@ class WorkflowService:
# 过滤 citations # 过滤 citations
citations = result.get("citations", []) citations = result.get("citations", [])
citation_cfg = feature_configs.get("citation", {}) citation_cfg = feature_configs.get("citation", {})
filtered_citations = ( if isinstance(citation_cfg, dict) and citation_cfg.get("enabled"):
citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else [] allow_download = citation_cfg.get("allow_download", False)
) if allow_download:
from app.core.config import settings
for c in citations:
if c.get("document_id"):
c["download_url"] = f"{settings.FILE_LOCAL_SERVER_URL}/apps/citations/{c['document_id']}/download"
filtered_citations = citations
else:
filtered_citations = []
assistant_meta = {"usage": token_usage, "audio_url": None} assistant_meta = {"usage": token_usage, "audio_url": None}
if filtered_citations: if filtered_citations:
assistant_meta["citations"] = filtered_citations assistant_meta["citations"] = filtered_citations
@@ -975,9 +982,16 @@ class WorkflowService:
# 过滤 citations # 过滤 citations
citations = event.get("data", {}).get("citations", []) citations = event.get("data", {}).get("citations", [])
citation_cfg = feature_configs.get("citation", {}) citation_cfg = feature_configs.get("citation", {})
filtered_citations = ( if isinstance(citation_cfg, dict) and citation_cfg.get("enabled"):
citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else [] allow_download = citation_cfg.get("allow_download", False)
) if allow_download:
from app.core.config import settings
for c in citations:
if c.get("document_id"):
c["download_url"] = f"{settings.FILE_LOCAL_SERVER_URL}/apps/citations/{c['document_id']}/download"
filtered_citations = citations
else:
filtered_citations = []
assistant_meta = {"usage": token_usage, "audio_url": None} assistant_meta = {"usage": token_usage, "audio_url": None}
if filtered_citations: if filtered_citations:
assistant_meta["citations"] = filtered_citations assistant_meta["citations"] = filtered_citations