github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 23:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain[patch]: smith.evaluation.progress.ProgressBarCallback: Make output after progress bar ends configurable (#31583)

Browse Source
This commit is contained in:
dennism-tulcolabs 2025-06-20 18:24:35 -05:00 committed by GitHub
parent 6105a5841b
commit 9de4f22205
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/langchain/langchain/smith/evaluation/progress.py
View File

@ -13,15 +13,19 @@ from langchain_core.outputs import LLMResult
class ProgressBarCallback(base_callbacks.BaseCallbackHandler): class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
"""A simple progress bar for the console.""" """A simple progress bar for the console."""
def __init__(self, total: int, ncols: int = 50, **kwargs: Any): def __init__(
self, total: int, ncols: int = 50, end_with: str = "\n", **kwargs: Any
):
"""Initialize the progress bar. """Initialize the progress bar.
Args: Args:
total: int, the total number of items to be processed. total: int, the total number of items to be processed.
ncols: int, the character width of the progress bar. ncols: int, the character width of the progress bar.
end_with: str, last string to print after progress bar reaches end.
""" """
self.total = total self.total = total
self.ncols = ncols self.ncols = ncols
self.end_with = end_with
self.counter = 0 self.counter = 0
self.lock = threading.Lock() self.lock = threading.Lock()
self._print_bar() self._print_bar()
@ -37,7 +41,8 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
progress = self.counter / self.total progress = self.counter / self.total
arrow = "-" * int(round(progress * self.ncols) - 1) + ">" arrow = "-" * int(round(progress * self.ncols) - 1) + ">"
spaces = " " * (self.ncols - len(arrow)) spaces = " " * (self.ncols - len(arrow))
print(f"\r[{arrow + spaces}] {self.counter}/{self.total}", end="") # noqa: T201 end = "" if self.counter < self.total else self.end_with
print(f"\r[{arrow + spaces}] {self.counter}/{self.total}", end=end) # noqa: T201
def on_chain_error( def on_chain_error(
self, self,