Fixes #9, new ENV var for schema

ai-cfia · Feb 8, 2024 · b1a769e · b1a769e
1 parent 4c02761
commit b1a769e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/DEVELOPER.md b/DEVELOPER.md
@@ -32,6 +32,8 @@
 
   * DB_SERVER_CONTAINER_NAME: The name of your database server container (ex: louis-db-server).
 
+  * AILAB_SCHEMA_VERSION: The version of the schema you want to use.
+
 * Run database locally (see bin/postgres.sh)
 * Restore latest schema dump
 

diff --git a/bin/generate_qna.py b/bin/generate_qna.py
@@ -24,9 +24,9 @@
 
 # Constants
 TEST_VERSION = date.today()
-REQUIRED_QUESTIONS = 1
+REQUIRED_QUESTIONS = 3
 CHARACTER_LIMIT = 14383
-DEFAULT_STORAGE_PATH = "/home/vscode/finesse-data-2/qna"
+DEFAULT_STORAGE_PATH = "../qna-test"
 SYSTEM_PROMPT_FILENAME = "qna_system_prompt.txt"
 USER_PROMPT_FILENAME = "qna_user_prompt.txt"
 
@@ -60,11 +60,8 @@ def generate_question(
 
     with project_db.cursor() as cursor:  # Open the cursor here
         for i in range(REQUIRED_QUESTIONS):
-            # Access the LOUIS_SCHEMA environment variable
-            louis_schema = os.getenv("LOUIS_SCHEMA")
-
-            # Extract version part from the schema name
-            schema_version = re.search(r"(\d+\.\d+\.\d+)", louis_schema).group(1)
+            # Access the AILAB_SCHEMA_VERSION environment variable
+            schema_version = os.getenv("AILAB_SCHEMA_VERSION")
 
             random_chunk = get_random_chunk(cursor, schema_version)
             if not random_chunk: