added queryHelperCot

app.py

@@ -15,6 +15,7 @@ from constants import *
 from utils import *
 from gptManager import ChatgptManager
 from queryHelperManager import QueryHelper
+from queryHelperManagerCoT import QueryHelperChainOfThought
 
 
 pd.set_option('display.max_columns', None)
@@ -45,8 +46,8 @@ queryHelper = QueryHelper(gptInstance=gptInstance,
 
 
 openAIClient2 = OpenAI(api_key=OPENAI_API_KEY)
-gptInstanceFineTuned = ChatgptManager(openAIClient2, model=GPT_MODEL_FINE_TUNED)
-queryHelperFineTuned = QueryHelper(gptInstance=gptInstanceFineTuned,
+gptInstance2 = ChatgptManager(openAIClient2, model=GPT_MODEL)
+queryHelperCot = QueryHelperChainOfThought(gptInstance=gptInstance2,
                           schemaName=SCHEMA_NAME,platform=PLATFORM, 
                           metadataLayout=metadataLayout, 
                           sampleDataRows=SAMPLE_ROW_MAX, 
@@ -83,20 +84,20 @@ def respond(message, chatHistory):
   return "", chatHistory
 
 # Function to save history of chat
-def respondFineTuned(message, chatHistory):
+def respondCoT(message, chatHistory):
   """gpt response handler for gradio ui"""
   global queryHelper
   try:
       if "modify" in message:
-        botMessage, prospectTablesAndCols  = queryHelperFineTuned.getQueryForUserInput(message, chatHistory)
+        botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInputCoT(message, chatHistory)
       else:
-         botMessage, prospectTablesAndCols  = queryHelperFineTuned.getQueryForUserInput(message, [])
+         botMessage, prospectTablesAndCols  = queryHelperCot.getQueryForUserInput(message, [])
   except Exception as e:
       errorMessage = {"function":"queryHelperFineTuned.getQueryForUserInput","error":str(e), "userInput":message}
       saveLog(errorMessage, 'error')
       raise ValueError(str(e))
   queryGenerated = extractSqlFromGptResponse(botMessage)
-  logMessage = {"userInput":message, "tablesColsSelectedByGpt":str(prospectTablesAndCols) , "queryGenerated":queryGenerated, "completeGptResponse":botMessage, "function":"queryHelperFineTuned.getQueryForUserInput"}
+  logMessage = {"userInput":message, "tablesColsSelectedByGpt":str(prospectTablesAndCols) , "queryGenerated":queryGenerated, "completeGptResponse":botMessage, "function":"queryHelperCot.getQueryForUserInputCoT"}
   saveLog(logMessage)
   chatHistory.append((message, botMessage))
   time.sleep(2)
@@ -225,12 +226,12 @@ with gr.Blocks() as demo:
         clear = gr.ClearButton([msg, chatbot])
         msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-    with gr.Tab("Query Helper Fine Tuned"):
+    with gr.Tab("Query Helper CoT"):
         gr.Markdown("""<h1><center> Query Helper Fine Tuned</center></h1>""")
         chatbot = gr.Chatbot()
         msg = gr.Textbox()
         clear = gr.ClearButton([msg, chatbot])
-        msg.submit(respondFineTuned, [msg, chatbot], [msg, chatbot])
+        msg.submit(respondCoT, [msg, chatbot], [msg, chatbot])
 
     # screen 2 : To run sql query against database
     with gr.Tab("Run Query"):