import names

yil532 · yil532 · commit 6fda536169ee · 2023-05-04T16:04:11.000-07:00
diff --git a/generativeai/chat.py b/generativeai/chat.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_chat]
-from vertex_ai.preview.language_models import ChatModel, InputOutputTextPair
+from vertexai.preview.language_models import ChatModel, InputOutputTextPair
 
 
 def science_tutoring(temperature=.2):
diff --git a/generativeai/chat_test.py b/generativeai/chat_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import ChatModel, InputcontentputTextPair
+from vertexai.preview.language_models import ChatModel, InputcontentputTextPair
 
 import chat
 
diff --git a/generativeai/classify_news_items.py b/generativeai/classify_news_items.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_classify_news_items]
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 
 def classify_news_items(temperature=0):
diff --git a/generativeai/classify_news_items_test.py b/generativeai/classify_news_items_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 import classify_news_items
 
diff --git a/generativeai/embedding.py b/generativeai/embedding.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_embedding]
-from vertex_ai.preview.language_models import TextEmbeddingModel
+from vertexai.preview.language_models import TextEmbeddingModel
 
 
 def text_embedding():
diff --git a/generativeai/embedding_test.py b/generativeai/embedding_test.py
@@ -15,7 +15,7 @@
 import pytest
 import numpy as np
 
-from vertex_ai.preview.language_models import TextEmbeddingModel
+from vertexai.preview.language_models import TextEmbeddingModel
 
 import embedding
 
diff --git a/generativeai/extraction.py b/generativeai/extraction.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_extraction]
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 
 def extractive_question_answering(temperature=0):
diff --git a/generativeai/extraction_test.py b/generativeai/extraction_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 import extraction
 
diff --git a/generativeai/ideation.py b/generativeai/ideation.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_ideation]
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 
 def interview(temperature=.2):
diff --git a/generativeai/ideation_test.py b/generativeai/ideation_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 import ideation
 
diff --git a/generativeai/list_tuned_models.py b/generativeai/list_tuned_models.py
@@ -16,13 +16,13 @@
 from typing import Union
 import pandas as pd
 
-from vertex_ai.preview.language_models import TextGenerationModel, TextEmbeddingModel
-from google.cloud import aiplatform
+from vertexai.preview.language_models import TextGenerationModel, TextEmbeddingModel
+from google.cloud import vertexai
 
 
 def list_tuned_models(project_id, location):
     """List tuned models."""
-    aiplatform.init(project=project_id, location=location)
+    vertexai.init(project=project_id, location=location)
     model = TextGenerationModel.from_pretrained("text-bison@001")
     tuned_model_names = model.list_tuned_model_names()
     print(tuned_model_names)
diff --git a/generativeai/list_tuned_models_test.py b/generativeai/list_tuned_models_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel, TextEmbeddingModel
+from vertexai.preview.language_models import TextGenerationModel, TextEmbeddingModel
 
 import list_tuned_models
 
diff --git a/generativeai/sentiment_analysis.py b/generativeai/sentiment_analysis.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_sentiment_analysis]
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 
 def sentiment_analysis(temperature=0):
diff --git a/generativeai/sentiment_analysis_test.py b/generativeai/sentiment_analysis_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 import sentiment_analysis
 
diff --git a/generativeai/summarization.py b/generativeai/summarization.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # [START generativeai_sdk_summarization]
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 
 def text_summarization(temperature=.2):
diff --git a/generativeai/summarization_test.py b/generativeai/summarization_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextGenerationModel
+from vertexai.preview.language_models import TextGenerationModel
 
 import summarization
 
diff --git a/generativeai/tuning.py b/generativeai/tuning.py
@@ -16,8 +16,8 @@
 from typing import Union
 import pandas as pd
 
-from vertex_ai.preview.language_models import TextGenerationModel, TextEmbeddingModel
-from google.cloud import aiplatform
+from vertexai.preview.language_models import TextGenerationModel, TextEmbeddingModel
+from google.cloud import vertexai
 
 
 def tuning(
@@ -41,12 +41,12 @@ def tuning(
   with rows for each training example.
 
   Args:
-    project_id: GCP Project ID, used to initialize aiplatform
-    location: GCP Region, used to initialize aiplatform
+    project_id: GCP Project ID, used to initialize vertexai
+    location: GCP Region, used to initialize vertexai
     training_data: GCS URI of training file or pandas dataframe of training data
     train_steps: Number of training steps to use when tuning the model.
   """
-  aiplatform.init(project=project_id, location=location)
+  vertexai.init(project=project_id, location=location)
   model = TextGenerationModel.from_pretrained("text-bison@001")
 
   model.tune_model(
diff --git a/generativeai/tuning_test.py b/generativeai/tuning_test.py
@@ -14,7 +14,7 @@
 
 import pytest
 
-from vertex_ai.preview.language_models import TextEmbeddingModel
+from vertexai.preview.language_models import TextEmbeddingModel
 
 import tuning
 
