rename and clean gemini example notebook

Author: dot-agi

…emini_examples/gemini_example_sync.ipynb …les/gemini_examples/gemini_example.ipynbexamples/gemini_examples/gemini_example_sync.ipynb renamed to examples/gemini_examples/gemini_example.ipynb examples/gemini_examples/gemini_example_sync.ipynb renamed to examples/gemini_examples/gemini_example.ipynb

@@ -19,28 +19,23 @@
    "source": [
     "import google.generativeai as genai\n",
     "import agentops\n",
-    "from agentops.llms.providers.gemini import GeminiProvider"
+    "from dotenv import load_dotenv\n",
+    "import os"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "a94545c9",
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Configure the Gemini API\n",
-    "import os\n",
+    "load_dotenv()\n",
     "\n",
-    "# Replace with your API key\n",
-    "# You can get one at: https://ai.google.dev/tutorials/setup\n",
-    "GEMINI_API_KEY = \"YOUR_API_KEY_HERE\"  # Replace with your API key\n",
-    "genai.configure(api_key=GEMINI_API_KEY)\n",
+    "GEMINI_API_KEY = os.getenv(\"GEMINI_API_KEY\") or \"your gemini api key\"\n",
+    "AGENTOPS_API_KEY = os.getenv(\"AGENTOPS_API_KEY\") or \"your agentops api key\"\n",
     "\n",
-    "# Note: In production, use environment variables:\n",
-    "# import os\n",
-    "# GEMINI_API_KEY = os.getenv(\"GEMINI_API_KEY\")\n",
-    "# genai.configure(api_key=GEMINI_API_KEY)"
+    "genai.configure(api_key=GEMINI_API_KEY)"
    ]
   },
   {
@@ -51,12 +46,8 @@
    "outputs": [],
    "source": [
     "# Initialize AgentOps and Gemini model\n",
-    "ao_client = agentops.init()\n",
-    "model = genai.GenerativeModel(\"gemini-1.5-flash\")\n",
-    "\n",
-    "# Initialize and override Gemini provider\n",
-    "provider = GeminiProvider(model)\n",
-    "provider.override()"
+    "agentops.init()\n",
+    "model = genai.GenerativeModel(\"gemini-1.5-flash\")"
    ]
   },
   {
@@ -70,7 +61,7 @@
     "print(\"Testing synchronous generation:\")\n",
     "response = model.generate_content(\n",
     "    \"What are the three laws of robotics?\",\n",
-    "    session=ao_client\n",
+    "    # session=ao_client\n",
     ")\n",
     "print(response.text)"
    ]
@@ -87,7 +78,7 @@
     "response = model.generate_content(\n",
     "    \"Explain the concept of machine learning in simple terms.\",\n",
     "    stream=True,\n",
-    "    session=ao_client\n",
+    "    # session=ao_client\n",
     ")\n",
     "\n",
     "for chunk in response:\n",
@@ -98,7 +89,7 @@
     "print(\"\\nTesting another synchronous generation:\")\n",
     "response = model.generate_content(\n",
     "    \"What is the difference between supervised and unsupervised learning?\",\n",
-    "    session=ao_client\n",
+    "    # session=ao_client\n",
     ")\n",
     "print(response.text)"
    ]
@@ -111,25 +102,29 @@
    "outputs": [],
    "source": [
     "# End session and check stats\n",
-    "agentops.end_session(\n",
-    "    end_state=\"Success\",\n",
-    "    end_state_reason=\"Gemini integration example completed successfully\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "b6d35f28",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Clean up\n",
-    "provider.undo_override()"
+    "agentops.end_session(end_state=\"Success\")"
    ]
   }
  ],
- "metadata": {},
+ "metadata": {
+  "kernelspec": {
+   "display_name": "ops",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.16"
+  }
+ },
  "nbformat": 4,
  "nbformat_minor": 5
 }