Update

app.py

@@ -32,7 +32,7 @@ if prompt := st.chat_input("What's up?"):
         web_query = llm_client.web_query_generator(query=prompt, history=st.session_state.messages)
         print(f"Web Query: {web_query}")
         
-        web_result = web_searcher.query_bing(query=prompt) if web_query != 'None' else "None"
+        web_result = web_searcher.query_bing(query=web_query) if web_query != 'None' else "None"
         # web_result = web_searcher.query_web_llm(query=prompt, num_web_results=10) if web_query != 'None' else "None"
         print(f"Web Result: {web_result}")
         

components.py

@@ -25,7 +25,7 @@ class LLMClient():
         messages += [{"role": h["role"], "content": h["content"]} for h in history[:-1]]
         
         # messages += [{"role": 'user', "content": f"Given the context, generate a proper web search query. If you know the answer already, return 'None' directly."}]
-        messages += [{"role": 'user', "content": f"Given the context, generate a proper web search query."}]
+        messages += [{"role": 'user', "content": f"Given the context, generate a proper web search query. Return 'None' if you can't generate a query based on the current context."}]
         
         response = self.llm_client.chat.completions.create(
             model = self.model_name,