scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

working code for vertex ai routes

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-03 16:56:33 -07:00
parent 64c008045f
commit 8051781af0
2 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/fine_tuning_apis/vertex_ai.py
View file

@ -266,7 +266,7 @@ class VertexFineTuningAPI(VertexLLM):
}
url = None
if request_route == "tuningJobs":
if request_route == "/tuningJobs":
url = f"https://{vertex_location}-aiplatform.googleapis.com/v1/projects/{vertex_project}/locations/{vertex_location}/tuningJobs"
if self.async_handler is None:

14
litellm/proxy/vertex_ai_endpoints/vertex_endpoints.py
View file

@ -34,14 +34,13 @@ def set_default_vertex_config(config):
if config is None:
return
if not isinstance(config, list):
raise ValueError("invalid files config, expected a list is not a list")
if not isinstance(config, dict):
raise ValueError("invalid config, vertex default config must be a dictionary")
for element in config:
if isinstance(element, dict):
for key, value in element.items():
if isinstance(value, str) and value.startswith("os.environ/"):
element[key] = litellm.get_secret(value)
if isinstance(config, dict):
for key, value in config.items():
if isinstance(value, str) and value.startswith("os.environ/"):
config[key] = litellm.get_secret(value)
default_vertex_config = config
@ -102,7 +101,6 @@ async def execute_post_vertex_ai_request(
async def vertex_create_fine_tuning_job(
request: Request,
fastapi_response: Response,
endpoint_name: str,
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
):
"""