diff --git a/src/ai_generators/ollama_md_generator.py b/src/ai_generators/ollama_md_generator.py index 18a5a8d..5c5dd0b 100644 --- a/src/ai_generators/ollama_md_generator.py +++ b/src/ai_generators/ollama_md_generator.py @@ -134,7 +134,7 @@ class OllamaGenerator: {self.content} """ try: - query_embed = self.ollama_client.embed(model=self.embed_model, input=prompt_system)['embeddings'] + query_embed = self.ollama_client.embed(model=self.embed_model, input=prompt_human)['embeddings'] collection = self.load_to_vector_db() collection_query = collection.query(query_embeddings=query_embed, n_results=100) print("Showing pertinent info from drafts used in final edited edition")