feat: improve question generation

BBVA · Nov 21, 2024 · 87d6e25 · 87d6e25
1 parent 6e3d051
commit 87d6e25
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/rankwise/generate/data.py b/rankwise/generate/data.py
@@ -12,4 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-DEFAULT_QUESTION_PROMPT = "You are a search engine user. Your task is to setup search terms or phrases for finding the given context document. The search terms should be diverse in nature across the document. Do not include any quote or format the terms, provide them in plain format. Restrict the terms to the context information provided."
+DEFAULT_QUESTION_PROMPT = (
+    "You are a search engine user. Your task is to setup search terms or phrases for finding the "
+    "given context document. The search terms should be diverse in nature across the document. Do "
+    "not include any quote or format the terms, provide them in plain format. Restrict the terms "
+    "to the context information provided."
+)
diff --git a/rankwise/generate/io.py b/rankwise/generate/io.py
@@ -25,4 +25,4 @@ def generate_dataset(model, contents, queries_count, question_gen_query):
         num_questions_per_chunk=queries_count,
         question_gen_query=question_gen_query,
     )
-    return dataset_generator.generate_dataset_from_nodes()
+    return dataset_generator.generate_questions_from_nodes()