mgil
/
LLm2Node
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: get float in filtred to threshold

This commit is contained in:
Mario Gil 2023-10-25 16:51:05 -05:00
parent 07c6fe9889
commit d7429cda09
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@ -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={} AllData={}
for dbt in dbs: for dbt in dbs:
Sal = dbt.similarity_search_with_score(query,4) 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(): #for item in AllData.items():
# print(item) # print(item)
if filtred: if filtred>0:
filtredData={} filtredData={}
for row in AllData.keys(): for row in AllData.keys():
if AllData[row]["d"]<th: if AllData[row]["d"]<filtred:
filtredData[row]=AllData[row] filtredData[row]=AllData[row]
filtredData=dict(sorted(filtredData.items(), key=lambda item: item[1]["d"])) filtredData=dict(sorted(filtredData.items(), key=lambda item: item[1]["d"]))
return filtredData,filtredData.keys() return filtredData,filtredData.keys()
@ -165,7 +165,7 @@ def read_main():
class Response(BaseModel): class Response(BaseModel):
query: str query: str
filtred : Optional[int] = 0 filtred : Optional[float] = -9.0
@ -175,11 +175,8 @@ def calculate_api(response: Response):
try: try:
filtred = response.filtred filtred = response.filtred
except: except:
filtred = 0 filtred = -9.0
if filtred==1:
filtred=True
else:
filtred=False
AllData=FinderDbs(query,[db2],filtred) AllData=FinderDbs(query,[db2],filtred)
versionL="_".join([model,entrenamiento]) versionL="_".join([model,entrenamiento])
if AllData: if AllData: