Harrison/improve cache (#368)

make it so everything goes through generate, which removes the need for
two types of caches
This commit is contained in:
Harrison Chase
2022-12-18 16:22:42 -05:00
committed by GitHub
parent 8d0869c6d3
commit 3474f39e21
28 changed files with 104 additions and 140 deletions
+2 -2
View File
@@ -7,7 +7,7 @@ from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
from langchain.chains.sql_database.prompt import PROMPT
from langchain.input import print_text
from langchain.llms.base import LLM
from langchain.llms.base import BaseLLM
from langchain.sql_database import SQLDatabase
@@ -22,7 +22,7 @@ class SQLDatabaseChain(Chain, BaseModel):
db_chain = SelfAskWithSearchChain(llm=OpenAI(), database=db)
"""
llm: LLM
llm: BaseLLM
"""LLM wrapper to use."""
database: SQLDatabase
"""SQL Database to connect to."""