test(test_amazing_vertex_completion.py): fix test

This commit is contained in:
Krrish Dholakia 2024-02-26 10:42:05 -08:00
parent 0f3630a5ac
commit adb4443ea6
2 changed files with 2 additions and 1 deletions

View file

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

View file

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