From 8b6385f2f9d1dda432250c2d2530924dda0c0aab Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Fri, 24 May 2024 09:27:50 -0700 Subject: [PATCH] Revert "remove max_tokens from example (#425)" This reverts commit e285bb4d7ea9aee324c448257b19312751109777. --- site/en/gemini-api/docs/get-started/python.ipynb | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/site/en/gemini-api/docs/get-started/python.ipynb b/site/en/gemini-api/docs/get-started/python.ipynb index 04c17525e..989cfd701 100644 --- a/site/en/gemini-api/docs/get-started/python.ipynb +++ b/site/en/gemini-api/docs/get-started/python.ipynb @@ -1752,6 +1752,7 @@ " # Only one candidate for now.\n", " candidate_count=1,\n", " stop_sequences=['x'],\n", + " max_output_tokens=20,\n", " temperature=1.0)\n", ")" ] @@ -1779,6 +1780,10 @@ ], "source": [ "text = response.text\n", + "\n", + "if response.candidates[0].finish_reason.name == \"MAX_TOKENS\":\n", + " text += '...'\n", + "\n", "to_markdown(text)" ] },