From 08f4e2ad70b09c516854970ae2b2b7ce9370e1b8 Mon Sep 17 00:00:00 2001 From: seehi <6580@pm.me> Date: Fri, 15 Mar 2024 19:14:59 +0800 Subject: [PATCH] rag_add_objs catch exception --- examples/rag_pipeline.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/rag_pipeline.py b/examples/rag_pipeline.py index 70d748b7d..1575bb47c 100644 --- a/examples/rag_pipeline.py +++ b/examples/rag_pipeline.py @@ -122,14 +122,15 @@ class RAGExample: logger.info("[After add objs]") self.engine.add_objs([player]) - nodes = await self._retrieve_and_print(question) - logger.info("[Object Detail]") try: + nodes = await self._retrieve_and_print(question) + + logger.info("[Object Detail]") player: Player = nodes[0].metadata["obj"] logger.info(player.name) except Exception as e: - logger.info(f"ERROR: nodes is empty, llm don't answer correctly, exception: {e}") + logger.error(f"nodes is empty, llm don't answer correctly, exception: {e}") async def rag_ini_objs(self): """This example show how to from objs, will print something like: