From a5aa9f7a447b0814f2303226e32d69bdbee7f3f4 Mon Sep 17 00:00:00 2001
From: bilge-ince <bilge.ince@enterprisedb.com>
Date: Wed, 31 Jul 2024 14:45:44 +0100
Subject: [PATCH] minor fix for env variable

---
 embedding.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/embedding.py b/embedding.py
index 01ecdee..3753351 100644
--- 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'],