nits (#2873)

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

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)

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",
 ]