bugresolve

langchain_pipeline.py

@@ -6,16 +6,21 @@ from langchain_openai import OpenAIEmbeddings
 from langchain_anthropic import ChatAnthropic
 from langchain_google_genai import ChatGoogleGenerativeAI
 import base64
-from google.cloud.aiplatform.generative_models import GenerativeModel, Part, FinishReason
-from google.cloud import aiplatform as generative_models
+from google.cloud import aiplatform
+from google.cloud.aiplatform.gapic.services.model_service import client as model_client
+from google.cloud.aiplatform.gapic.services.model_service.types import generative_model
 
 def generate(document_parts, prompt_text):
     aiplatform.init(project="akroda", location="us-central1")
-    model = GenerativeModel("gemini-1.5-pro-001")
-    responses = model.generate_content(
-        document_parts,
-        generation_config=generation_config,
-        safety_settings=safety_settings,
+    model_service_client = model_client.ModelServiceClient()
+    model_resource_name = model_service_client.model_path("akroda", "us-central1", "gemini-1.5-pro-001")
+    response = model_service_client.generate_content(
+        request=generative_model.GenerateContentRequest(
+            model=model_resource_name,
+            documents=document_parts,
+            prompt=prompt_text,
+            generation_config=generation_config,
+            safety_settings=safety_settings,
         stream=True,
     )