|
@@ -77,10 +77,10 @@ class Ident:
|
|
identified_fields = identified_fields.replace(", ", "\n")
|
|
identified_fields = identified_fields.replace(", ", "\n")
|
|
print(f"New query: {identified_fields}")
|
|
print(f"New query: {identified_fields}")
|
|
|
|
|
|
- text_splitter = CharacterTextSplitter(chunk_size=500, chunk_overlap=100)
|
|
|
|
|
|
+ text_splitter = CharacterTextSplitter()
|
|
docs = text_splitter.split_documents(documents)
|
|
docs = text_splitter.split_documents(documents)
|
|
# print(documents)
|
|
# print(documents)
|
|
- self.vectorstore = Chroma.from_documents(docs, embedding=OllamaEmbeddings(model="nomic-embed-text"))
|
|
|
|
|
|
+ self.vectorstore = Chroma.from_documents(docs, embedding=OllamaEmbeddings(model="mxbai-embed-large"))
|
|
# print(f"Response (2): {self.vectorstore.similarity_search_with_score(identified_fields)}")
|
|
# print(f"Response (2): {self.vectorstore.similarity_search_with_score(identified_fields)}")
|
|
self.retriever = self.vectorstore.as_retriever()
|
|
self.retriever = self.vectorstore.as_retriever()
|
|
|
|
|