Update app.py

app.py

@@ -137,7 +137,7 @@ def search_and_reconstruct(query, df_chunks):
 
     return reconstructed_results
 
-def call_chat_api(data: ChatRequestClient):
+def call_chat_api(data: ChatRequestClient, k):
     url = "https://agent-builder-api.greensea-b20be511.northeurope.azurecontainerapps.io/chat/"
     # Validate and convert the data to a dictionary
     validated_data = data.dict()
@@ -148,7 +148,7 @@ def call_chat_api(data: ChatRequestClient):
     if response.status_code == 200:
         body = response.json()
         query = body.get("content")
-        final_results = search_and_reconstruct(query, df_chunks)
+        final_results = search_and_reconstruct(query, df_chunks, k)
         return body, final_results # Return the JSON response if successful
     else:
         return "An error occured"  # Return the raw response text if not successful
@@ -170,7 +170,8 @@ persona1SystemMessage = st.sidebar.text_area("Query Designer System Message", va
 
 llm1 = st.sidebar.selectbox("Model Selection", ['GPT-4', 'GPT3.5'], key='persona1_size')
 temp1 = st.sidebar.slider("Temperature", min_value=0.0, max_value=1.0, step=0.1, value=0.6, key='persona1_temp')
-tokens1 = st.sidebar.slider("Tokens", min_value=0, max_value=4000, step=100, value=500, key='persona1_tokens')
+tokens1 = st.sidebar.slider("Tokens", min_value=1, max_value=4000, step=100, value=500, key='persona1_tokens')
+k = st.sidebar.slider("Tokens", min_value=1, max_value=10, step=1, value=3, key='k')
 
 st.sidebar.caption(f"Session ID: {genuuid()}")
 
@@ -223,7 +224,7 @@ else:
             temperature2=0.2
         )
 
-        response, retrival = call_chat_api(data)
+        response, retrival = call_chat_api(data, k)
         agent_message = response.get("content", "No response received from the agent.")
         elapsed_time = response.get("elapsed_time", 0)
         st.session_state.messages.append({"role": "assistant", "content": agent_message})