Update app.py

app.py

@@ -67,12 +67,22 @@ def IBM_chat (messages):
     response = response.json()
     return response["choices"][0]["message"]["content"]
 
+def IBM_query (prompt):
+    messages = [{"role": "user", "content": prompt}]
+    return IBM_chat(messages)
+    
 def get_credentials():
 	return {
 		"url" : "https://us-south.ml.cloud.ibm.com",
 		"apikey" : os.getenv("IBM_API_KEY")
 	}
 
+##############################################
+##
+##   Vector DB
+##
+##############################################
+
 from ibm_watsonx_ai.client import APIClient
 from ibm_watsonx_ai.foundation_models.embeddings.sentence_transformer_embeddings import SentenceTransformerEmbeddings
 
@@ -249,7 +259,7 @@ if st.sidebar.button("🔗 Connect with IBM Chat"):
         
         # Get response from IBM
         with st.spinner("Analyzing..."):
-            result = IBM_chat(prompt)
+            result = IBM_query(prompt)
             query = "Find 3 projects that best match these criteria: " + result
             st.session_state["extended_query"] = query
     else: