fix: correct argument limit error (#2758)

Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
Emmanuel Ferdman 2025-06-10 10:08:17 +03:00 committed by GitHub
parent 326d6a4248
commit 569770e47d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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