diff --git a/main.py b/main.py index 968c7b3..e22a907 100644 --- a/main.py +++ b/main.py @@ -32,7 +32,7 @@ def chunk_data(texts: list[str], collection): chunker = Chunker(collection) print(f"chunking {len(texts)} documents") - for text in texts[: len(texts) // 2]: + for text in texts: chunker.chunk_document(document=text) @@ -40,7 +40,6 @@ def consult_oracle(input: str, collection): # Ask embeddings = Chunker.embedding_fx(input=[input]) results = collection.query(query_texts=[input], query_embeddings=embeddings) - print(results) # Generate output = ollama.generate( diff --git a/request.py b/request.py index 229d619..9357096 100644 --- a/request.py +++ b/request.py @@ -21,4 +21,4 @@ class PaperlessNGXService: if __name__ == "__main__": pp = PaperlessNGXService() - print(pp.get_data()[0].keys()) + pp.get_data()