Update app.py

app.py

@@ -22,16 +22,16 @@ auth = (user,password)
 
 client = openai.OpenAI(api_key=openai_api_key)
 
-os_client = OpenSearch(
-    hosts = [{'host': host, 'port': port}],
-    http_auth = auth,
-    use_ssl = True,
-    verify_certs = False
-)
-indices = os_client.cat.indices(format="json")
-list_of_indeces = []
-for index in indices:
-    list_of_indeces.append(index['index'])
+# os_client = OpenSearch(
+#     hosts = [{'host': host, 'port': port}],
+#     http_auth = auth,
+#     use_ssl = True,
+#     verify_certs = False
+# )
+# indices = os_client.cat.indices(format="json")
+# list_of_indeces = []
+# for index in indices:
+#     list_of_indeces.append(index['index'])
 
 def rag_app(user_input: str) -> str:
     gr_client = Client("anasmkh/QdrantVectorStore_Llamaindex")
@@ -49,12 +49,12 @@ def query_generator(user_input:str) -> str:
     completion = client.chat.completions.create(
     model=response.fine_tuned_model,
     messages=[
-    {"role": "system", "content": f"""You are a highly skilled assistant trained to translate natural language requests into accurate and efficient OpenSearch JSON queries. Follow a clear, step-by-step process to:
+    {"role": "system", "content": """You are a highly skilled assistant trained to translate natural language requests into accurate and efficient OpenSearch JSON queries. Follow a clear, step-by-step process to:
 
 Understand the user's request by breaking it down into components such as filters, aggregations, sort criteria, and specific fields.
 Pay special attention to fields with unique names, such as Date (instead of timestamp) and Stream (instead of type), and ensure they are used correctly in the query.
 Recognize that the user operates within two main opcos: Zambia and Eswatini, each containing ptm_counters, ptm_events, and multiple streams like ers-daily.
-Generate a valid JSON query strictly based on the provided indices, ensuring it aligns with the user's prompt. The available indices are: {list_of_indeces}.
+Generate a valid JSON query strictly based on the provided indices, ensuring it aligns with the user's prompt.
 When generating the query:
 Be precise and include only necessary fields and components relevant to the request.
 Assume any unspecified context or detail needs clarification and provide a clear explanation of your assumptions if needed.
@@ -72,27 +72,27 @@ query_tool = FunctionTool.from_defaults(fn=query_generator)
 llm = OpenAI(model="gpt-3.5-turbo", temperature=0)
 agent = ReActAgent.from_tools([query_tool,rag_tool], llm=llm, verbose=True)
 
-def implement_query(generated_query):
-    try:
-        st.write("Raw Query:", generated_query)
+# def implement_query(generated_query):
+#     try:
+#         st.write("Raw Query:", generated_query)
 
-        if isinstance(generated_query, str):
-            generated_query = generated_query.replace("'", '"')
-            query = json.loads(generated_query)
-        else:
-            query = generated_query
+#         if isinstance(generated_query, str):
+#             generated_query = generated_query.replace("'", '"')
+#             query = json.loads(generated_query)
+#         else:
+#             query = generated_query
 
-        st.write("Validated Query:", query)
+#         st.write("Validated Query:", query)
 
 
-        response = os_client.search(body=query)
-        return response
-    except json.JSONDecodeError as e:
-        st.error("Error: The generated query is not valid JSON.")
-        st.write(f"JSONDecodeError Details: {e}")
-    except Exception as e:
-        st.error(f"Error executing OpenSearch query: {e}")
-        st.write(f"Exception Details: {e}")
+#         response = os_client.search(body=query)
+#         return response
+#     except json.JSONDecodeError as e:
+#         st.error("Error: The generated query is not valid JSON.")
+#         st.write(f"JSONDecodeError Details: {e}")
+#     except Exception as e:
+#         st.error(f"Error executing OpenSearch query: {e}")
+#         st.write(f"Exception Details: {e}")
 
 
 st.subheader('OpenSearch Assistant')
@@ -111,9 +111,9 @@ if st.button("Submit"):
                     st.write('Used Tool: ',source.tool_name)
                     if source.tool_name =='query_generator':
                         st.write(source.raw_output.content)
-                        os_response = implement_query(source.raw_output.content)
-                        st.subheader('OS Response')
-                        st.write(os_response)
+                        # os_response = implement_query(source.raw_output.content)
+                        # st.subheader('OS Response')
+                        # st.write(os_response)
                     else:
                         st.write(source.raw_output[0][0][1])
             except Exception as e: