Upload prompts.py

prompts.py

@@ -0,0 +1,27 @@
+from typing import List
+
+def get_classification_prompt(categories_list: List[str]) -> str:
+    """Generate classification prompt based on the categories list."""
+    categories_str = ', '.join([f"'{category}'" for category in categories_list])
+    return (
+        f"Classify the following query into one of the following categories: {categories_str}. "
+        f"If it doesn't fit into any category, respond with 'None'. "
+        f"Return the classification, do not output absolutely anything else."
+    )
+
+
+def get_query_generation_prompt(query_str: str, num_queries: int) -> str:
+    """Generate query generation prompt based on query string and number of sub-queries."""
+    return (
+        f"You are an expert at distilling a user question into sub-questions that can be used to fully answer the original query. "
+        f"First, identify the key words from the original question below: \n"
+        f"{query_str}"
+        f"Generate {num_queries} sub-queries that cover the different aspects needed to fully address the user's query.\n\n"
+        f"Here is an example: \n"
+        f"Original Question: What does test data mean and what do I need to know about it?\n"
+        f"Output:\n"
+        f"definition of 'test data'\n"
+        f"test data requirements and conditions for a bias audit\n"
+        f"examples of the use of test data in a bias audit\n\n"
+        f"Output the rewritten sub-queries, one on each line, do not output absolutely anything else."
+    )