Skip to content

Commit

Permalink
Fix display chunk issue
Browse files Browse the repository at this point in the history
Signed-off-by: jinhai <[email protected]>
  • Loading branch information
JinHai-CN committed Nov 28, 2024
1 parent d8a5b40 commit 3d17726
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion rag/utils/infinity_conn.py
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,9 @@ def get(
table_name = f"{indexName}_{knowledgebaseId}"
table_instance = db_instance.get_table(table_name)
kb_res = table_instance.output(["*"]).filter(f"id = '{chunkId}'").to_pl()
df_list.append(kb_res)
if len(kb_res) != 0 and kb_res.shape[0] > 0:
df_list.append(kb_res)

self.connPool.release_conn(inf_conn)
res = concat_dataframes(df_list, ["id"])
res_fields = self.getFields(res, res.columns)
Expand Down

0 comments on commit 3d17726

Please sign in to comment.