vertex

langchain_pipeline.py

@@ -6,11 +6,11 @@ from langchain_openai import OpenAIEmbeddings
 from langchain_anthropic import ChatAnthropic
 from langchain_google_genai import ChatGoogleGenerativeAI
 import base64
-from vertexai.generative_models import GenerativeModel, Part, FinishReason
-import vertexai.preview.generative_models as generative_models
+from google.cloud.aiplatform.generative_models import GenerativeModel, Part, FinishReason
+from google.cloud import aiplatform as generative_models
 
 def generate(document_parts, prompt_text):
-    vertexai.init(project="akroda", location="us-central1")
+    aiplatform.init(project="akroda", location="us-central1")
     model = GenerativeModel("gemini-1.5-pro-001")
     responses = model.generate_content(
         document_parts,