allow other function names

This commit is contained in:
balalo
2025-03-18 20:45:31 +01:00
parent dc5067407d
commit 48b6850df4
2 changed files with 2 additions and 2 deletions

View File

@@ -367,7 +367,7 @@ class LLMSetToolsFrame(DataFrame):
class LLMSetToolChoiceFrame(DataFrame):
"""A frame containing a tool choice for an LLM to use for function calling."""
tool_choice: Literal["none", "auto", "required"]
tool_choice: Literal["none", "auto", "required"] | str
@dataclass

View File

@@ -253,7 +253,7 @@ class LLMContextResponseAggregator(BaseLLMResponseAggregator):
def set_tools(self, tools: List):
self._context.set_tools(tools)
def set_tool_choice(self, tool_choice: Literal["none", "auto", "required"]):
def set_tool_choice(self, tool_choice: Literal["none", "auto", "required"] | str):
self._context.set_tool_choice(tool_choice)
def reset(self):