From d0d8ba46c570726869a7dd376834076e0f794863 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 4 Dec 2023 11:35:36 -0800 Subject: [PATCH] test(test_proxy_server_keys.py): fix relative import --- litellm/tests/test_proxy_server_keys.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_proxy_server_keys.py b/litellm/tests/test_proxy_server_keys.py index b0e7d33bd2..806b5f43eb 100644 --- a/litellm/tests/test_proxy_server_keys.py +++ b/litellm/tests/test_proxy_server_keys.py @@ -24,8 +24,8 @@ logging.basicConfig( from fastapi.testclient import TestClient from fastapi import FastAPI from litellm.proxy.proxy_server import router, save_worker_config, startup_event # Replace with the actual module where your FastAPI router is defined -cwd = os.getcwd() -config_fp = f"{cwd}/test_config.yaml" +filepath = os.path.dirname(os.path.abspath(__file__)) +config_fp = f"{filepath}/test_config.yaml" save_worker_config(config=config_fp, model=None, alias=None, api_base=None, api_version=None, debug=False, temperature=None, max_tokens=None, request_timeout=600, max_budget=None, telemetry=False, drop_params=True, add_function_to_prompt=False, headers=None, save=False, use_queue=False) app = FastAPI() app.include_router(router) # Include your router in the test app