Merge pull request #97 from vanderbilt-data-science/return-db-texts

free_speech_app/FreeSpeechPromptsResponses.py

@@ -55,7 +55,10 @@ def query_retriever(db, query, num_results = 3):
     retriever = db.as_retriever(search_kwargs={"k": num_results})
     docs = retriever.get_relevant_documents(query)
 
-    return docs
+    docs_as_text = [doc.page_content for doc in docs]
+    docs_as_one_string = ' '.join(str(i)+ "\n" for i in docs_as_text)
+
+    return docs_as_one_string
 
 # %% ../nbs/free-speech-prompts.ipynb 10
 def generate_custom_prompt(original_post, principles=None, writing_style=None, word_limit=None):