changed chat followup prompt

GoogleCloudPlatform · May 22, 2024 · f4f3c8c · f4f3c8c
1 parent 2260630
commit f4f3c8c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/notebooks/vertex_genai/solutions/grounding_vertex_agent_builder.ipynb b/notebooks/vertex_genai/solutions/grounding_vertex_agent_builder.ipynb
@@ -108,7 +108,6 @@
    },
    "outputs": [],
    "source": [
-    "import vertexai\n",
     "from IPython.display import Markdown, display\n",
     "from vertexai.language_models import (\n",
     "    ChatModel,\n",
@@ -134,7 +133,6 @@
    "source": [
     "parameters = {\n",
     "    \"temperature\": 0.2,  # Temperature controls the degree of randomness in token selection.\n",
-    "    \"max_output_tokens\": 256,  # Token limit determines the maximum amount of text output.\n",
     "    \"top_p\": 0.8,  # Tokens are selected from most probable to least until the sum of their probabilities equals the top_p value.\n",
     "    \"top_k\": 40,  # A top_k of 1 means the selected token is the most probable among all tokens.\n",
     "}\n",
@@ -301,7 +299,7 @@
    "outputs": [],
    "source": [
     "PROMPT = \"How can I ground LLM responses in Vertex AI?\"\n",
-    "PROMPT_FOLLOWUP = \"What types of data can I use for grounding?\""
+    "PROMPT_FOLLOWUP = \"Is grounding available in PaLM models?\""
    ]
   },
   {