lm-deluge 0.0.16__py3-none-any.whl → 0.0.18__py3-none-any.whl

lm_deluge/built_in_tools/anthropic/editor.py

@@ -0,0 +1,559 @@
+# from collections import defaultdict
+# from pathlib import Path
+# from typing import Any, Literal, get_args
+
+# Command_20250124 = Literal[
+#     "view",
+#     "create",
+#     "str_replace",
+#     "insert",
+#     "undo_edit",
+# ]
+
+# Command_20250429 = Literal[
+#     "view",
+#     "create",
+#     "str_replace",
+#     "insert",
+# ]
+# SNIPPET_LINES: int = 4
+
+
+# class EditTool:
+#     """
+#     An filesystem editor tool that allows the agent to view,
+#     create, and edit files. The tool parameters are defined by
+#     Anthropic and are not editable.
+#     """
+
+#     api_type: Literal["text_editor_20250124"] = "text_editor_20250124"
+#     name: Literal["str_replace_editor"] = "str_replace_editor"
+
+#     _file_history: dict[Path, list[str]]
+
+#     def __init__(self):
+#         self._file_history = defaultdict(list)
+#         super().__init__()
+
+#     def to_params(self) -> Any:
+#         return {
+#             "name": self.name,
+#             "type": self.api_type,
+#         }
+
+#     async def __call__(
+#         self,
+#         *,
+#         command: Command_20250124,
+#         path: str,
+#         file_text: str | None = None,
+#         view_range: list[int] | None = None,
+#         old_str: str | None = None,
+#         new_str: str | None = None,
+#         insert_line: int | None = None,
+#         **kwargs,
+#     ):
+#         _path = Path(path)
+#         self.validate_path(command, _path)
+#         if command == "view":
+#             return await self.view(_path, view_range)
+#         elif command == "create":
+#             if file_text is None:
+#                 raise ToolError("Parameter `file_text` is required for command: create")
+#             self.write_file(_path, file_text)
+#             self._file_history[_path].append(file_text)
+#             return ToolResult(output=f"File created successfully at: {_path}")
+#         elif command == "str_replace":
+#             if old_str is None:
+#                 raise ToolError(
+#                     "Parameter `old_str` is required for command: str_replace"
+#                 )
+#             return self.str_replace(_path, old_str, new_str)
+#         elif command == "insert":
+#             if insert_line is None:
+#                 raise ToolError(
+#                     "Parameter `insert_line` is required for command: insert"
+#                 )
+#             if new_str is None:
+#                 raise ToolError("Parameter `new_str` is required for command: insert")
+#             return self.insert(_path, insert_line, new_str)
+#         elif command == "undo_edit":
+#             return self.undo_edit(_path)
+#         raise ToolError(
+#             f"Unrecognized command {command}. The allowed commands for the {self.name} tool are: {', '.join(get_args(Command_20250124))}"
+#         )
+
+#     def validate_path(self, command: str, path: Path):
+#         """
+#         Check that the path/command combination is valid.
+#         """
+#         # Check if its an absolute path
+#         if not path.is_absolute():
+#             suggested_path = Path("") / path
+#             raise ToolError(
+#                 f"The path {path} is not an absolute path, it should start with `/`. Maybe you meant {suggested_path}?"
+#             )
+#         # Check if path exists
+#         if not path.exists() and command != "create":
+#             raise ToolError(
+#                 f"The path {path} does not exist. Please provide a valid path."
+#             )
+#         if path.exists() and command == "create":
+#             raise ToolError(
+#                 f"File already exists at: {path}. Cannot overwrite files using command `create`."
+#             )
+#         # Check if the path points to a directory
+#         if path.is_dir():
+#             if command != "view":
+#                 raise ToolError(
+#                     f"The path {path} is a directory and only the `view` command can be used on directories"
+#                 )
+
+#     async def view(self, path: Path, view_range: list[int] | None = None):
+#         """Implement the view command"""
+#         if path.is_dir():
+#             if view_range:
+#                 raise ToolError(
+#                     "The `view_range` parameter is not allowed when `path` points to a directory."
+#                 )
+
+#             _, stdout, stderr = await run(
+#                 rf"find {path} -maxdepth 2 -not -path '*/\.*'"
+#             )
+#             if not stderr:
+#                 stdout = f"Here's the files and directories up to 2 levels deep in {path}, excluding hidden items:\n{stdout}\n"
+#             return CLIResult(output=stdout, error=stderr)
+
+#         file_content = self.read_file(path)
+#         init_line = 1
+#         if view_range:
+#             if len(view_range) != 2 or not all(isinstance(i, int) for i in view_range):
+#                 raise ToolError(
+#                     "Invalid `view_range`. It should be a list of two integers."
+#                 )
+#             file_lines = file_content.split("\n")
+#             n_lines_file = len(file_lines)
+#             init_line, final_line = view_range
+#             if init_line < 1 or init_line > n_lines_file:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its first element `{init_line}` should be within the range of lines of the file: {[1, n_lines_file]}"
+#                 )
+#             if final_line > n_lines_file:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its second element `{final_line}` should be smaller than the number of lines in the file: `{n_lines_file}`"
+#                 )
+#             if final_line != -1 and final_line < init_line:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its second element `{final_line}` should be larger or equal than its first `{init_line}`"
+#                 )
+
+#             if final_line == -1:
+#                 file_content = "\n".join(file_lines[init_line - 1 :])
+#             else:
+#                 file_content = "\n".join(file_lines[init_line - 1 : final_line])
+
+#         return CLIResult(
+#             output=self._make_output(file_content, str(path), init_line=init_line)
+#         )
+
+#     def str_replace(self, path: Path, old_str: str, new_str: str | None):
+#         """Implement the str_replace command, which replaces old_str with new_str in the file content"""
+#         # Read the file content
+#         file_content = self.read_file(path).expandtabs()
+#         old_str = old_str.expandtabs()
+#         new_str = new_str.expandtabs() if new_str is not None else ""
+
+#         # Check if old_str is unique in the file
+#         occurrences = file_content.count(old_str)
+#         if occurrences == 0:
+#             raise ToolError(
+#                 f"No replacement was performed, old_str `{old_str}` did not appear verbatim in {path}."
+#             )
+#         elif occurrences > 1:
+#             file_content_lines = file_content.split("\n")
+#             lines = [
+#                 idx + 1
+#                 for idx, line in enumerate(file_content_lines)
+#                 if old_str in line
+#             ]
+#             raise ToolError(
+#                 f"No replacement was performed. Multiple occurrences of old_str `{old_str}` in lines {lines}. Please ensure it is unique"
+#             )
+
+#         # Replace old_str with new_str
+#         new_file_content = file_content.replace(old_str, new_str)
+
+#         # Write the new content to the file
+#         self.write_file(path, new_file_content)
+
+#         # Save the content to history
+#         self._file_history[path].append(file_content)
+
+#         # Create a snippet of the edited section
+#         replacement_line = file_content.split(old_str)[0].count("\n")
+#         start_line = max(0, replacement_line - SNIPPET_LINES)
+#         end_line = replacement_line + SNIPPET_LINES + new_str.count("\n")
+#         snippet = "\n".join(new_file_content.split("\n")[start_line : end_line + 1])
+
+#         # Prepare the success message
+#         success_msg = f"The file {path} has been edited. "
+#         success_msg += self._make_output(
+#             snippet, f"a snippet of {path}", start_line + 1
+#         )
+#         success_msg += "Review the changes and make sure they are as expected. Edit the file again if necessary."
+
+#         return CLIResult(output=success_msg)
+
+#     def insert(self, path: Path, insert_line: int, new_str: str):
+#         """Implement the insert command, which inserts new_str at the specified line in the file content."""
+#         file_text = self.read_file(path).expandtabs()
+#         new_str = new_str.expandtabs()
+#         file_text_lines = file_text.split("\n")
+#         n_lines_file = len(file_text_lines)
+
+#         if insert_line < 0 or insert_line > n_lines_file:
+#             raise ToolError(
+#                 f"Invalid `insert_line` parameter: {insert_line}. It should be within the range of lines of the file: {[0, n_lines_file]}"
+#             )
+
+#         new_str_lines = new_str.split("\n")
+#         new_file_text_lines = (
+#             file_text_lines[:insert_line]
+#             + new_str_lines
+#             + file_text_lines[insert_line:]
+#         )
+#         snippet_lines = (
+#             file_text_lines[max(0, insert_line - SNIPPET_LINES) : insert_line]
+#             + new_str_lines
+#             + file_text_lines[insert_line : insert_line + SNIPPET_LINES]
+#         )
+
+#         new_file_text = "\n".join(new_file_text_lines)
+#         snippet = "\n".join(snippet_lines)
+
+#         self.write_file(path, new_file_text)
+#         self._file_history[path].append(file_text)
+
+#         success_msg = f"The file {path} has been edited. "
+#         success_msg += self._make_output(
+#             snippet,
+#             "a snippet of the edited file",
+#             max(1, insert_line - SNIPPET_LINES + 1),
+#         )
+#         success_msg += "Review the changes and make sure they are as expected (correct indentation, no duplicate lines, etc). Edit the file again if necessary."
+#         return CLIResult(output=success_msg)
+
+#     def undo_edit(self, path: Path):
+#         """Implement the undo_edit command."""
+#         if not self._file_history[path]:
+#             raise ToolError(f"No edit history found for {path}.")
+
+#         old_text = self._file_history[path].pop()
+#         self.write_file(path, old_text)
+
+#         return CLIResult(
+#             output=f"Last edit to {path} undone successfully. {self._make_output(old_text, str(path))}"
+#         )
+
+#     def read_file(self, path: Path):
+#         """Read the content of a file from a given path; raise a ToolError if an error occurs."""
+#         try:
+#             return path.read_text()
+#         except Exception as e:
+#             raise ToolError(f"Ran into {e} while trying to read {path}") from None
+
+#     def write_file(self, path: Path, file: str):
+#         """Write the content of a file to a given path; raise a ToolError if an error occurs."""
+#         try:
+#             path.write_text(file)
+#         except Exception as e:
+#             raise ToolError(f"Ran into {e} while trying to write to {path}") from None
+
+#     def _make_output(
+#         self,
+#         file_content: str,
+#         file_descriptor: str,
+#         init_line: int = 1,
+#         expand_tabs: bool = True,
+#     ):
+#         """Generate output for the CLI based on the content of a file."""
+#         file_content = maybe_truncate(file_content)
+#         if expand_tabs:
+#             file_content = file_content.expandtabs()
+#         file_content = "\n".join(
+#             [
+#                 f"{i + init_line:6}\t{line}"
+#                 for i, line in enumerate(file_content.split("\n"))
+#             ]
+#         )
+#         return (
+#             f"Here's the result of running `cat -n` on {file_descriptor}:\n"
+#             + file_content
+#             + "\n"
+#         )
+
+
+# class EditTool20250429(BaseAnthropicTool):
+#     """
+#     An filesystem editor tool that allows the agent to view, create, and edit files.
+#     The tool parameters are defined by Anthropic and are not editable.
+#     """
+
+#     api_type: Literal["str_replace_based_edit_tool"] = "str_replace_based_edit_tool"
+#     name: Literal["str_replace_based_edit_tool"] = "str_replace_based_edit_tool"
+
+#     _file_history: dict[Path, list[str]]
+
+#     def __init__(self):
+#         self._file_history = defaultdict(list)
+#         super().__init__()
+
+#     def to_params(self) -> Any:
+#         return {
+#             "name": self.name,
+#             "type": self.api_type,
+#         }
+
+#     async def __call__(
+#         self,
+#         *,
+#         command: Command_20250429,
+#         path: str,
+#         file_text: str | None = None,
+#         view_range: list[int] | None = None,
+#         old_str: str | None = None,
+#         new_str: str | None = None,
+#         insert_line: int | None = None,
+#         **kwargs,
+#     ):
+#         _path = Path(path)
+#         self.validate_path(command, _path)
+#         if command == "view":
+#             return await self.view(_path, view_range)
+#         elif command == "create":
+#             if file_text is None:
+#                 raise ToolError("Parameter `file_text` is required for command: create")
+#             self.write_file(_path, file_text)
+#             self._file_history[_path].append(file_text)
+#             return ToolResult(output=f"File created successfully at: {_path}")
+#         elif command == "str_replace":
+#             if old_str is None:
+#                 raise ToolError(
+#                     "Parameter `old_str` is required for command: str_replace"
+#                 )
+#             return self.str_replace(_path, old_str, new_str)
+#         elif command == "insert":
+#             if insert_line is None:
+#                 raise ToolError(
+#                     "Parameter `insert_line` is required for command: insert"
+#                 )
+#             if new_str is None:
+#                 raise ToolError("Parameter `new_str` is required for command: insert")
+#             return self.insert(_path, insert_line, new_str)
+#         # Note: undo_edit command was removed in this version
+#         raise ToolError(
+#             f"Unrecognized command {command}. The allowed commands for the {self.name} tool are: {', '.join(get_args(Command_20250429))}"
+#         )
+
+#     def validate_path(self, command: str, path: Path):
+#         """
+#         Check that the path/command combination is valid.
+#         """
+#         # Check if its an absolute path
+#         if not path.is_absolute():
+#             suggested_path = Path("") / path
+#             raise ToolError(
+#                 f"The path {path} is not an absolute path, it should start with `/`. Maybe you meant {suggested_path}?"
+#             )
+#         # Check if path exists
+#         if not path.exists() and command != "create":
+#             raise ToolError(
+#                 f"The path {path} does not exist. Please provide a valid path."
+#             )
+#         if path.exists() and command == "create":
+#             raise ToolError(
+#                 f"File already exists at: {path}. Cannot overwrite files using command `create`."
+#             )
+#         # Check if the path points to a directory
+#         if path.is_dir():
+#             if command != "view":
+#                 raise ToolError(
+#                     f"The path {path} is a directory and only the `view` command can be used on directories"
+#                 )
+
+#     async def view(self, path: Path, view_range: list[int] | None = None):
+#         """Implement the view command"""
+#         if path.is_dir():
+#             if view_range:
+#                 raise ToolError(
+#                     "The `view_range` parameter is not allowed when `path` points to a directory."
+#                 )
+
+#             _, stdout, stderr = await run(
+#                 rf"find {path} -maxdepth 2 -not -path '*/\.*'"
+#             )
+#             if not stderr:
+#                 stdout = f"Here's the files and directories up to 2 levels deep in {path}, excluding hidden items:\n{stdout}\n"
+#             return CLIResult(output=stdout, error=stderr)
+
+#         file_content = self.read_file(path)
+#         init_line = 1
+#         if view_range:
+#             if len(view_range) != 2 or not all(isinstance(i, int) for i in view_range):
+#                 raise ToolError(
+#                     "Invalid `view_range`. It should be a list of two integers."
+#                 )
+#             file_lines = file_content.split("\n")
+#             n_lines_file = len(file_lines)
+#             init_line, final_line = view_range
+#             if init_line < 1 or init_line > n_lines_file:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its first element `{init_line}` should be within the range of lines of the file: {[1, n_lines_file]}"
+#                 )
+#             if final_line > n_lines_file:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its second element `{final_line}` should be smaller than the number of lines in the file: `{n_lines_file}`"
+#                 )
+#             if final_line != -1 and final_line < init_line:
+#                 raise ToolError(
+#                     f"Invalid `view_range`: {view_range}. Its second element `{final_line}` should be larger or equal than its first `{init_line}`"
+#                 )
+
+#             if final_line == -1:
+#                 file_content = "\n".join(file_lines[init_line - 1 :])
+#             else:
+#                 file_content = "\n".join(file_lines[init_line - 1 : final_line])
+
+#         return CLIResult(
+#             output=self._make_output(file_content, str(path), init_line=init_line)
+#         )
+
+#     def str_replace(self, path: Path, old_str: str, new_str: str | None):
+#         """Implement the str_replace command, which replaces old_str with new_str in the file content"""
+#         # Read the file content
+#         file_content = self.read_file(path).expandtabs()
+#         old_str = old_str.expandtabs()
+#         new_str = new_str.expandtabs() if new_str is not None else ""
+
+#         # Check if old_str is unique in the file
+#         occurrences = file_content.count(old_str)
+#         if occurrences == 0:
+#             raise ToolError(
+#                 f"No replacement was performed, old_str `{old_str}` did not appear verbatim in {path}."
+#             )
+#         elif occurrences > 1:
+#             file_content_lines = file_content.split("\n")
+#             lines = [
+#                 idx + 1
+#                 for idx, line in enumerate(file_content_lines)
+#                 if old_str in line
+#             ]
+#             raise ToolError(
+#                 f"No replacement was performed. Multiple occurrences of old_str `{old_str}` in lines {lines}. Please ensure it is unique"
+#             )
+
+#         # Replace old_str with new_str
+#         new_file_content = file_content.replace(old_str, new_str)
+
+#         # Write the new content to the file
+#         self.write_file(path, new_file_content)
+
+#         # Save the content to history
+#         self._file_history[path].append(file_content)
+
+#         # Create a snippet of the edited section
+#         replacement_line = file_content.split(old_str)[0].count("\n")
+#         start_line = max(0, replacement_line - SNIPPET_LINES)
+#         end_line = replacement_line + SNIPPET_LINES + new_str.count("\n")
+#         snippet = "\n".join(new_file_content.split("\n")[start_line : end_line + 1])
+
+#         # Prepare the success message
+#         success_msg = f"The file {path} has been edited. "
+#         success_msg += self._make_output(
+#             snippet, f"a snippet of {path}", start_line + 1
+#         )
+#         success_msg += "Review the changes and make sure they are as expected. Edit the file again if necessary."
+
+#         return CLIResult(output=success_msg)
+
+#     def insert(self, path: Path, insert_line: int, new_str: str):
+#         """Implement the insert command, which inserts new_str at the specified line in the file content."""
+#         file_text = self.read_file(path).expandtabs()
+#         new_str = new_str.expandtabs()
+#         file_text_lines = file_text.split("\n")
+#         n_lines_file = len(file_text_lines)
+
+#         if insert_line < 0 or insert_line > n_lines_file:
+#             raise ToolError(
+#                 f"Invalid `insert_line` parameter: {insert_line}. It should be within the range of lines of the file: {[0, n_lines_file]}"
+#             )
+
+#         new_str_lines = new_str.split("\n")
+#         new_file_text_lines = (
+#             file_text_lines[:insert_line]
+#             + new_str_lines
+#             + file_text_lines[insert_line:]
+#         )
+#         snippet_lines = (
+#             file_text_lines[max(0, insert_line - SNIPPET_LINES) : insert_line]
+#             + new_str_lines
+#             + file_text_lines[insert_line : insert_line + SNIPPET_LINES]
+#         )
+
+#         new_file_text = "\n".join(new_file_text_lines)
+#         snippet = "\n".join(snippet_lines)
+
+#         self.write_file(path, new_file_text)
+#         self._file_history[path].append(file_text)
+
+#         success_msg = f"The file {path} has been edited. "
+#         success_msg += self._make_output(
+#             snippet,
+#             "a snippet of the edited file",
+#             max(1, insert_line - SNIPPET_LINES + 1),
+#         )
+#         success_msg += "Review the changes and make sure they are as expected (correct indentation, no duplicate lines, etc). Edit the file again if necessary."
+#         return CLIResult(output=success_msg)
+
+#     # Note: undo_edit method is not implemented in this version as it was removed
+
+#     def read_file(self, path: Path):
+#         """Read the content of a file from a given path; raise a ToolError if an error occurs."""
+#         try:
+#             return path.read_text()
+#         except Exception as e:
+#             raise ToolError(f"Ran into {e} while trying to read {path}") from None
+
+#     def write_file(self, path: Path, file: str):
+#         """Write the content of a file to a given path; raise a ToolError if an error occurs."""
+#         try:
+#             path.write_text(file)
+#         except Exception as e:
+#             raise ToolError(f"Ran into {e} while trying to write to {path}") from None
+
+#     def _make_output(
+#         self,
+#         file_content: str,
+#         file_descriptor: str,
+#         init_line: int = 1,
+#         expand_tabs: bool = True,
+#     ):
+#         """Generate output for the CLI based on the content of a file."""
+#         file_content = maybe_truncate(file_content)
+#         if expand_tabs:
+#             file_content = file_content.expandtabs()
+#         file_content = "\n".join(
+#             [
+#                 f"{i + init_line:6}\t{line}"
+#                 for i, line in enumerate(file_content.split("\n"))
+#             ]
+#         )
+#         return (
+#             f"Here's the result of running `cat -n` on {file_descriptor}:\n"
+#             + file_content
+#             + "\n"
+#         )
+
+
+# class EditTool20241022(EditTool20250124):
+#     api_type: Literal["text_editor_20250429"] = "text_editor_20250429"  # pyright: ignore[reportIncompatibleVariableOverride]

