mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-19 21:33:31 -04:00
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:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user