Update openai.py generator

Added `user_question` to assistant prompt for completeness
defog-ai · Dec 2, 2023 · a71bf14 · a71bf14
1 parent 024ff01
commit a71bf14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/query_generators/openai.py b/query_generators/openai.py
@@ -126,6 +126,10 @@ def generate_query(self, question: str, instructions: str) -> dict:
                 table_metadata_string=prune_metadata_str(question, self.db_name),
             )
 
+            assistant_prompt = assistant_prompt.format(
+                user_question=question,
+            )
+
             messages = []
             messages.append({"role": "system", "content": sys_prompt})
             messages.append({"role": "user", "content": user_prompt})