Update usage of OpenAI client (#166)

AgentOps-AI · Apr 28, 2024 · 712bb68 · 712bb68
1 parent 52a70ee
commit 712bb68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/multi_agent_example.ipynb b/examples/multi_agent_example.ipynb
@@ -19,7 +19,7 @@
     "from agentops.agent import track_agent\n",
     "from dotenv import load_dotenv\n",
     "import os\n",
-    "import openai\n",
+    "from openai import OpenAI\n",
     "import logging\n",
     "\n",
     "from IPython.display import display, Markdown"
@@ -50,7 +50,7 @@
    "outputs": [],
    "source": [
     "agentops.init(AGENTOPS_API_KEY)\n",
-    "openai_client = openai.Client()"
+    "openai_client = openai_client = OpenAI(api_key = OPENAI_API_KEY )"
    ],
    "metadata": {
     "collapsed": false