diff --git a/packages/angular/src/lib/provider.ts b/packages/angular/src/lib/provider.ts
index 7363fe3c..8a208b95 100644
--- a/packages/angular/src/lib/provider.ts
+++ b/packages/angular/src/lib/provider.ts
@@ -48,6 +48,8 @@ import {
   getTranslation,
   getBehavior,
   type CountryData,
+  type TranslationCategory,
+  type TranslationKey,
 } from "@firebase-oss/ui-core";
 
 const FIREBASE_UI_STORE = new InjectionToken<FirebaseUIStore>("firebaseui.store");
@@ -185,9 +187,9 @@ export function injectRecaptchaVerifier(element: () => ElementRef<HTMLDivElement
  * @param key - The translation key within the category.
  * @returns A computed signal containing the translated string.
  */
-export function injectTranslation(category: string, key: string) {
+export function injectTranslation<T extends TranslationCategory>(category: T, key: TranslationKey<T>) {
   const ui = injectUI();
-  return computed(() => getTranslation(ui(), category as any, key as any));
+  return computed(() => getTranslation(ui(), category, key));
 }
 
 /**
diff --git a/packages/core/src/translations.ts b/packages/core/src/translations.ts
index 0bf63ece..5743afe4 100644
--- a/packages/core/src/translations.ts
+++ b/packages/core/src/translations.ts
@@ -21,6 +21,9 @@ import {
 } from "@firebase-oss/ui-translations";
 import { type FirebaseUI } from "./config";
 
+// Re-export types for use in other packages
+export type { TranslationCategory, TranslationKey };
+
 /**
  * Gets a translated string for a given category and key.
  *