more updates

run-llama · Dec 16, 2024 · 9f23093 · 9f23093
1 parent 8d1a81e
commit 9f23093
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 19 deletions.
diff --git a/llama-index-integrations/tools/llama-index-tools-tavily-research/README.md b/llama-index-integrations/tools/llama-index-tools-tavily-research/README.md
@@ -20,12 +20,16 @@ Here's an example usage of the TavilyToolSpec.
 
 ```python
 from llama_index.tools.tavily_research import TavilyToolSpec
-from llama_index.agent.openai import OpenAIAgent
+from llama_index.core.agent import FunctionCallingAgent
+from llama_index.llms.openai import OpenAI
 
 tavily_tool = TavilyToolSpec(
     api_key="your-key",
 )
-agent = OpenAIAgent.from_tools(tavily_tool.to_tool_list())
+agent = FunctionCallingAgent.from_tools(
+    tavily_tool.to_tool_list(),
+    llm=OpenAI(model="gpt-4o"),
+)
 
 agent.chat("What happened in the latest Burning Man festival?")
 ```

diff --git a/llama-index-integrations/tools/llama-index-tools-tavily-research/examples/tavily.ipynb b/llama-index-integrations/tools/llama-index-tools-tavily-research/examples/tavily.ipynb
@@ -23,19 +23,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install llama-index-tools-tavily-research llama-index llama-hub tavily-python"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ce6c6e5d-f61e-461e-8e8d-99d27fb3fcaf",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Set up OpenAI\n",
-    "import openai\n",
-    "from llama_index.agent.openai import OpenAIAgent"
+    "%pip install llama-index-tools-tavily-research llama-index"
    ]
   },
   {
@@ -45,7 +33,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "openai.api_key = \"sk-api-key\""
+    "# set your openai key, if using openai\n",
+    "import os\n",
+    "\n",
+    "os.environ[\"OPENAI_API_KEY\"] = \"sk-...\"\n",
+    "os.environ[\"TAVILY_API_KEY\"] = \"...\""
    ]
   },
   {
@@ -125,8 +117,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "agent = OpenAIAgent.from_tools(\n",
+    "from llama_index.core.agent import FunctionCallingAgent\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "agent = FunctionCallingAgent.from_tools(\n",
     "    tavily_tool_list,\n",
+    "    llm=OpenAI(model=\"gpt-4o\"),\n",
     ")"
    ]
   },
@@ -176,9 +172,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "llama_hub",
+   "display_name": "llama-index-caVs7DDe-py3.10",
    "language": "python",
-   "name": "llama_hub"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {