Add type parameter for key to createContext() (#3628)

lit · Feb 2, 2023 · 61ce814 · 61ce814
1 parent 4d69843
commit 61ce814
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/.changeset/happy-apricots-pay.md b/.changeset/happy-apricots-pay.md
@@ -0,0 +1,5 @@
+---
+'@lit-labs/context': patch
+---
+
+Add type parameter for key to createContext()
diff --git a/packages/labs/context/src/lib/create-context.ts b/packages/labs/context/src/lib/create-context.ts
@@ -49,8 +49,8 @@ export type ContextType<Key extends Context<unknown, unknown>> =
  *
  * @param key a context key value
  * @template ValueType the type of value that can be provided by this context.
- * @returns the context key value with the correct type
+ * @returns the context key value cast to `Context<K, ValueType>`
  */
-export function createContext<ValueType>(key: unknown) {
-  return key as Context<typeof key, ValueType>;
+export function createContext<ValueType, K = unknown>(key: K) {
+  return key as Context<K, ValueType>;
 }