Merge branch 'main' into main

Author: averikitsch

docs/graph_qa_chain.ipynb

@@ -499,7 +499,7 @@
     "from IPython.core.display import HTML\n",
     "\n",
     "# Initialize llm object\n",
-    "llm = ChatVertexAI(model=\"gemini-1.5-flash-002\", temperature=0)\n",
+    "llm = ChatVertexAI(model=\"gemini-2.0-flash-001\", temperature=0)\n",
     "\n",
     "# Initialize GraphQAChain\n",
     "chain = SpannerGraphQAChain.from_llm(\n",

tests/integration/test_spanner_graph_qa.py

@@ -37,7 +37,7 @@ def random_string(num_char=3):
 
 def get_llm():
     llm = ChatVertexAI(
-        model="gemini-1.5-flash-002",
+        model="gemini-2.0-flash-001",
         temperature=0,
     )
     return llm

tests/integration/test_spanner_graph_retriever.py

@@ -39,7 +39,7 @@ def random_string(num_char=3):
 
 def get_llm():
     llm = ChatVertexAI(
-        model="gemini-1.5-flash-002",
+        model="gemini-2.0-flash-001",
         temperature=0,
     )
     return llm