github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-02 19:47:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

notebook fmt (#12498)

Browse Source
This commit is contained in:
Bagatur
2023-10-29 15:50:09 -07:00
committed by GitHub
parent 56cc5b847c
commit 2424fff3f1
342 changed files with 8261 additions and 6796 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
templates/sql-llamacpp/sql_llamacpp/__init__.py
View File

@@ -1,3 +1,3 @@
from sql_llamacpp.chain import chain
__all__ = ["chain"]
__all__ = ["chain"]

15
templates/sql-llamacpp/sql_llamacpp/chain.py
View File

@@ -13,8 +13,10 @@ from langchain.utilities import SQLDatabase
# File name and URL
file_name = "mistral-7b-instruct-v0.1.Q4_K_M.gguf"
url = ("https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.1-GGUF/resolve/main/"
"mistral-7b-instruct-v0.1.Q4_K_M.gguf")
url = (
"https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.1-GGUF/resolve/main/"
"mistral-7b-instruct-v0.1.Q4_K_M.gguf"
)
# Check if file is present in the current directory
if not os.path.exists(file_name):
print(f"'{file_name}' not found. Downloading...")
@@ -114,12 +116,17 @@ prompt_response = ChatPromptTemplate.from_messages(
("human", template),
]
)
# Supply the input types to the prompt
# Supply the input types to the prompt
class InputType(BaseModel):
question: str
chain = (
RunnablePassthrough.assign(query=sql_response_memory).with_types(input_type=InputType)
RunnablePassthrough.assign(query=sql_response_memory).with_types(
input_type=InputType
)
| RunnablePassthrough.assign(
schema=get_schema,
response=lambda x: db.run(x["query"]),