extracting both text and summary in RAG

karthiksoman · Sep 24, 2024 · fafe989 · fafe989
1 parent 7a79cfd
commit fafe989
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/code/notebook/pinecone_retrieval.ipynb b/code/notebook/pinecone_retrieval.ipynb
@@ -95,6 +95,25 @@
     ")\n"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "562a40c6-0e5a-45eb-b22a-8be3edf1cfdd",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "0\n"
+     ]
+    }
+   ],
+   "source": [
+    "for i in range(1):\n",
+    "    print(i)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 71,

diff --git a/lambda-pipeline/app.py b/lambda-pipeline/app.py
@@ -32,7 +32,7 @@ def get_model_response(input_text, temperature: float = 0.7):
         logging.error(f"Failed to get model response: {str(e)}")
         return None
 
-def get_rag_context(query, top_k=3):
+def get_rag_context(query, top_k=2):
     try:
         embed_model = OpenAIEmbedding(
             model='text-embedding-ada-002',