diff --git a/rag/prompts.py b/rag/prompts.py index 389d6a66d..66c8a4ef4 100644 --- a/rag/prompts.py +++ b/rag/prompts.py @@ -427,7 +427,8 @@ Output: res = {} for k, v in obj.items(): try: - res[str(k)] = int(v) + if int(v) > 0: + res[str(k)] = int(v) except Exception: pass return res