fix: allow variable message to be any (#13494)

1 year ago · 50cfb7c9ec
parent 8281c688ca
commit 50cfb7c9ec
1 changed files with 1 additions and 1 deletions
--- a/api/core/tools/entities/tool_entities.py
+++ b/api/core/tools/entities/tool_entities.py
@ -125,7 +125,7 @@ class ToolInvokeMessage(BaseModel):

    class VariableMessage(BaseModel):
        variable_name: str = Field(..., description="The name of the variable")
-        variable_value: str = Field(..., description="The value of the variable")
+        variable_value: Any = Field(..., description="The value of the variable")
        stream: bool = Field(default=False, description="Whether the variable is streamed")

        @model_validator(mode="before")