fix pyright

Author: adhami3310

packages/reflex-core/src/reflex_core/vars/base.py

@@ -37,7 +37,7 @@
 )
 
 from rich.markup import escape
-from typing_extensions import dataclass_transform, override
+from typing_extensions import LiteralString, dataclass_transform, override
 
 from reflex_core import constants
 from reflex_core.constants.compiler import Hooks
@@ -85,6 +85,7 @@
 VAR_TYPE = TypeVar("VAR_TYPE", covariant=True)
 OTHER_VAR_TYPE = TypeVar("OTHER_VAR_TYPE")
 STRING_T = TypeVar("STRING_T", bound=str)
+LITERAL_STRING_T = TypeVar("LITERAL_STRING_T", bound=LiteralString)
 SEQUENCE_TYPE = TypeVar("SEQUENCE_TYPE", bound=Sequence)
 
 warnings.filterwarnings("ignore", message="fields may not start with an underscore")
@@ -651,17 +652,17 @@ def create(  # pyright: ignore [reportOverlappingOverload]
     @classmethod
     def create(  # pyright: ignore [reportOverlappingOverload]
         cls,
-        value: str,
+        value: LITERAL_STRING_T,
         _var_data: VarData | None = None,
-    ) -> LiteralStringVar: ...
+    ) -> LiteralStringVar[LITERAL_STRING_T]: ...
 
     @overload
     @classmethod
     def create(  # pyright: ignore [reportOverlappingOverload]
         cls,
         value: STRING_T,
         _var_data: VarData | None = None,
-    ) -> StringVar[STRING_T]: ...
+    ) -> LiteralStringVar[STRING_T]: ...
 
     @overload
     @classmethod

packages/reflex-core/src/reflex_core/vars/sequence.py

@@ -573,7 +573,7 @@ def create(
         )
 
 
-STRING_TYPE = TypingExtensionsTypeVar("STRING_TYPE", default=str)
+STRING_TYPE = TypingExtensionsTypeVar("STRING_TYPE", default=str, covariant=True)
 
 
 class StringVar(Var[STRING_TYPE], python_types=str):
@@ -1153,7 +1153,7 @@ def get_decimal_string_operation(
     frozen=True,
     slots=True,
 )
-class LiteralStringVar(LiteralVar[str], StringVar[str]):
+class LiteralStringVar(LiteralVar[STRING_TYPE], StringVar[STRING_TYPE]):
     """Base class for immutable literal string vars."""
 
     _var_value: str = dataclasses.field(default="")

tests/units/components/core/test_cond.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any
+from typing import Any, Literal
 
 import pytest
 from reflex_components_core.base.fragment import Fragment
@@ -184,7 +184,7 @@ def test_cond_assert_types() -> None:
     _ = assert_type(cond(True, "hello", var_int), Var[str | int])
 
     # Var[T], U -> Var[T | U]
-    _ = assert_type(cond(True, var_str, 3), Var[str | int])
+    _ = assert_type(cond(True, var_str, 3), Var[int | Literal["a"]])
 
     # Var[T], Var[U] -> Var[T | U]
-    _ = assert_type(cond(True, var_int, var_str), Var[int | str])
+    _ = assert_type(cond(True, var_int, var_str), Var[int | Literal["a"]])

tests/units/experimental/test_memo.py

@@ -117,7 +117,7 @@ def conditional_slot(
         show: rx.Var[bool],
         first: rx.Var[rx.Component],
         second: rx.Var[rx.Component],
-    ) -> rx.Component:
+    ) -> rx.Var[rx.Component]:
         return rx.cond(show, first, second)
 
     definition = EXPERIMENTAL_MEMOS["ConditionalSlot"]