lm_deluge/built_in_tools/base.py

@@ -0,0 +1,9 @@
+class BuiltInTool:
+    def for_anthropic(self):
+        raise NotImplementedError("Not implemented")
+
+    def for_openai_chat(self):
+        raise NotImplementedError("Not implemented")
+
+    def for_openai_responses(self):
+        raise NotImplementedError("Not implemented")

lm_deluge/client.py

@@ -50,6 +50,7 @@ class LLMClient(BaseModel):
     max_attempts: int = 5
     request_timeout: int = 30
     cache: Any = None
+    extra_headers: dict[str, str] | None = None
     # sampling params - if provided, and sampling_params is not,
     # these override the defaults
     temperature: float = 0.75
@@ -364,6 +365,7 @@ class LLMClient(BaseModel):
                             tools=tools,
                             cache=cache,
                             use_responses_api=use_responses_api,
+                            extra_headers=self.extra_headers,
                         )
                     except StopIteration:
                         prompts_not_finished = False
@@ -451,13 +453,98 @@ class LLMClient(BaseModel):
         model, sampling_params = self._select_model()
         if isinstance(prompt, str):
             prompt = Conversation.user(prompt)
-        async for item in stream_chat(model, prompt, sampling_params, tools, None):
+        async for item in stream_chat(
+            model, prompt, sampling_params, tools, None, self.extra_headers
+        ):
             if isinstance(item, str):
                 print(item, end="", flush=True)
             else:
                 # final item
                 return item
 
