datacipen commited on
Commit
462f117
·
verified ·
1 Parent(s): 20bba35

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -730,7 +730,7 @@ async def API_France_Travail(romeListArray):
730
 
731
  @cl.step(type="tool")
732
  async def creation_liste_code_Rome(competence, categorie):
733
- docsearch = await connexion_catalogue_Rome()
734
  retrieve_comp = docsearch.similarity_search(competence, k=30, filter={"categorie": {"$eq": categorie}})
735
  retrieve = pd.DataFrame(retrieve_comp)
736
  codeRome = []
@@ -785,7 +785,7 @@ async def creation_liste_code_Rome(competence, categorie):
785
 
786
  @cl.step(type="tool")
787
  async def creation_liste_skills_Esco(competence, categorie):
788
- docsearch = await connexion_catalogue_Rome()
789
  retrieve_comp = docsearch.similarity_search(competence, k=40, filter={"categorie": {"$eq": categorie}})
790
  retrieve = pd.DataFrame(retrieve_comp)
791
  competence = []
@@ -846,7 +846,7 @@ async def connexion_France_Travail():
846
  return client
847
 
848
  @cl.step(type="tool")
849
- async def connexion_catalogue_Rome():
850
  os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
851
  os.environ['PINECONE_INDEX_NAME'] = os.environ['PINECONE_INDEX_NAME']
852
  embeddings = HuggingFaceEmbeddings()
 
730
 
731
  @cl.step(type="tool")
732
  async def creation_liste_code_Rome(competence, categorie):
733
+ docsearch = await connexion_vector_database()
734
  retrieve_comp = docsearch.similarity_search(competence, k=30, filter={"categorie": {"$eq": categorie}})
735
  retrieve = pd.DataFrame(retrieve_comp)
736
  codeRome = []
 
785
 
786
  @cl.step(type="tool")
787
  async def creation_liste_skills_Esco(competence, categorie):
788
+ docsearch = await connexion_vector_database()
789
  retrieve_comp = docsearch.similarity_search(competence, k=40, filter={"categorie": {"$eq": categorie}})
790
  retrieve = pd.DataFrame(retrieve_comp)
791
  competence = []
 
846
  return client
847
 
848
  @cl.step(type="tool")
849
+ async def connexion_vector_database():
850
  os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
851
  os.environ['PINECONE_INDEX_NAME'] = os.environ['PINECONE_INDEX_NAME']
852
  embeddings = HuggingFaceEmbeddings()