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
This commit is contained in:
aditya thomas 2024-03-23 05:35:27 +05:30 committed by GitHub
parent 0cc0467267
commit 375ab7bf59
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -54,6 +54,8 @@
"import getpass\n", "import getpass\n",
"import os\n", "import os\n",
"\n", "\n",
"from langchain_fireworks import Fireworks\n",
"\n",
"if \"FIREWORKS_API_KEY\" not in os.environ:\n", "if \"FIREWORKS_API_KEY\" not in os.environ:\n",
" os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n", " os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n",
"\n", "\n",
@ -181,7 +183,7 @@
], ],
"source": [ "source": [
"from langchain.prompts import PromptTemplate\n", "from langchain.prompts import PromptTemplate\n",
"from langchain_community.llms.fireworks import Fireworks\n", "from langchain_fireworks import Fireworks\n",
"\n", "\n",
"llm = Fireworks(\n", "llm = Fireworks(\n",
" model=\"accounts/fireworks/models/mixtral-8x7b-instruct\",\n", " model=\"accounts/fireworks/models/mixtral-8x7b-instruct\",\n",
@ -249,7 +251,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.11.4" "version": "3.9.6"
} }
}, },
"nbformat": 4, "nbformat": 4,