Skip to content

Commit 593163f

Browse files
authored
Fix output documents (#1839)
1 parent a6ea9c9 commit 593163f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gen-ai/orchestrator-server/src/main/python/tock-llm-indexing-tools/export_run_results.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def append_runs_langfuse(dataset_item, _runs_names):
132132
[f'[{doc["metadata"]["title"]}]({doc["metadata"]["source"]}) : {doc["page_content"]}'
133133
f'\n################################################################################'
134134
f'################################################################################' for doc in
135-
trace.output["source_documents"]]))
135+
trace.output["documents"]]))
136136

137137
return csv_line
138138

@@ -276,7 +276,7 @@ def append_runs_langsmith(dataset_example, _session_ids):
276276
csv_line.append('')
277277
else:
278278
csv_line.append(run["outputs"]["answer"])
279-
csv_line.append(','.join([doc["metadata"]["url"] for doc in run["outputs"]["source_documents"]]))
279+
csv_line.append(','.join([doc["metadata"]["url"] for doc in run["outputs"]["documents"]]))
280280

281281
return csv_line
282282

0 commit comments

Comments
 (0)