mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-10 05:20:39 +00:00
fix length based example selector (#862)
This commit is contained in:
parent
3f48eed5bd
commit
8df6b68093
@ -51,7 +51,7 @@ class LengthBasedExampleSelector(BaseExampleSelector, BaseModel):
|
||||
examples = []
|
||||
while remaining_length > 0 and i < len(self.examples):
|
||||
new_length = remaining_length - self.example_text_lengths[i]
|
||||
if i < 0:
|
||||
if new_length < 0:
|
||||
break
|
||||
else:
|
||||
examples.append(self.examples[i])
|
||||
|
@ -17,7 +17,7 @@ def selector() -> LengthBasedExampleSelector:
|
||||
selector = LengthBasedExampleSelector(
|
||||
examples=EXAMPLES,
|
||||
example_prompt=prompts,
|
||||
max_length=25,
|
||||
max_length=30,
|
||||
)
|
||||
return selector
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user