kolzchut-ragbot 1.7.4__py3-none-any.whl → 1.7.5__py3-none-any.whl

kolzchut_ragbot/engine.py

@@ -306,7 +306,13 @@ class Engine:
 
         # Combine documents
         top_k_documents_and_additional_document = top_k_documents.copy()
+        # Remove documents with the same page_id as the additional_document before appending
         if additional_document:
+            additional_page_id = additional_document.get("page_id")
+            top_k_documents_and_additional_document = [
+                doc for doc in top_k_documents_and_additional_document
+                if doc.get("page_id") != additional_page_id
+            ]
             top_k_documents_and_additional_document.append(additional_document)
 
         # Query LLM
@@ -326,14 +332,13 @@ class Engine:
         }
 
         request_params['timers_ms'] = {
-            "answer_time":  int(answer_time*1000),
-            "retrieval_time": int(retrieval_time*1000),
-            "llm_time": int(gpt_elapsed*1000)
+            "answer_time": int(answer_time * 1000),
+            "retrieval_time": int(retrieval_time * 1000),
+            "llm_time": int(gpt_elapsed * 1000)
         }
 
         if additional_page_time:
-            request_params['timers_ms']['additional_page_time'] = int(additional_page_time*1000)
-
+            request_params['timers_ms']['additional_page_time'] = int(additional_page_time * 1000)
 
         return top_k_documents, gpt_answer, stats, all_docs_and_scores, request_params
 

{kolzchut_ragbot-1.7.4.dist-info → kolzchut_ragbot-1.7.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kolzchut-ragbot
-Version: 1.7.4
+Version: 1.7.5
 Summary: A search engine using machine learning models and Elasticsearch for advanced document retrieval.
 Home-page: https://github.com/shmuelrob/rag-bot
 Author: Shmuel Robinov

{kolzchut_ragbot-1.7.4.dist-info → kolzchut_ragbot-1.7.5.dist-info}/RECORD

@@ -2,11 +2,11 @@ kolzchut_ragbot/Document.py,sha256=5OyBBTZyAJFM_1Pjs3SUC-_s5zEJ5U6wjhw12_FFkdE,3
 kolzchut_ragbot/IntegrateService.py,sha256=rcwUY2RkclCY3l8BGAmNbstdxhxwhLO9oA8BofqLyts,96
 kolzchut_ragbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kolzchut_ragbot/config.py,sha256=uILFvgn9W92-NRaKXYtaoQXpn3KOWKK8SZYRsIAa5Yw,133
-kolzchut_ragbot/engine.py,sha256=0tkznQd2kCawnCru3nuGUfG66H_iRaPkFOtk0r5kxvg,16015
+kolzchut_ragbot/engine.py,sha256=-qNSnyq4A3rQZz_HU_5Djwc35Su0M4tAqNvqLlDQ_UU,16383
 kolzchut_ragbot/get_full_documents_utilities.py,sha256=YWljmGWM6h1ghLDCAUnDdhmn-0k6R_t7b1g7wSojzvg,1882
 kolzchut_ragbot/llm_client.py,sha256=JdDeOn2THpkOM2Mwe2DucTaYXul1fL2agIisBuHFtc8,347
 kolzchut_ragbot/model.py,sha256=HCi3r4YztPknnbgTOA7I-GVaqxn8CzrTeLFkEg-7fg0,6320
-kolzchut_ragbot-1.7.4.dist-info/METADATA,sha256=xviSAqe9g2sy4DbsKT13cVmHN4fEdgkZFv3-i2MjPK4,1999
-kolzchut_ragbot-1.7.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kolzchut_ragbot-1.7.4.dist-info/top_level.txt,sha256=NTZoY4GGw3v_7jm0MgcdHw8simoZ78PsR7Meqmkgd_Q,16
-kolzchut_ragbot-1.7.4.dist-info/RECORD,,
+kolzchut_ragbot-1.7.5.dist-info/METADATA,sha256=9jpBjTYYUzmuNuQ_t5L-fPjOJfKoCfBSEree9DnuazE,1999
+kolzchut_ragbot-1.7.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kolzchut_ragbot-1.7.5.dist-info/top_level.txt,sha256=NTZoY4GGw3v_7jm0MgcdHw8simoZ78PsR7Meqmkgd_Q,16
+kolzchut_ragbot-1.7.5.dist-info/RECORD,,