From 375ab7bf59e7f17eec7ab98edcfb62a900d8456e Mon Sep 17 00:00:00 2001 From: aditya thomas Date: Sat, 23 Mar 2024 05:35:27 +0530 Subject: [PATCH] docs: update module imports for fireworks documentation (#19377) **Description:** Update module imports for Fireworks documentation **Issue:** Module imports not present or in incorrect location **Dependencies:** None --- docs/docs/integrations/llms/fireworks.ipynb | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/docs/docs/integrations/llms/fireworks.ipynb b/docs/docs/integrations/llms/fireworks.ipynb index 2aaae1b699c..57e772f760c 100644 --- a/docs/docs/integrations/llms/fireworks.ipynb +++ b/docs/docs/integrations/llms/fireworks.ipynb @@ -54,6 +54,8 @@ "import getpass\n", "import os\n", "\n", + "from langchain_fireworks import Fireworks\n", + "\n", "if \"FIREWORKS_API_KEY\" not in os.environ:\n", " os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n", "\n", @@ -181,7 +183,7 @@ ], "source": [ "from langchain.prompts import PromptTemplate\n", - "from langchain_community.llms.fireworks import Fireworks\n", + "from langchain_fireworks import Fireworks\n", "\n", "llm = Fireworks(\n", " model=\"accounts/fireworks/models/mixtral-8x7b-instruct\",\n", @@ -249,7 +251,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.4" + "version": "3.9.6" } }, "nbformat": 4,