tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-24 08:30:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

vertex testing

Browse Source
This commit is contained in:
Ishaan Jaff
2024-06-27 20:20:46 -07:00
parent 76490690a1
commit c27bfd76b6
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_amazing_vertex_completion.py
+13 -5
View File
@@ -329,11 +329,14 @@ def test_vertex_ai():
"code-gecko@001",
"code-gecko@002",
"code-gecko@latest",
"codechat-bison@latest",
"code-bison@001",
"text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
]:
] or (
"gecko" in model or "32k" in model or "ultra" in model or "002" in model
):
# our account does not have access to this model
continue
print("making request", model)
@@ -381,12 +384,15 @@ def test_vertex_ai_stream():
"code-gecko@001",
"code-gecko@002",
"code-gecko@latest",
"codechat-bison@latest",
"code-bison@001",
"text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
]:
# ouraccount does not have access to this model
] or (
"gecko" in model or "32k" in model or "ultra" in model or "002" in model
):
# our account does not have access to this model
continue
print("making request", model)
response = completion(
@@ -433,11 +439,12 @@ async def test_async_vertexai_response():
"code-gecko@001",
"code-gecko@002",
"code-gecko@latest",
"codechat-bison@latest",
"code-bison@001",
"text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
]:
] or ("gecko" in model or "32k" in model or "ultra" in model or "002" in model):
# our account does not have access to this model
continue
try:
@@ -479,11 +486,12 @@ async def test_async_vertexai_streaming_response():
"code-gecko@001",
"code-gecko@002",
"code-gecko@latest",
"codechat-bison@latest",
"code-bison@001",
"text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-preview-0215",
]:
] or ("gecko" in model or "32k" in model or "ultra" in model or "002" in model):
# our account does not have access to this model
continue
try: