Update to use Voyage

Author: sis0k0

notebooks/ai-agents-lab.ipynb

@@ -34,10 +34,27 @@
    "cell_type": "code",
    "execution_count": 1,
    "metadata": {},
-   "outputs": [],
-   "source": [
-    "import os, json\n",
-    "from pymongo import MongoClient"
+   "outputs": [
+    {
+     "ename": "ModuleNotFoundError",
+     "evalue": "No module named 'pymongo'",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[31m---------------------------------------------------------------------------\u001b[39m",
+      "\u001b[31mModuleNotFoundError\u001b[39m                       Traceback (most recent call last)",
+      "\u001b[36mCell\u001b[39m\u001b[36m \u001b[39m\u001b[32mIn[1]\u001b[39m\u001b[32m, line 3\u001b[39m\n\u001b[32m      1\u001b[39m \u001b[38;5;28;01mimport\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34;01mos\u001b[39;00m\n\u001b[32m      2\u001b[39m \u001b[38;5;28;01mimport\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34;01msys\u001b[39;00m\n\u001b[32m----> \u001b[39m\u001b[32m3\u001b[39m \u001b[38;5;28;01mfrom\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34;01mpymongo\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;28;01mimport\u001b[39;00m MongoClient\n\u001b[32m      5\u001b[39m \u001b[38;5;66;03m# Add parent directory to path to import from utils\u001b[39;00m\n\u001b[32m      6\u001b[39m sys.path.append(os.path.join(os.path.dirname(os.getcwd())))\n",
+      "\u001b[31mModuleNotFoundError\u001b[39m: No module named 'pymongo'"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "import sys\n",
+    "from pymongo import MongoClient\n",
+    "\n",
+    "# Add parent directory to path to import from utils\n",
+    "sys.path.append(os.path.join(os.path.dirname(os.getcwd())))\n",
+    "from utils import track_progress, set_env"
    ]
   },
   {
@@ -47,13 +64,35 @@
    "outputs": [],
    "source": [
     "# If you are using your own MongoDB Atlas cluster, use the connection string for your cluster here\n",
-    "MONGODB_URI = os.environ.get(\"MONGODB_URI\")\n",
+    "MONGODB_URI =  os.environ.get(\"MONGODB_URI\")\n",
     "# Initialize a MongoDB Python client\n",
     "mongodb_client = MongoClient(MONGODB_URI)\n",
     "# Check the connection to the server\n",
     "mongodb_client.admin.command(\"ping\")"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Set the LLM provider and passkey provided by your workshop instructor\n",
+    "# NOTE: LLM_PROVIDER can be set to one of \"aws\"/ \"microsoft\" / \"google\"\n",
+    "LLM_PROVIDER = \"aws\"\n",
+    "PASSKEY = \"replace-with-passkey\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Obtain API keys from our AI model proxy and set them as environment variables-- DO NOT CHANGE\n",
+    "set_env([LLM_PROVIDER,\"voyageai\"], PASSKEY)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -106,31 +145,18 @@
     "### Pick an LLM provider of your choice below"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "SERVERLESS_URL = os.environ.get(\"SERVERLESS_URL\")\n",
-    "# Can be one of \"aws\", \"google\" or \"microsoft\"\n",
-    "LLM_PROVIDER = \"aws\""
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
     "from langchain_core.load import load\n",
-    "import requests\n",
+    "from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder\n",
+    "from utils import get_llm\n",
     "\n",
-    "# Obtain the Langchain LLM object from our serverless endpoint\n",
-    "llm_dict = requests.post(\n",
-    "    url=SERVERLESS_URL, json={\"task\": \"get_llm\", \"data\": LLM_PROVIDER}\n",
-    ").json()\n",
-    "llm = load(llm_dict[\"llm\"], secrets_map=llm_dict[\"secrets_map\"])"
+    "# Obtain the Langchain LLM object using the `get_llm` function from the `utils`` module.\n",
+    "llm = get_llm(LLM_PROVIDER)"
    ]
   },
   {
@@ -1741,7 +1767,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "venv",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -1755,7 +1781,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.11"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,