allow other function names
This commit is contained in:
@@ -367,7 +367,7 @@ class LLMSetToolsFrame(DataFrame):
|
|||||||
class LLMSetToolChoiceFrame(DataFrame):
|
class LLMSetToolChoiceFrame(DataFrame):
|
||||||
"""A frame containing a tool choice for an LLM to use for function calling."""
|
"""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
|
@dataclass
|
||||||
|
|||||||
@@ -253,7 +253,7 @@ class LLMContextResponseAggregator(BaseLLMResponseAggregator):
|
|||||||
def set_tools(self, tools: List):
|
def set_tools(self, tools: List):
|
||||||
self._context.set_tools(tools)
|
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)
|
self._context.set_tool_choice(tool_choice)
|
||||||
|
|
||||||
def reset(self):
|
def reset(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user