minor fix for env variable

bilge-ince · Jul 31, 2024 · a5aa9f7 · a5aa9f7
1 parent 73b96dc
commit a5aa9f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/embedding.py b/embedding.py
@@ -11,7 +11,7 @@ def generate_embeddings():
                         'documents_embeddings',
                         'public',
                         'id',
-                        '{os.getenv("EMBEDDING_MODEL")}',
+                        '{os.getenv("AIDB_MODEL_NAME")}',
                         'text',
                         'documents',
                         ARRAY['id', 'doc_fragment'],