Update functions.py

functions.py

@@ -177,8 +177,12 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type='stuff'):
         
     docs = docsearch.similarity_search_with_score(query, k=3, namespace = f'{title}-earnings')
 
+    print(docs)
+
     if chain_type == 'Normal':
 
+        docs = [d[0] for d in docs]
+
         PROMPT = PromptTemplate(template=template, 
                                 input_variables=["summaries", "question"],
                                 output_parser=output_parser)