Spaces:

SorbonneUniversity
/

SorboBot

Sleeping

Léo Bourrel commited on Nov 15, 2023

Commit

dc294ab

1 Parent(s): 9f90955

fix: return abstract list first element only

Files changed (1) hide show

sorbobotapp/vector_store.py CHANGED Viewed

@@ -244,7 +244,7 @@ class CustomVectorStore(VectorStore):
                         "doi": result["doi"],
                         "hal_id": result["hal_id"],
                         "distance": result["distance"],
-                        "abstract": result["abstract"],
                     },
                 ),
                 result["distance"] if self.embedding_function is not None else None,

                         "doi": result["doi"],
                         "hal_id": result["hal_id"],
                         "distance": result["distance"],
+                        "abstract": result["abstract"][0],
                     },
                 ),
                 result["distance"] if self.embedding_function is not None else None,