minor fix on query tool fn

GoogleCloudPlatform · Jul 2, 2024 · 191e555 · 191e555
1 parent 75f7751
commit 191e555
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/vertex_genai/solutions/gemini_function_calling.ipynb b/notebooks/vertex_genai/solutions/gemini_function_calling.ipynb
@@ -1017,7 +1017,7 @@
    },
    "outputs": [],
    "source": [
-    "def handle_query_fn_call(fn_name: str, fn_args: dict) -> Union[int, float]:\n",
+    "def handle_query_fn_call(fn_name: str, fn_args: dict):\n",
     "    \"\"\"Handles query tool function calls.\"\"\"\n",
     "\n",
     "    print(f\"Function calling: {fn_name} with args: {str(fn_args)}\\n\")\n",