fix bug of inserting cites (#76)

2024-02-27 17:51:54 +08:00
parent 4568a4b2cb
commit 1567e881de
5 changed files with 31 additions and 16 deletions
--- a/rag/nlp/search.py
+++ b/rag/nlp/search.py
@@ -226,7 +226,7 @@ class Dealer:
                continue
            if i not in cites:
                continue
-            assert int(cites[i]) < len(chunk_v)
+            for c in cites[i]: assert int(c) < len(chunk_v)
            res += "##%s$$" % "$".join(cites[i])

        return res