+    async def run_agent_loop(
+        self,
+        conversation: str | Conversation,
+        *,
+        tools: list[Tool | dict] | None = None,
+        max_rounds: int = 5,
+        show_progress: bool = False,
+    ) -> tuple[Conversation, APIResponse]:
+        """Run a simple agent loop until no more tool calls are returned.
+
+        The provided ``conversation`` will be mutated and returned alongside the
+        final ``APIResponse`` from the model. ``tools`` may include ``Tool``
+        instances or built‑in tool dictionaries.
+        """
+
+        if isinstance(conversation, str):
+            conversation = Conversation.user(conversation)
+
+        last_response: APIResponse | None = None
+
+        for _ in range(max_rounds):
+            responses = await self.process_prompts_async(
+                [conversation],
+                tools=tools,  # type: ignore
+                return_completions_only=False,
+                show_progress=show_progress,
+            )
+
+            last_response = responses[0]
+            if last_response is None or last_response.content is None:
+                break
+
+            conversation.add(last_response.content)
+
+            tool_calls = last_response.content.tool_calls
+            if not tool_calls:
+                break
+
+            for call in tool_calls:
+                tool_obj = None
+                if tools:
+                    for t in tools:
+                        if isinstance(t, Tool) and t.name == call.name:
+                            tool_obj = t
+                            break
+
+                if isinstance(tool_obj, Tool) and tool_obj.run is not None:
+                    try:
+                        result = await tool_obj.acall(**call.arguments)
+                    except Exception as e:  # pragma: no cover - best effort
+                        result = f"Error: {e}"
+                else:
+                    result = f"Tool {call.name} not found"
+
+                if not isinstance(result, (str, dict, list)):
+                    result = str(result)
+
+                conversation.add_tool_result(call.id, result)  # type: ignore
+
+        if last_response is None:
+            raise RuntimeError("model did not return a response")
+
+        return conversation, last_response
+
+    def run_agent_loop_sync(
+        self,
+        conversation: str | Conversation,
+        *,
+        tools: list[Tool | dict | MCPServer] | None = None,
+        max_rounds: int = 5,
+        show_progress: bool = False,
+    ) -> tuple[Conversation, APIResponse]:
+        """Synchronous wrapper for :meth:`run_agent_loop`."""
+
+        return asyncio.run(
+            self.run_agent_loop(
+                conversation,
+                tools=tools,  # type: ignore
+                max_rounds=max_rounds,
+                show_progress=show_progress,
+            )
+        )
+
     async def submit_batch_job(
         self,
         prompts: Sequence[str | list[dict] | Conversation],