phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_amazing_vertex_completion.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-26 10:42:05 -08:00
parent f460da4770
commit 28f4b5809c
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/main.py
View file

@ -10,7 +10,6 @@
import os, openai, sys, json, inspect, uuid, datetime, threading import os, openai, sys, json, inspect, uuid, datetime, threading
from typing import Any, Literal, Union from typing import Any, Literal, Union
from functools import partial from functools import partial
import dotenv, traceback, random, asyncio, time, contextvars import dotenv, traceback, random, asyncio, time, contextvars
from copy import deepcopy from copy import deepcopy
import httpx import httpx

2
litellm/tests/test_amazing_vertex_completion.py
View file

@ -250,6 +250,8 @@ async def test_async_vertexai_streaming_response():
"code-gecko@latest", "code-gecko@latest",
"code-bison@001", "code-bison@001",
"text-bison@001", "text-bison@001",
"gemini-1.5-pro",
"gemini-1.5-pro-vision",
]: ]:
# our account does not have access to this model # our account does not have access to this model
continue continue