diff --git a/docs/modules/indexes/document_loaders/examples/git.ipynb b/docs/modules/indexes/document_loaders/examples/git.ipynb index 87a6feb8fd..306a68fa46 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 56ce0d27a0..aa1a2e6dc5 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 17d2cde9b8..b5924cefed 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", ]