From d7429cda09f459d7729a4ff38353305b1fdac2aa Mon Sep 17 00:00:00 2001
From: marioggil <marioggil@gmail.com>
Date: Wed, 25 Oct 2023 16:51:05 -0500
Subject: [PATCH] feat: get float in filtred to threshold

---
 main.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/main.py b/main.py
index 23bf17e..e67e6f2 100644
--- a/main.py
+++ b/main.py
@@ -132,7 +132,7 @@ db2=makeFaissdb(documents2,"Intencionality3",emb2)
 
 
 
-def FinderDbs(query,dbs,filtred=False,th=1.2):
+def FinderDbs(query,dbs,filtred=1.2):
     AllData={}
     for dbt in dbs:
         Sal = dbt.similarity_search_with_score(query,4)
@@ -144,10 +144,10 @@ def FinderDbs(query,dbs,filtred=False,th=1.2):
     #for item in AllData.items():
     #    print(item)
     
-    if filtred:
+    if filtred>0:
         filtredData={}
         for row in AllData.keys():
-            if AllData[row]["d"]<th:
+            if AllData[row]["d"]<filtred:
                 filtredData[row]=AllData[row]
         filtredData=dict(sorted(filtredData.items(), key=lambda item: item[1]["d"]))
         return  filtredData,filtredData.keys()
@@ -165,7 +165,7 @@ def read_main():
 
 class Response(BaseModel):
     query: str
-    filtred : Optional[int] = 0
+    filtred : Optional[float] = -9.0
 
 
 
@@ -175,11 +175,8 @@ def calculate_api(response: Response):
     try:
         filtred = response.filtred
     except:
-        filtred = 0
-    if filtred==1:
-        filtred=True
-    else:
-        filtred=False
+        filtred = -9.0
+    
     AllData=FinderDbs(query,[db2],filtred)
     versionL="_".join([model,entrenamiento])
     if AllData: