Updated Basic_Evaluation.ipynb to use google-genai (#642)

* Updated module google-generativeai to use google-genai
* Refactored imports
* Modified how configs are being assigned to models
* corrected indentation for better readability
* add 2.5 flash preview to model id param

---------

Co-authored-by: Nikita Namjoshi <[email protected]>

Author: Shi-pra-19

examples/prompting/Basic_Evaluation.ipynb

@@ -11,7 +11,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 30,
       "metadata": {
         "cellView": "form",
         "id": "2K6KP3hUl7_l"
@@ -66,25 +66,25 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 31,
       "metadata": {
         "id": "Ne-3gnXqR0hI"
       },
       "outputs": [],
       "source": [
-        "%pip install -U -q \"google-generativeai>=0.7.2\""
+        "%pip install -U -q \"google-genai>=1.7.0\""
       ]
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 32,
       "metadata": {
         "id": "EconMHePQHGw"
       },
       "outputs": [],
       "source": [
-        "import google.generativeai as genai\n",
-        "\n",
+        "from google import genai\n",
+        "from google.genai import types\n",
         "from IPython.display import Markdown"
       ]
     },
@@ -101,7 +101,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 33,
       "metadata": {
         "id": "v-JZzORUpVR2"
       },
@@ -110,7 +110,7 @@
         "from google.colab import userdata\n",
         "GOOGLE_API_KEY=userdata.get('GOOGLE_API_KEY')\n",
         "\n",
-        "genai.configure(api_key=GOOGLE_API_KEY)"
+        "client = genai.Client(api_key=GOOGLE_API_KEY)"
       ]
     },
     {
@@ -121,77 +121,92 @@
       "source": [
         "## Example\n",
         "\n",
-        "Start by defining some system instructions for this problem. For demonstration purposes, the use case involves prompting the model to write an essay with some mistakes. Remember that for generation tasks like writing an essay, you can change the temperature of the model to get more creative answers. Here, you can use `\"temperature\": 0` for predictability."
+        "Start by defining some system instructions for this problem. For demonstration purposes, the use case involves prompting the model to write an essay with some mistakes. Remember that for generation tasks like writing an essay, you can change the temperature of the model to get more creative answers."
       ]
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 34,
       "metadata": {
         "id": "zNw1Hele0Bne"
       },
       "outputs": [
         {
           "data": {
-            "text/markdown": "## The Power of Words: Why Reading is the Best Thing Ever\n\nReading is a super important thing. It's like, the best thing ever. Everyone should read more, because it's good for you. Like, really good. Reading can make you smarter, more knowledgeable, and even a better person. It's like a magic potion that can transform your brain.\n\nFirstly, reading helps you learn new things. When you read, you're basically absorbing information like a sponge. You can learn about history, science, art, and even how to cook a mean lasagna. It's like having a personal tutor in your pocket, except it doesn't cost anything.\n\nSecondly, reading can make you a better writer. By reading, you learn how to use language effectively. You can see how different authors use words to create different effects. It's like learning from the masters. Plus, reading can help you expand your vocabulary, which is super important for sounding smart.\n\nThirdly, reading can help you relax and de-stress. When you're reading a good book, you can escape from the real world and get lost in another one. It's like taking a vacation without leaving your couch. Reading can also help you sleep better, because it can calm your mind and help you unwind.\n\nIn conclusion, reading is awesome. It's like a superpower that can make you smarter, more knowledgeable, and a better person. So, next time you're feeling bored or stressed, pick up a book and give it a try. You won't regret it. \n",
+            "text/markdown": "Reading is like, a really good thing. It’s beneficial, you know? Like, a lot. That's what I'm gonna talk about. So, like, my thesis is that reading is really important for a bunch of reasons that I will now tell you.\n\nFirst off, reading makes you smart. I read this somewhere, I forget where. Probably on the internet. But, like, think about it. When you read, you learn new words. And new words are good because you can use them to impress people. The more words you know, the smarter you seem. Its like a secret code to being smart, I think. My English teacher is always saying, “use a thesaurus!” So, like, I do. Sometimes. Its useful. Plus, reading exposes you to new ideas, and new ideas are good for…stuff.\n\nSecondly, reading books is way better than watching TV. TV rots your brain. Everyone knows that. I mean, have you ever met someone who watches TV all day and is, like, a genius? No, right? Exactly. But books, books are different. Books make you think, even if the stuff you read doesn't make much sense. I read this book last week, something with like, a red cover, and it didn't make any sense. But I know it was making me smarter, because I kept having to go back and try to understand it. Therefore, books good, TV bad.\n\nFinally, reading can help you with writing. The more you read, the more you pick up on how other people write, their, like, style and stuff. So when it comes time for you to write something, like this essay even, it will be easier. It's like you’re absorbing their talent through your eyeballs. I think it's osmosis or something. And also, it's because you learn all these new words, that I mentioned before. I always learn a whole bunch of new words whenever I am reading. Its just natural, you know.\n\nIn conclusion, reading is really, really good. It makes you smart, its better than watching TV, and it helps you with writing. So everyone should read more. Seriously, go read a book. I did, and look at me, writing this essay! Like, I am super benefiting from reading right now. You should to. Also, I hope I get a good grade on this essay.\n",
             "text/plain": [
               "<IPython.core.display.Markdown object>"
             ]
           },
-          "execution_count": 8,
+          "execution_count": 34,
           "metadata": {},
           "output_type": "execute_result"
         }
       ],
       "source": [
-        "student_system_prompt = \"\"\"You're a college student. Your job is to write an essay riddled with common mistakes and a few major ones.\n",
-        "The essay should have mistakes regarding clarity, grammar, argumentation, and vocabulary.\n",
-        "Ensure your essay includes a clear thesis statement. You should write only an essay, so do not include any notes.\"\"\"\n",
+        "student_system_prompt = \"\"\"\n",
+        "    You're a college student. Your job is to write an essay riddled with common mistakes and a few major ones.\n",
+        "    The essay should have mistakes regarding clarity, grammar, argumentation, and vocabulary.\n",
+        "    Ensure your essay includes a clear thesis statement. You should write only an essay, so do not include any notes.\n",
+        "\"\"\"\n",
         "\n",
-        "student_model = genai.GenerativeModel(model_name='gemini-2.0-flash', generation_config={\"temperature\": 0},\n",
-        "                              system_instruction=student_system_prompt)\n",
+        "MODEL_ID=\"gemini-2.0-flash\" # @param [\"gemini-2.5-flash-preview-04-17\",\"gemini-2.0-flash-lite\",\"gemini-2.0-flash\",\"gemini-2.5-pro-exp-03-25\"] {\"allow-input\":true, isTemplate: true}\n",
         "\n",
-        "essay = student_model.generate_content(\"Write an essay about benefits of reading.\").text\n",
+        "student_response = client.models.generate_content(\n",
+        "    model=MODEL_ID,\n",
+        "    contents='Write an essay about the benefits of reading.',\n",
+        "    config=types.GenerateContentConfig(\n",
+        "        system_instruction=student_system_prompt\n",
+        "    ),\n",
+        ")\n",
+        "\n",
+        "essay = student_response.text\n",
         "Markdown(essay)"
       ]
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 35,
       "metadata": {
         "id": "mzp8ZFXf0-OT"
       },
       "outputs": [
         {
           "data": {
-            "text/markdown": "## Evaluation:\n\n**Thesis Statement:** 2/5 - The essay states that reading is \"the best thing ever\" but doesn't offer a specific argument or claim. It needs a more focused thesis statement.\n\n**Clarity and Precision of Language:** 2/5 - The language is informal and uses slang (\"super important,\" \"like,\" \"really good,\" \"super important\"). It lacks the precision and formality expected in academic writing.\n\n**Grammar and Punctuation:** 2/5 - There are several grammatical errors, including comma splices and run-on sentences. The essay also lacks proper punctuation, particularly with the use of commas.\n\n**Argumentation:** 3/5 - The essay presents three benefits of reading, but the arguments are not fully developed. They rely on generalizations and lack specific examples or evidence.\n\n## Corrected Version:\n\n**The Power of Words: The Transformative Benefits of Reading**\n\nReading is a fundamental human activity with profound benefits for individuals and society. It is not merely a pastime but a powerful tool that can enhance cognitive abilities, broaden perspectives, and foster empathy. This essay will explore three key ways in which reading transforms individuals: by expanding knowledge, improving communication skills, and promoting emotional well-being.\n\nFirstly, reading serves as a gateway to knowledge and understanding. Through books, we can explore diverse cultures, historical events, scientific discoveries, and philosophical ideas. By immersing ourselves in different worlds and perspectives, we expand our intellectual horizons and develop a deeper understanding of the complexities of the world around us. For example, reading historical accounts can provide valuable insights into the past, while scientific texts can deepen our understanding of the natural world.\n\nSecondly, reading enhances communication skills. By engaging with different writing styles and literary techniques, readers develop a greater appreciation for the nuances of language. They learn how to use words effectively to convey ideas, emotions, and arguments. Moreover, reading exposes individuals to a wider vocabulary, enriching their communication and enabling them to express themselves with greater clarity and precision.\n\nThirdly, reading promotes emotional well-being. Engaging with a good book can provide a much-needed escape from the stresses of daily life. It allows individuals to immerse themselves in fictional worlds, connect with relatable characters, and experience a range of emotions. This process can be therapeutic, fostering empathy, reducing stress, and promoting relaxation. Furthermore, reading can inspire creativity and imagination, enriching our inner lives and fostering a sense of wonder.\n\nIn conclusion, reading is not simply a leisure activity but a transformative experience that enriches our lives in countless ways. By expanding our knowledge, improving our communication skills, and promoting emotional well-being, reading empowers us to become more informed, articulate, and compassionate individuals. \n\n**Changes Made:**\n\n* **Thesis Statement:** The thesis statement is now more specific and focused, outlining the three key benefits of reading that will be discussed in the essay.\n* **Clarity and Precision of Language:** The language is more formal and precise, avoiding slang and informal expressions.\n* **Grammar and Punctuation:** The essay has been corrected for grammatical errors, including comma splices and run-on sentences. Proper punctuation has been implemented throughout.\n* **Argumentation:** The arguments are more developed, providing specific examples and evidence to support the claims. The essay now uses a more academic tone and structure. \n",
+            "text/markdown": "Okay, here's a breakdown of your essay, along with a corrected version.\n\n**Evaluation:**\n\n*   **Thesis Statement (2/5):** The thesis is present but lacks sophistication and clarity. It's too conversational and doesn't precisely outline the specific arguments.\n*   **Clarity and Precision of Language (2/5):** The language is overly informal, repetitive (\"like,\" \"you know\"), and lacks precision. The writing could benefit from more specific examples and a more academic tone.\n*   **Grammar and Punctuation (3/5):** There are noticeable errors in grammar and punctuation, including incorrect use of \"its\" vs. \"it's,\" run-on sentences, and comma splices.\n*   **Argumentation (2/5):** The arguments are simplistic and lack strong evidence. The connections between reading and intelligence, writing ability, etc., are not fully developed or supported with credible sources.\n\n**Corrected Essay:**\n\nReading offers multifaceted benefits that extend beyond mere entertainment. Its importance is underscored by its capacity to enhance cognitive abilities, provide a superior alternative to passive screen time, and cultivate effective writing skills.\n\nFirstly, reading significantly contributes to intellectual development. Exposure to a wide range of texts expands vocabulary and enhances comprehension. As one encounters new words and concepts within various contexts, they are integrated into one's cognitive framework, facilitating more nuanced and articulate communication. As such, reading becomes a powerful tool for self-improvement and intellectual growth. Furthermore, reading exposes individuals to diverse perspectives and ideas, which stimulates critical thinking and broadens one's understanding of the world.\n\nSecondly, engaging with literature offers a more enriching experience compared to passive forms of media consumption like television. While television often presents information in a simplified and visually driven format, reading requires active engagement and interpretation. This active engagement fosters concentration, analytical skills, and imagination, leading to deeper cognitive processing. Moreover, the ability to select books allows for a tailored and purposeful learning experience, unlike the often random and unfiltered nature of television programming.\n\nFinally, reading plays a crucial role in the development of writing proficiency. By immersing themselves in the works of skilled authors, readers subconsciously absorb stylistic elements, narrative structures, and effective rhetorical techniques. This implicit learning process enhances one's ability to articulate thoughts clearly, construct compelling arguments, and employ language effectively. Furthermore, the increased vocabulary acquired through reading directly translates into more precise and varied writing.\n\nIn conclusion, reading offers a wealth of cognitive, intellectual, and creative advantages. Its capacity to enhance intellect, surpass the limitations of passive media, and foster effective writing skills underscores its importance in personal and academic development. Therefore, individuals are encouraged to embrace reading as a means of lifelong learning and intellectual enrichment.\n\n**Explanation of Changes:**\n\n*   **Thesis Statement:** The original thesis was very informal. The corrected version is more formal and specific, clearly stating the three main benefits.\n*   **Language:** All instances of \"like,\" \"you know,\" and other informal expressions were removed. The language was made more precise and academic.\n*   **Grammar and Punctuation:** Grammatical errors were corrected (e.g., \"its\" changed to \"it's\"). Run-on sentences were split, and commas were added or removed for clarity.\n*   **Argumentation:** The arguments were strengthened by providing more specific explanations of how reading leads to intelligence, better writing, etc. The vague references to \"reading it somewhere\" were removed. While still lacking direct citations, the corrected version presents more logical reasoning.\n*   **Structure:** The essay was structured with clear topic sentences for each paragraph and a more formal concluding paragraph.\n*   **Tone:** The overall tone was shifted from conversational to academic.\n\n",
             "text/plain": [
               "<IPython.core.display.Markdown object>"
             ]
           },
-          "execution_count": 9,
+          "execution_count": 35,
           "metadata": {},
           "output_type": "execute_result"
         }
       ],
       "source": [
-        "teacher_system_prompt = f\"\"\"\n",
-        "As a teacher, you are tasked with grading students' essays.\n",
-        "Please follow these instructions for evaluation:\n",
+        "teacher_system_prompt = \"\"\"\n",
+        "    As a teacher, you are tasked with grading students' essays.\n",
+        "    Please follow these instructions for evaluation:\n",
         "\n",
-        "1. Evaluate the essay on a scale of 1-5 based on the following criteria:\n",
-        "- Thesis statement,\n",
-        "- Clarity and precision of language,\n",
-        "- Grammar and punctuation,\n",
-        "- Argumentation\n",
+        "    1. Evaluate the essay on a scale of 1-5 based on the following criteria:\n",
+        "       - Thesis statement,\n",
+        "       - Clarity and precision of language,\n",
+        "       - Grammar and punctuation,\n",
+        "       - Argumentation\n",
         "\n",
-        "2. Write a corrected version of the essay, addressing any identified issues\n",
-        "in the original submission. Point what changes were made.\n",
+        "    2. Write a corrected version of the essay, addressing any identified issues\n",
+        "       in the original submission. Point out what changes were made.\n",
         "\"\"\"\n",
-        "teacher_model = genai.GenerativeModel(model_name='gemini-2.0-flash', generation_config={\"temperature\": 0},\n",
-        "                                         system_instruction=teacher_system_prompt)\n",
+        "teacher_response = client.models.generate_content(\n",
+        "    model=MODEL_ID,\n",
+        "    contents=essay,\n",
+        "    config=types.GenerateContentConfig(\n",
+        "        system_instruction=teacher_system_prompt\n",
+        "    ),\n",
+        ")\n",
+        "\n",
         "\n",
-        "Markdown(teacher_model.generate_content(essay).text)"
+        "Markdown(teacher_response.text)"
       ]
     },
     {