From 5c2243ee91d28fa2821c11a6e61b9e60e02ff555 Mon Sep 17 00:00:00 2001
From: Lance Martin <122662504+rlancemartin@users.noreply.github.com>
Date: Fri, 27 Oct 2023 16:54:54 -0700
Subject: [PATCH] Update llama.cpp and Ollama templates (#12466)

---
 templates/sql-llamacpp/sql-llamacpp.ipynb   | 15 +++++++++++++--
 templates/sql-ollama/pyproject.toml         |  2 +-
 templates/sql-ollama/sql-ollama.ipynb       | 15 +++++++++++++--
 templates/sql-ollama/sql_ollama/__init__.py |  2 +-
 4 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/templates/sql-llamacpp/sql-llamacpp.ipynb b/templates/sql-llamacpp/sql-llamacpp.ipynb
index b3bf2c3e4..412177f9c 100644
--- a/templates/sql-llamacpp/sql-llamacpp.ipynb
+++ b/templates/sql-llamacpp/sql-llamacpp.ipynb
@@ -19,10 +19,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "ff5869c6-2065-48f3-bb43-52a515968276",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'\\nNatural Language Response: Klay Thompson plays for the Golden State Warriors.'"
+      ]
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "from langserve.client import RemoteRunnable\n",
     "sql_app = RemoteRunnable('http://0.0.0.0:8001/sql_llamacpp')\n",
diff --git a/templates/sql-ollama/pyproject.toml b/templates/sql-ollama/pyproject.toml
index 0ad568975..e8ed637ff 100644
--- a/templates/sql-ollama/pyproject.toml
+++ b/templates/sql-ollama/pyproject.toml
@@ -1,5 +1,5 @@
 [tool.poetry]
-name = "sql_ollama"
+name = "sql-ollama"
 version = "0.1.0"
 description = ""
 authors = ["Lance Martin <lance@langchain.dev>"]
diff --git a/templates/sql-ollama/sql-ollama.ipynb b/templates/sql-ollama/sql-ollama.ipynb
index e41c07e8f..6498e051c 100644
--- a/templates/sql-ollama/sql-ollama.ipynb
+++ b/templates/sql-ollama/sql-ollama.ipynb
@@ -19,10 +19,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "50c27e82-92d8-4fa1-8bc4-b6544e59773d",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "AIMessage(content=' Sure! Here\\'s the natural language response based on the input:\\n\\n\"Klay Thompson plays for the Golden State Warriors.\"')"
+      ]
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "from langserve.client import RemoteRunnable\n",
     "sql_app = RemoteRunnable('http://0.0.0.0:8001/sql_ollama')\n",
diff --git a/templates/sql-ollama/sql_ollama/__init__.py b/templates/sql-ollama/sql_ollama/__init__.py
index d92f80552..25bdf9683 100644
--- a/templates/sql-ollama/sql_ollama/__init__.py
+++ b/templates/sql-ollama/sql_ollama/__init__.py
@@ -1,3 +1,3 @@
-from ollama.chain import chain
+from sql_ollama.chain import chain
 
 __all__ = ["chain"]
\ No newline at end of file