Update app.py

app.py

@@ -194,6 +194,25 @@ def rag_workflow(query):
     print(f"References for the query:\n{references}\n")
     prompt = f"You are an intelligent AI assistant who is very good in giving answers for anything asked or instructed by the user. Provide a clear and concise answer based only on the pieces of retrieved context. You must follow this very strictly, do not use anything else other than the retrieved context. If no related Information is found from the context, reply that you do not know. \n\nContext:\n{context}\n\nQuery: {query}"
 
+
+def rag_workflow(query):
+    docs = query_chroma(vectorstore, query, k=10)
+    context = "\n\n".join([doc for doc, _ in docs])
+    references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(docs)])
+    print(f"Context for the query:\n{context}\n")
+    print(f"References for the query:\n{references}\n")
+    
+    prompt = f"""You are an intelligent Coding AI assistant who is very good in giving answers for anything asked or instructed by the user. 
+                    The retrieved context contains source code and documenation of an api library. Provide a clear and consice answer based on the information in the retrieved context.
+                    If no related Information is found from the context, reply that you do not know.
+
+            Context:
+            {context}
+
+            Query: 
+            {query}
+    """
+
     
     response = llm.invoke(prompt)
     return response.content, references