adding cloud pickle in requirements (#12303)

GoogleCloudPlatform · Aug 22, 2024 · 6d2cb2f · 6d2cb2f
1 parent 31ca8fb
commit 6d2cb2f
Showing 1 changed file with 2 additions and 1 deletion.
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.",