issue #49 - secrets correction

ai-cfia · Nov 22, 2023 · 963c049 · 963c049
1 parent 4363ef1
commit 963c049
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/pull-request.yml b/.github/workflows/pull-request.yml
@@ -10,7 +10,7 @@ on:
 jobs:
   lint-test:
     # uses: ai-cfia/github-workflows/.github/workflows/workflow-lint-test-node.yml@main
-    uses: ai-cfia/github-workflows/.github/workflows/workflow-lint-test-python.yml@main
+    uses: ai-cfia/github-workflows/.github/workflows/workflow-lint-test-python.yml@issue-34-add-environment-variable-for-ailab-db-lint-test
     secrets: inherit
 
   build:

diff --git a/ailab/db/api/__init__.py b/ailab/db/api/__init__.py
@@ -8,7 +8,7 @@
 dotenv.load_dotenv()
 
 FINESSE_WEIGHTS = os.environ.get("FINESSE_WEIGHTS") \
-    or db.raise_db_error("FINESSE_WEIGHTS is not set")
+    or db.raise_error("FINESSE_WEIGHTS is not set")
 
 
 if FINESSE_WEIGHTS:
@@ -105,4 +105,4 @@ def search_from_text_query(cursor, query):
     else:
         data.update(db_data)
     docs = search(cursor, data['embedding'])
-    return docs
+    return docs
diff --git a/ailab/models/openai.py b/ailab/models/openai.py
@@ -56,4 +56,4 @@ def get_chat_answer(system_prompt, user_prompt, max_token):
             {"role": "user", "content": (user_prompt)}
         ]
     )
-    return response
+    return response