datacipen commited on
Commit
bf8dc9c
1 Parent(s): bfcda27

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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
- return codeSecteur
 
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):