allow other function names
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user