mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 20:26:15 +00:00
fix: correct argument limit error (#2758)
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
parent
326d6a4248
commit
569770e47d
@ -56,7 +56,7 @@ class LLMExtractor(ExtractorBase, ABC):
|
|||||||
"""Inner extract by LLM."""
|
"""Inner extract by LLM."""
|
||||||
# limit check
|
# limit check
|
||||||
if limit and limit < 1:
|
if limit and limit < 1:
|
||||||
ValueError("optional argument limit >= 1")
|
raise ValueError("optional argument limit >= 1")
|
||||||
|
|
||||||
template = HumanPromptTemplate.from_template(self._prompt_template)
|
template = HumanPromptTemplate.from_template(self._prompt_template)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user