Fixes: #1867 InferenceRouter has no attribute formatter

This commit is contained in:
Rohan Awhad 2025-06-09 15:49:41 -04:00
parent 28ca00d0d9
commit 38faf57db6

View file

@ -163,6 +163,9 @@ class InferenceRouter(Inference):
messages: list[Message] | InterleavedContent, messages: list[Message] | InterleavedContent,
tool_prompt_format: ToolPromptFormat | None = None, tool_prompt_format: ToolPromptFormat | None = None,
) -> int | None: ) -> int | None:
if not hasattr(self, "formatter") or self.formatter is None:
return None
if isinstance(messages, list): if isinstance(messages, list):
encoded = self.formatter.encode_dialog_prompt(messages, tool_prompt_format) encoded = self.formatter.encode_dialog_prompt(messages, tool_prompt_format)
else: else: