mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 04:12:13 +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."""
|
||||
# limit check
|
||||
if limit and limit < 1:
|
||||
ValueError("optional argument limit >= 1")
|
||||
raise ValueError("optional argument limit >= 1")
|
||||
|
||||
template = HumanPromptTemplate.from_template(self._prompt_template)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user