Fix docs/example notebook (#4)

* Fix * Fix
langchain-ai · Sep 26, 2024 · 98883e6 · 98883e6
1 parent 7e2674f
commit 98883e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/libs/snowflake/docs/retrievers.ipynb b/libs/snowflake/docs/retrievers.ipynb
@@ -104,9 +104,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_snowflake import SnowflakeRetriever\n",
+    "from langchain_snowflake import CortexSearchRetriever\n",
     "\n",
-    "retriever = SnowflakeRetriever(\n",
+    "retriever = CortexSearchRetriever(\n",
     "    ## Snowflake connection parameters. Uncomment out if not set on the environment.\n",
     "    \n",
     "    # username=\"snowflake-username\",\n",

diff --git a/libs/snowflake/langchain_snowflake/retrievers.py b/libs/snowflake/langchain_snowflake/retrievers.py
@@ -78,9 +78,9 @@ class CortexSearchRetriever(BaseRetriever):
     Instantiate:
         .. code-block:: python
 
-            from langchain-snowflake import SnowflakeRetriever
+            from langchain-snowflake import CortexSearchRetriever
 
-            retriever = SnowflakeRetriever(
+            retriever = CortexSearchRetriever(
                 authenticator="externalbrowser",
                 columns=["name", "description", "era"],
                 search_column="description",