From 8fef69296db61a705123d9f3d13f94edbec3bd80 Mon Sep 17 00:00:00 2001
From: Harrison Chase <hw.chase.17@gmail.com>
Date: Fri, 14 Apr 2023 07:55:12 -0700
Subject: [PATCH] nits (#2873)

---
 .../indexes/document_loaders/examples/git.ipynb     | 13 +++++--------
 langchain/chains/sql_database/base.py               |  6 +-----
 langchain/document_loaders/__init__.py              |  2 ++
 3 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/docs/modules/indexes/document_loaders/examples/git.ipynb b/docs/modules/indexes/document_loaders/examples/git.ipynb
index 87a6feb8..306a68fa 100644
--- a/docs/modules/indexes/document_loaders/examples/git.ipynb
+++ b/docs/modules/indexes/document_loaders/examples/git.ipynb
@@ -1,7 +1,6 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -11,7 +10,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -38,7 +36,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.git import GitLoader"
+    "from langchain.document_loaders import GitLoader"
    ]
   },
   {
@@ -109,7 +107,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.git import GitLoader"
+    "from langchain.document_loaders import GitLoader"
    ]
   },
   {
@@ -164,7 +162,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "ai",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -178,9 +176,8 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.6"
-  },
-  "orig_nbformat": 4
+   "version": "3.9.1"
+  }
  },
  "nbformat": 4,
  "nbformat_minor": 2
diff --git a/langchain/chains/sql_database/base.py b/langchain/chains/sql_database/base.py
index 56ce0d27..aa1a2e6d 100644
--- a/langchain/chains/sql_database/base.py
+++ b/langchain/chains/sql_database/base.py
@@ -65,11 +65,7 @@ class SQLDatabaseChain(Chain):
             return [self.output_key, "intermediate_steps"]
 
     def _call(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
-        try:
-            prompt = self.prompt or SQL_PROMPTS[self.database.dialect]
-        except KeyError:
-            # fallback to generic prompt if dialect-specific prompt doesn't exist yet
-            prompt = PROMPT
+        prompt = self.prompt or SQL_PROMPTS.get(self.database.dialect, PROMPT)
         llm_chain = LLMChain(llm=self.llm, prompt=prompt)
         input_text = f"{inputs[self.input_key]}\nSQLQuery:"
         self.callback_manager.on_text(input_text, verbose=self.verbose)
diff --git a/langchain/document_loaders/__init__.py b/langchain/document_loaders/__init__.py
index 17d2cde9..b5924cef 100644
--- a/langchain/document_loaders/__init__.py
+++ b/langchain/document_loaders/__init__.py
@@ -27,6 +27,7 @@ from langchain.document_loaders.evernote import EverNoteLoader
 from langchain.document_loaders.facebook_chat import FacebookChatLoader
 from langchain.document_loaders.gcs_directory import GCSDirectoryLoader
 from langchain.document_loaders.gcs_file import GCSFileLoader
+from langchain.document_loaders.git import GitLoader
 from langchain.document_loaders.gitbook import GitbookLoader
 from langchain.document_loaders.googledrive import GoogleDriveLoader
 from langchain.document_loaders.gutenberg import GutenbergLoader
@@ -142,4 +143,5 @@ __all__ = [
     "BigQueryLoader",
     "BiliBiliLoader",
     "SlackDirectoryLoader",
+    "GitLoader",
 ]