Merge branch 'develop_qatalive' of https://github.com/vishalgtingre/a…

…zure-search-openai-demo into develop_qatalive
vishalgtingre · Nov 5, 2023 · d226448 · d226448
2 parents a656624 + 42a8d89
commit d226448
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/app/backend/approaches/chatreadretrieveread.py b/app/backend/approaches/chatreadretrieveread.py
@@ -298,11 +298,13 @@ async def run_until_final_call(
             functions=self.functions,
             function_call="auto",
         )
-
-        query_text = self.get_search_query(chat_completion, original_user_query)
-
         response_message = chat_completion["choices"][0]["message"]
-        print (chat_completion)
+        print ("response_message")
+        print (response_message)
+        query_text = self.get_search_query(chat_completion, original_user_query)
+        print ("chat_completion")
+        print(chat_completion)
+        # STEP 2: Retrieve relevant documents from the search index with the GPT optimized query
 
         query_text = None
         if response_message.get("function_call"):
@@ -554,6 +556,8 @@ def get_search_query(self, chat_completion: dict[str, Any], user_query: str):
                 arg = json.loads(function_call["arguments"])
                 search_query = arg.get("search_query", self.NO_RESPONSE)
                 if search_query != self.NO_RESPONSE:
+                    print("the search_query")
+                    print(search_query)
                     return search_query
         elif query_text := response_message.get("content"):
             if query_text.strip() != self.NO_RESPONSE: