phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 12:06:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

incorporate feedback

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-10-07 13:54:41 -04:00
parent eb8acd24b0
commit 08305203d3
1 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

25
llama_stack/providers/inline/agents/meta_reference/responses/tool_executor.py
View file

@ -158,9 +158,9 @@ class ToolExecutor:
citation_instruction = ""
if unique_files:
citation_instruction = " Cite sources immediately at the end of sentences before punctuation, using `<|file-id|>` format (e.g., 'This is a fact <|file-Cn3MSNn72ENTiiq11Qda4A|>.')."
citation_instruction += (
" Do not add extra punctuation. Use only the file IDs provided (do not invent new ones)."
citation_instruction = (
" Cite sources immediately at the end of sentences before punctuation, using `<|file-id|>` format (e.g., 'This is a fact <|file-Cn3MSNn72ENTiiq11Qda4A|>.'). "
"Do not add extra punctuation. Use only the file IDs provided (do not invent new ones)."
)
content_items.append(
@ -170,12 +170,19 @@ class ToolExecutor:
)
# handling missing attributes for old versions
citation_files = {
(r.file_id or (r.attributes.get("document_id") if r.attributes else None)): r.filename
or (r.attributes.get("filename") if r.attributes else None)
or "unknown"
for r in search_results
}
citation_files = {}
for result in search_results:
file_id = result.file_id
if not file_id and result.attributes:
file_id = result.attributes.get("document_id")
filename = result.filename
if not filename and result.attributes:
filename = result.attributes.get("filename")
if not filename:
filename = "unknown"
citation_files[file_id] = filename
return ToolInvocationResult(
content=content_items,