Introduce lf.ModelInfo.

PiperOrigin-RevId: 724599567
google · Feb 11, 2025 · 7e76c24 · 7e76c24
1 parent 9065143
commit 7e76c24
Show file tree

Hide file tree

Showing 22 changed files with 3,131 additions and 1,578 deletions.
diff --git a/langfun/core/__init__.py b/langfun/core/__init__.py
@@ -93,14 +93,18 @@
 from langfun.core.modality import ModalityError
 
 # Interfaces for languge models.
+from langfun.core.language_model import ModelInfo
 from langfun.core.language_model import LanguageModel
+
 from langfun.core.language_model import LMSample
 from langfun.core.language_model import LMSamplingOptions
+from langfun.core.language_model import LMSamplingResult
+from langfun.core.language_model import LMScoringResult
+
 from langfun.core.language_model import LMSamplingUsage
 from langfun.core.language_model import UsageNotAvailable
 from langfun.core.language_model import UsageSummary
-from langfun.core.language_model import LMSamplingResult
-from langfun.core.language_model import LMScoringResult
+
 from langfun.core.language_model import LMCache
 from langfun.core.language_model import LMDebugMode