|
|
|
@ -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)
|
|
|
|
|