Browse Source

Fix redundant code lines

pull/162/head
nssuu 3 months ago
parent
commit
a1cd181f2e
  1. 59
      week1/community-contributions/day2-webpage-summarizer-ollama-gemini.ipynb

59
week1/community-contributions/day2-webpage-summarizer-ollama-gemini.ipynb

@ -20,7 +20,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 1, "execution_count": 12,
"id": "be06ce76-20ee-4066-9582-a4ed745f278f", "id": "be06ce76-20ee-4066-9582-a4ed745f278f",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -35,33 +35,32 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": 13,
"id": "f3a93b10-f528-4e68-bb95-942596cad52e", "id": "99e42519-5dac-4b13-8a26-8a635753343b",
"metadata": {},
"outputs": [],
"source": [
"load_dotenv()\n",
"api_key = os.getenv(\"GEMINI_API_KEY\")\n",
"\n",
"if not api_key or len(api_key) < 39:\n",
" print(\"No correct api key was found\")\n",
"else:\n",
" print(\"Api key found. Good to go!\")"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "37e7b93f-faf7-415d-8d8c-35e81c6388d3",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"client = genai.Client(api_key=api_key)" "def gemini_invoke(website):\n",
" load_dotenv()\n",
" api_key = os.getenv(\"GEMINI_API_KEY\")\n",
" if not api_key or len(api_key) < 39:\n",
" print(\"No correct api key was found\")\n",
" return\n",
" else:\n",
" print(\"Api key found. Good to go!\")\n",
" client = genai.Client(api_key=api_key)\n",
" response = client.models.generate_content(\n",
" model=\"gemini-2.0-flash\",\n",
" config=types.GenerateContentConfig(\n",
" system_instruction=system_prompt),\n",
" contents=user_prompt_for(website)\n",
" )\n",
" return response.text"
] ]
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 4, "execution_count": 14,
"id": "95a6ece8-8402-4cad-96b9-36a6ea444c54", "id": "95a6ece8-8402-4cad-96b9-36a6ea444c54",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -97,7 +96,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 6, "execution_count": 15,
"id": "233b8904-7a4a-4265-8b0d-20934ae4b29c", "id": "233b8904-7a4a-4265-8b0d-20934ae4b29c",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -109,7 +108,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 13, "execution_count": 16,
"id": "5c996c03-84ab-4378-8a55-026d94404d35", "id": "5c996c03-84ab-4378-8a55-026d94404d35",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -119,7 +118,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 14, "execution_count": 17,
"id": "abf9464e-dc8d-4099-aeb6-495498326673", "id": "abf9464e-dc8d-4099-aeb6-495498326673",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -135,7 +134,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 15, "execution_count": 20,
"id": "32ab2d29-02d1-43c5-b920-f2621f292b23", "id": "32ab2d29-02d1-43c5-b920-f2621f292b23",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -149,13 +148,7 @@
" response = ollama.chat(model=Model, messages=messages)\n", " response = ollama.chat(model=Model, messages=messages)\n",
" return response[\"message\"][\"content\"]\n", " return response[\"message\"][\"content\"]\n",
" else:\n", " else:\n",
" response = client.models.generate_content(\n", " return gemini_invoke(website)"
" model=\"gemini-2.0-flash\",\n",
" config=types.GenerateContentConfig(\n",
" system_instruction=system_prompt),\n",
" contents=user_prompt_for(website)\n",
" )\n",
" return response.text"
] ]
}, },
{ {
@ -165,7 +158,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"summarize(\"https://edwarddonner.com\", \"gemini\")" "summarize(\"https://edwarddonner.com\", \"ollama\")"
] ]
} }
], ],

Loading…
Cancel
Save