Fix FAI retrieval (#4869)

tsbhangu · web-flow · commit 906fea9c04b2 · 2025-11-06T18:14:16.000-05:00
diff --git a/servers/fai/src/fai/utils/chat/retrieve/retrieve.py b/servers/fai/src/fai/utils/chat/retrieve/retrieve.py
@@ -12,7 +12,7 @@
     get_tpuf_namespace,
 )
 
-DOCUMENT_ATTRIBUTES = ["document", "url", "version", "product", "roles", "title", "keywords"]
+DOCUMENT_ATTRIBUTES = ["chunk", "document", "url", "version", "product", "roles", "title", "keywords"]
 
 
 def _results_to_ranks(items: list[Row]) -> dict[str, int]:

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`get_tpuf_namespace,`
`13`	`13`	`)`
`14`	`14`
`15`		`-DOCUMENT_ATTRIBUTES = ["document", "url", "version", "product", "roles", "title", "keywords"]`
	`15`	`+DOCUMENT_ATTRIBUTES = ["chunk", "document", "url", "version", "product", "roles", "title", "keywords"]`
`16`	`16`
`17`	`17`
`18`	`18`	`def _results_to_ranks(items: list[Row]) -> dict[str, int]:`