Update main.py
Browse files
main.py
CHANGED
@@ -748,8 +748,9 @@ 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 |
-
|
|
|
753 |
|
754 |
@cl.step(type="tool")
|
755 |
async def creation_liste_code_Rome(competence, categorie):
|
|
|
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 = int(retrieve_comp[0].metadata['codefinal'])
|
752 |
+
codeSecteurSTR = "0" + str(codeSecteur)
|
753 |
+
return int(codeSecteurSTR)
|
754 |
|
755 |
@cl.step(type="tool")
|
756 |
async def creation_liste_code_Rome(competence, categorie):
|