Update main.py
Browse files
main.py
CHANGED
@@ -746,8 +746,8 @@ async def API_France_Travail(romeListArray):
|
|
746 |
@cl.step(type="tool")
|
747 |
async def creation_liste_entreprises(arrayOfsecteur):
|
748 |
docsearch = await connexion_vector_database()
|
749 |
-
for i in range(0, len(
|
750 |
-
retrieve_comp = docsearch.similarity_search(
|
751 |
codeSecteur = retrieve_comp[0].metadata['codefinal']
|
752 |
return codeSecteur
|
753 |
|
|
|
746 |
@cl.step(type="tool")
|
747 |
async def creation_liste_entreprises(arrayOfsecteur):
|
748 |
docsearch = await connexion_vector_database()
|
749 |
+
for i in range(0, len(arrayOfsecteur)):
|
750 |
+
retrieve_comp = docsearch.similarity_search(arrayOfsecteur[i][0], k=1, filter={"categorie": {"$eq": "inpiSecteur"}})
|
751 |
codeSecteur = retrieve_comp[0].metadata['codefinal']
|
752 |
return codeSecteur
|
753 |
|