ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix optional params

Browse Source
This commit is contained in:
Rena Lu 2024-02-13 20:50:26 +00:00
parent 3fc0d8fda9
commit ce7ce3b719
2 changed files with 1170 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1168
litellm/llms/.ipynb_checkpoints/vertex_ai-checkpoint.py Normal file
View File

File diff suppressed because it is too large Load Diff

10
litellm/llms/vertex_ai.py
View File

@ -560,8 +560,7 @@ def completion(
)
response = llm_model.predict(
endpoint=endpoint_path,
instances=instances,
**optional_params,
instances=instances
).predictions
completion_response = response[0]
@ -590,8 +589,7 @@ def completion(
f"llm_model.predict(instances={instances}, **{optional_params})\n"
)
response = llm_model.predict(
instances=instances,
**optional_params,
instances=instances
).predictions
completion_response = response[0]
@ -791,7 +789,6 @@ async def async_completion(
response_obj = await llm_model.predict(
endpoint=endpoint_path,
instances=instances,
**optional_params,
)
response = response_obj.predictions
completion_response = response[0]
@ -807,7 +804,6 @@ async def async_completion(
)
response_obj = await llm_model.predict_async(
instances=instances,
**optional_params,
)
response = response_obj.predictions
@ -980,7 +976,6 @@ async def async_streaming(
response_obj = await llm_model.predict(
endpoint=endpoint_path,
instances=instances,
**optional_params,
)
response = response_obj.predictions
@ -999,7 +994,6 @@ async def async_streaming(
request_str += f"llm_model.predict_async(instances={instances}, **{optional_params})\n"
response_obj = await llm_model.predict_async(
instances=instances,
**optional_params,
)
response = response_obj.predictions
completion_response = response[0]