integrating agentic chroma

2025-02-26 23:16:00 +10:00
parent 9b11fea0e7
commit e7f7a79d86
3 changed files with 32 additions and 91 deletions
@@ -111,7 +111,9 @@ class OllamaGenerator:
            collection_query = collection.query(query_embeddings=query_embed, n_results=100)
            print("Showing pertinent info from drafts used in final edited edition")
            for document in collection_query:
-                print (document)
+                print (document['ids'])
+                print (document['embeddings'])
+                print (document['documents'])
            pertinent_draft_info = '\n\n'.join(collection.query(query_embeddings=query_embed, n_results=100)['documents'][0])
            prompt_enhanced = f"{prompt} - Generate the final document using this information from the drafts: {pertinent_draft_info} - ONLY OUTPUT THE MARKDOWN"
            print("Generating final document")