mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 20:48:32 +00:00
fix: move to new vertex credentials
This commit is contained in:
@@ -3,7 +3,7 @@ const { VertexAI, RequestOptions } = require('@google-cloud/vertexai');
|
||||
|
||||
|
||||
const vertexAI = new VertexAI({
|
||||
project: 'pathrise-convert-1606954137718',
|
||||
project: 'litellm-ci-cd',
|
||||
location: 'us-central1',
|
||||
apiEndpoint: "127.0.0.1:4000/vertex-ai"
|
||||
});
|
||||
|
||||
@@ -61,7 +61,7 @@ beforeAll(() => {
|
||||
describe('Vertex AI Tests', () => {
|
||||
test('should successfully generate content from Vertex AI', async () => {
|
||||
const vertexAI = new VertexAI({
|
||||
project: 'pathrise-convert-1606954137718',
|
||||
project: 'litellm-ci-cd',
|
||||
location: 'us-central1',
|
||||
apiEndpoint: "localhost:4000/vertex-ai"
|
||||
});
|
||||
@@ -100,7 +100,7 @@ describe('Vertex AI Tests', () => {
|
||||
|
||||
|
||||
test('should successfully generate non-streaming content from Vertex AI', async () => {
|
||||
const vertexAI = new VertexAI({project: 'pathrise-convert-1606954137718', location: 'us-central1', apiEndpoint: "localhost:4000/vertex-ai"});
|
||||
const vertexAI = new VertexAI({project: 'litellm-ci-cd', location: 'us-central1', apiEndpoint: "localhost:4000/vertex-ai"});
|
||||
const customHeaders = new Headers({"x-litellm-api-key": "sk-1234"});
|
||||
const requestOptions = {customHeaders: customHeaders};
|
||||
const generativeModel = vertexAI.getGenerativeModel({model: 'gemini-2.5-flash-lite'}, requestOptions);
|
||||
|
||||
@@ -98,7 +98,7 @@ async def test_basic_vertex_ai_pass_through_with_spendlog():
|
||||
load_vertex_ai_credentials()
|
||||
|
||||
vertexai.init(
|
||||
project="pathrise-convert-1606954137718",
|
||||
project="litellm-ci-cd",
|
||||
location="us-central1",
|
||||
api_endpoint=f"{LITE_LLM_ENDPOINT}/vertex_ai",
|
||||
api_transport="rest",
|
||||
@@ -138,7 +138,7 @@ async def test_basic_vertex_ai_pass_through_streaming_with_spendlog():
|
||||
load_vertex_ai_credentials()
|
||||
|
||||
vertexai.init(
|
||||
project="pathrise-convert-1606954137718",
|
||||
project="litellm-ci-cd",
|
||||
location="us-central1",
|
||||
api_endpoint=f"{LITE_LLM_ENDPOINT}/vertex_ai",
|
||||
api_transport="rest",
|
||||
@@ -177,7 +177,7 @@ async def test_vertex_ai_pass_through_endpoint_context_caching():
|
||||
# load_vertex_ai_credentials()
|
||||
|
||||
vertexai.init(
|
||||
project="pathrise-convert-1606954137718",
|
||||
project="litellm-ci-cd",
|
||||
location="us-central1",
|
||||
api_endpoint=f"{LITE_LLM_ENDPOINT}/vertex_ai",
|
||||
api_transport="rest",
|
||||
|
||||
@@ -70,7 +70,7 @@ jest.retryTimes(3);
|
||||
describe('Vertex AI Tests', () => {
|
||||
test('should successfully generate non-streaming content with tags', async () => {
|
||||
const vertexAI = new VertexAI({
|
||||
project: 'pathrise-convert-1606954137718',
|
||||
project: 'litellm-ci-cd',
|
||||
location: 'us-central1',
|
||||
apiEndpoint: "127.0.0.1:4000/vertex_ai"
|
||||
});
|
||||
@@ -129,7 +129,7 @@ describe('Vertex AI Tests', () => {
|
||||
|
||||
test('should successfully generate streaming content with tags', async () => {
|
||||
const vertexAI = new VertexAI({
|
||||
project: 'pathrise-convert-1606954137718',
|
||||
project: 'litellm-ci-cd',
|
||||
location: 'us-central1',
|
||||
apiEndpoint: "127.0.0.1:4000/vertex_ai"
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user