# import sys, os, time
# import traceback
# from dotenv import load_dotenv

# load_dotenv()
# import os, io

# # this file is to test litellm/proxy

# sys.path.insert(
#     0, os.path.abspath("../..")
# )  # Adds the parent directory to the system path
# import pytest, logging, requests
# import litellm
# from litellm import embedding, completion, completion_cost, Timeout
# from litellm import RateLimitError


# def test_add_new_key():
#     max_retries = 3
#     retry_delay = 1  # seconds

#     for retry in range(max_retries + 1):
#         try:
#             # Your test data
#             test_data = {
#                 "models": ["gpt-3.5-turbo", "gpt-4", "claude-2", "azure-model"],
#                 "aliases": {"mistral-7b": "gpt-3.5-turbo"},
#                 "duration": "20m",
#             }
#             print("testing proxy server")

#             # Your bearer token
#             token = os.getenv("PROXY_MASTER_KEY")
#             headers = {"Authorization": f"Bearer {token}"}

#             staging_endpoint = "https://litellm-litellm-pr-1376.up.railway.app"
#             main_endpoint = "https://litellm-staging.up.railway.app"

#             # Make a request to the staging endpoint
#             response = requests.post(
#                 main_endpoint + "/key/generate", json=test_data, headers=headers
#             )

#             print(f"response: {response.text}")

#             if response.status_code == 200:
#                 result = response.json()
#                 break  # Successful response, exit the loop
#             elif response.status_code == 503 and retry < max_retries:
#                 print(
#                     f"Retrying in {retry_delay} seconds... (Retry {retry + 1}/{max_retries})"
#                 )
#                 time.sleep(retry_delay)
#             else:
#                 assert False, f"Unexpected response status code: {response.status_code}"

#         except Exception as e:
#             print(traceback.format_exc())
#             pytest.fail(f"An error occurred {e}")


# test_add_new_key()