adding cloud pickle in requirements (GoogleCloudPlatform#12303)

CadillacBurgess1 · web-flow · commit 6d2cb2fae503 · 2024-08-21T19:56:51.000-07:00
diff --git a/generative_ai/gemini_reasoning_engine.py b/generative_ai/gemini_reasoning_engine.py
@@ -56,7 +56,7 @@ def query(self, a: int, b: int) -> str:
         SimpleAdditionApp(),
         display_name="Demo Addition App",
         description="A simple demo addition app",
-        requirements=[],
+        requirements=["cloudpickle==3"],
         extra_packages=[],
     )
     # [END generativeaionvertexai_create_reasoning_engine_basic]
@@ -124,6 +124,7 @@ def query(self, question: str) -> Union[str, List[Union[str, Dict]]]:
             "google-cloud-aiplatform==1.50.0",
             "langchain-google-vertexai",
             "langchain-core",
+            "cloudpickle==3",
         ],
         display_name="Demo LangChain App",
         description="This is a simple LangChain app.",
