github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-11 22:04:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix length based example selector (#862)

Browse Source
This commit is contained in:
Harrison Chase 2023-02-02 22:06:56 -08:00 committed by GitHub
parent 3f48eed5bd
commit 8df6b68093
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/prompts/example_selector/length_based.py
View File

@ -51,7 +51,7 @@ class LengthBasedExampleSelector(BaseExampleSelector, BaseModel):
examples = [] examples = []
while remaining_length > 0 and i < len(self.examples): while remaining_length > 0 and i < len(self.examples):
new_length = remaining_length - self.example_text_lengths[i] new_length = remaining_length - self.example_text_lengths[i]
if i < 0: if new_length < 0:
break break
else: else:
examples.append(self.examples[i]) examples.append(self.examples[i])

2
tests/unit_tests/prompts/test_length_based_example_selector.py
View File

@ -17,7 +17,7 @@ def selector() -> LengthBasedExampleSelector:
selector = LengthBasedExampleSelector( selector = LengthBasedExampleSelector(
examples=EXAMPLES, examples=EXAMPLES,
example_prompt=prompts, example_prompt=prompts,
max_length=25, max_length=30,
) )
return selector return selector