From 5a3dbc15d031b77d0ad0dc3af4fbaf24ab8d27b4 Mon Sep 17 00:00:00 2001 From: pm3310 Date: Mon, 19 Feb 2024 21:19:01 +0000 Subject: [PATCH] Add env vars for OpenAI models --- sagify/commands/llm.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sagify/commands/llm.py b/sagify/commands/llm.py index 1930313..476716e 100644 --- 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'),