test(test_openai_endpoints.py): fix test

This commit is contained in:
Krrish Dholakia 2024-02-23 22:57:20 -08:00
parent f7b0ad1e53
commit 20def3805c

View file

@ -97,10 +97,10 @@ async def test_chat_completion_old_key():
""" """
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
try: try:
key = "sk-yNXvlRO4SxIGG0XnRMYxTw" key = "sk-ecMXHujzUtKCvHcwacdaTw"
await chat_completion(session=session, key=key) await chat_completion(session=session, key=key)
except Exception as e: except Exception as e:
key = "sk-2KV0sAElLQqMpLZXdNf3yw" # try diff db key (in case db url is for the other db) key = "sk-ecMXHujzUtKCvHcwacdaTw" # try diff db key (in case db url is for the other db)
await chat_completion(session=session, key=key) await chat_completion(session=session, key=key)