Final updates to get working, tested locally

Dockerfile

@@ -34,8 +34,11 @@ EXPOSE 8501
 # The HEALTHCHECK instruction tells Docker how to test a container to check that it is still working. Your container needs to listen to Streamlit’s (default) port 8501:
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
+# Update working directory to be consistent with where Start.py is
+WORKDIR /usr/src/app/scripts
+
 # An ENTRYPOINT allows you to configure a container that will run as an executable. Here, it also contains the entire streamlit run command for your app, so you don’t have to call it from the command line
-ENTRYPOINT ["streamlit", "run", "./scripts/Start.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "Start.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
 # Execute with:
 # docker build -t <image_name> .    

scripts/queries.py

@@ -74,7 +74,10 @@ class QA_Model:
             logging.info('Chat vectorstore: '+str(self.vectorstore))
 
             # Test query
-            test_query = self.vectorstore.similarity_search(TEST_QUERY_PROMPT)
+            try:
+                test_query = self.vectorstore.similarity_search(TEST_QUERY_PROMPT)
+            except:
+                raise Exception("Pinecone vector database is not configured properly. Test query failed. Likely the index does not exist.")
             logging.info('Test query: '+str(test_query))
             if not test_query:
                 raise ValueError("Pinecone vector database is not configured properly. Test query failed.")
@@ -94,7 +97,10 @@ class QA_Model:
             logging.info('Chat vectorstore: '+str(self.vectorstore))
 
             # Test query
-            test_query = self.vectorstore.similarity_search(TEST_QUERY_PROMPT)
+            try:
+                test_query = self.vectorstore.similarity_search(TEST_QUERY_PROMPT)
+            except:
+                raise Exception("Chroma vector database is not configured properly. Test query failed. Likely the index does not exist.")
             logging.info('Test query: '+str(test_query))
             if not test_query:
                 raise ValueError("Chroma vector database is not configured properly. Test query failed.")
@@ -110,7 +116,10 @@ class QA_Model:
             logging.info('Chat query model:'+str(query_model))
 
              # Test query
-            test_query = self.vectorstore.search(TEST_QUERY_PROMPT)
+            try:
+                test_query = self.vectorstore.search(TEST_QUERY_PROMPT)
+            except:
+                raise Exception("RAGatouille vector database is not configured properly.")
             logging.info('Test query: '+str(test_query))
             if not test_query:
                 raise ValueError("Chroma vector database is not configured properly. Test query failed.")