From 48951dbabd4ca561063350df68b189673a2edefe Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Tue, 12 Nov 2024 16:11:30 -0800 Subject: [PATCH] try importing everything from extensions --- reflex/vars/function.py | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) diff --git a/reflex/vars/function.py b/reflex/vars/function.py index fb9b0d769..8fb6cec17 100644 --- a/reflex/vars/function.py +++ b/reflex/vars/function.py @@ -4,9 +4,22 @@ from __future__ import annotations import dataclasses import sys -from typing import Any, Generic, Optional, Sequence, Tuple, Type, Union, overload -from typing_extensions import Callable, Concatenate, ParamSpec, Protocol, TypeVar +from typing_extensions import ( + Any, + Callable, + Concatenate, + Generic, + Optional, + ParamSpec, + Protocol, + Sequence, + Tuple, + Type, + TypeVar, + Union, + overload, +) from reflex.utils import format from reflex.utils.types import GenericType @@ -458,7 +471,10 @@ class ArgsFunctionOperationBuilder(CachedVarOperation, BuilderFunctionVar): JSON_STRINGIFY = FunctionStringVar.create( "JSON.stringify", _var_type=ReflexCallable[[Any], str] ) -ARRAY_ISARRAY = FunctionStringVar.create("Array.isArray") -PROTOTYPE_TO_STRING = FunctionStringVar.create( - "((__to_string) => __to_string.toString())" +ARRAY_ISARRAY = FunctionStringVar.create( + "Array.isArray", _var_type=ReflexCallable[[Any], bool] +) +PROTOTYPE_TO_STRING = FunctionStringVar.create( + "((__to_string) => __to_string.toString())", + _var_type=ReflexCallable[[Any], str], )