use use

src/react.ts

@@ -1,5 +1,3 @@
-/// <reference types="react/experimental" />
-
 import { useCallback, useDebugValue, useEffect, useMemo, useRef } from 'react'
 import {
   affectedToPathList,

tests/async.test.tsx

@@ -1,3 +1,5 @@
+/// <reference types="react/canary" />
+
 import { StrictMode, Suspense, use } from 'react'
 import { fireEvent, render, waitFor } from '@testing-library/react'
 import { it } from 'vitest'
@@ -8,7 +10,7 @@ const sleep = (ms: number) =>
     setTimeout(resolve, ms)
   })
 
-const use2 = <T,>(x: T): T => (x instanceof Promise ? use(x) : x)
+const use2 = <T,>(x: T): Awaited<T> => (x instanceof Promise ? use(x) : x)
 
 it('delayed increment', async () => {
   const state = proxy<any>({ count: 0 })

tests/computed.test.tsx

@@ -1,10 +1,14 @@
-import { StrictMode, Suspense } from 'react'
+/// <reference types="react/canary" />
+
+import { StrictMode, Suspense, use } from 'react'
 import { fireEvent, render } from '@testing-library/react'
 import { memoize } from 'proxy-memoize'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 import { proxy, snapshot, subscribe, useSnapshot } from 'valtio'
 import { addComputed, proxyWithComputed, subscribeKey } from 'valtio/utils'
 
+const use2 = <T,>(x: T): Awaited<T> => (x instanceof Promise ? use(x) : x)
+
 const consoleWarn = console.warn
 beforeEach(() => {
   console.warn = vi.fn((message: string) => {
@@ -217,7 +221,7 @@ describe('DEPRECATED addComputed', () => {
       return (
         <>
           <div>
-            count: {snap.count}, delayedCount: {snap.delayedCount}
+            count: {snap.count}, delayedCount: {use2(snap.delayedCount)}
           </div>
           <button onClick={() => ++state.count}>button</button>
         </>

tests/derive.test.tsx

@@ -1,4 +1,6 @@
-import { StrictMode, Suspense, useEffect, useRef } from 'react'
+/// <reference types="react/canary" />
+
+import { StrictMode, Suspense, use, useEffect, useRef } from 'react'
 import { fireEvent, render } from '@testing-library/react'
 import { describe, expect, it, vi } from 'vitest'
 import { proxy, snapshot, subscribe, useSnapshot } from 'valtio'
@@ -11,6 +13,8 @@ const sleep = (ms: number) =>
     setTimeout(resolve, ms)
   })
 
+const use2 = <T,>(x: T): Awaited<T> => (x instanceof Promise ? use(x) : x)
+
 it('basic derive', async () => {
   const computeDouble = vi.fn((x: number) => x * 2)
   const state = proxy({
@@ -168,7 +172,7 @@ it('async derive', async () => {
     return (
       <>
         <div>
-          count: {snap.count}, delayedCount: {snap.delayedCount}
+          count: {snap.count}, delayedCount: {use2(snap.delayedCount)}
         </div>
         <button onClick={() => ++state.count}>button</button>
       </>