Skip to content

Commit

Permalink
bug fix
Browse files Browse the repository at this point in the history
gecBurton committed Jan 28, 2025
1 parent ffd242e commit 4c6a094
Showing 3 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions redbox-core/redbox/app.py
Original file line number Diff line number Diff line change
@@ -36,9 +36,9 @@ def __init__(self, debug: bool = False):
self.debug = debug

def _get_runnable(self, state: RedboxState):
azure_endpoint = os.environ["AZURE_OPENAI_ENDPOINT"]
azure_endpoint = os.environ["LITELLM_URL"]
api_key = os.environ["LITELLM_MASTER_KEY"]
logger.info("AZURE_OPENAI_ENDPOINT=%s, LITELLM_MASTER_KEY=%s", azure_endpoint, api_key)
logger.info("LITELLM_URL=%s, LITELLM_MASTER_KEY=%s", azure_endpoint, api_key)

llm = init_chat_model(
azure_endpoint=azure_endpoint,
2 changes: 1 addition & 1 deletion tests/.env.integration
Original file line number Diff line number Diff line change
@@ -38,6 +38,6 @@ EMAIL_FILE_PATH='/app/mail'
FILE_EXPIRY_IN_DAYS=30
MAX_SECURITY_CLASSIFICATION=OFFICIAL_SENSITIVE

AZURE_OPENAI_ENDPOINT=http://litellm:4000
LITELLM_URL=http://litellm:4000
LITELLM_MASTER_KEY=sk-1234
POSTGRES_URL=postgresql://redbox-core:insecure@db:5432/litellm
2 changes: 1 addition & 1 deletion tests/.env.test
Original file line number Diff line number Diff line change
@@ -53,6 +53,6 @@ USER_EMAIL=
# OFFICIAL, OFFICIAL_SENSITIVE, SECRET or TOP_SECRET
MAX_SECURITY_CLASSIFICATION=OFFICIAL_SENSITIVE

AZURE_OPENAI_ENDPOINT=http://0.0.0.0:4000
LITELLM_URL=http://0.0.0.0:4000
LITELLM_MASTER_KEY=sk-1234
POSTGRES_URL=postgresql://redbox-core:insecure@localhost:5432/litellm

0 comments on commit 4c6a094

Please sign in to comment.