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

mark as async

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 16:40:41 -07:00
parent 414d2dcb52
commit f43060e8df
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/pass_through_tests/test_vertex_ai.py
View file

@ -9,8 +9,7 @@ from vertexai.preview.generative_models import GenerativeModel
import tempfile import tempfile
import json import json
import os import os
from google.oauth2 import service_account import pytest
import google.auth.transport.requests
# Path to your service account JSON file # Path to your service account JSON file
@ -77,6 +76,7 @@ async def call_spend_logs_endpoint():
LITE_LLM_ENDPOINT = "http://localhost:4000" LITE_LLM_ENDPOINT = "http://localhost:4000"
@pytest.mark.asyncio()
async def test_basic_vertex_ai_pass_through_with_spendlog(): async def test_basic_vertex_ai_pass_through_with_spendlog():
vertexai.init( vertexai.init(