Skip to content

Commit 8df6b68

Browse files
authored
fix length based example selector (#862)
1 parent 3f48eed commit 8df6b68

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

langchain/prompts/example_selector/length_based.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def select_examples(self, input_variables: Dict[str, str]) -> List[dict]:
5151
examples = []
5252
while remaining_length > 0 and i < len(self.examples):
5353
new_length = remaining_length - self.example_text_lengths[i]
54-
if i < 0:
54+
if new_length < 0:
5555
break
5656
else:
5757
examples.append(self.examples[i])

tests/unit_tests/prompts/test_length_based_example_selector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def selector() -> LengthBasedExampleSelector:
1717
selector = LengthBasedExampleSelector(
1818
examples=EXAMPLES,
1919
example_prompt=prompts,
20-
max_length=25,
20+
max_length=30,
2121
)
2222
return selector
2323

0 commit comments

Comments
 (0)