Add env vars for OpenAI models

Kenza-AI · Feb 19, 2024 · 5a3dbc1 · 5a3dbc1
1 parent a6fd9e5
commit 5a3dbc1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sagify/commands/llm.py b/sagify/commands/llm.py
@@ -609,6 +609,9 @@ def start_local_gateway(image, dockerfile_dir, platform):
     logger.info(ASCII_LOGO)
     environment_vars = {
         'OPENAI_API_KEY': os.environ.get('OPENAI_API_KEY'),
+        'OPENAI_CHAT_COMPLETIONS_MODEL': os.environ.get('OPENAI_CHAT_COMPLETIONS_MODEL'),
+        'OPENAI_EMBEDDINGS_MODEL': os.environ.get('OPENAI_EMBEDDINGS_MODEL'),
+        'OPENAI_IMAGE_CREATION_MODEL': os.environ.get('OPENAI_IMAGE_CREATION_MODEL'),
         'AWS_ACCESS_KEY_ID': os.environ.get('AWS_ACCESS_KEY_ID'),
         'AWS_SECRET_ACCESS_KEY': os.environ.get('AWS_SECRET_ACCESS_KEY'),
         'AWS_REGION_NAME': os.environ.get('AWS_REGION_NAME'),