Skip to content
Snippets Groups Projects
Select Git revision
  • 44f3ce79b06df2d15e66a04fdce5f1e38bba53f0
  • master default protected
2 results

test_heap.py

Blame
    • doc_output.py 955 B
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28 
    

    

    

    print("----------------------------------------------> Importation des modules")
import os

from langchain.embeddings import HuggingFaceEmbeddings
from langchain.vectorstores import Chroma

embedding_function = HuggingFaceEmbeddings()
VECTORS_DIRECTORY = "text_files"  # Change this directory if needed

# Charge des documents locaux

print("----------------------------------------------> Charge des documents locaux")
persist_directory = os.path.join(VECTORS_DIRECTORY, "vector")
new_db = Chroma(persist_directory=persist_directory, embedding_function=embedding_function)
print("Entrez votre requête :")

query = input()
embedding_vector = embedding_function.embed_query(query)

tot_docs = ""
print("----------------------------------------------> Recherche dans la documentation")
docs = new_db.similarity_search_by_vector(embedding_vector, k=3)

for i in range(len(docs)):
    tot_docs += f'data {str(i)} : {docs[i].page_content}\n'

print(tot_docs)