Internal clean up.

PiperOrigin-RevId: 916121032

Author: junjiang-lab

litert_cli/commands/run/desktop.py

@@ -275,8 +275,8 @@ def run_desktop(
   with ctx:
     try:
       env = None
-      if constants.IN_GOOGLE3:
-        # In Google3 environment, we need to fallback to LD_LIBRARY_PATH for
+      if constants.IS_INTERNAL_ENV:
+        # In internal environment, we need to fallback to LD_LIBRARY_PATH for
         # loading GPU accelerators in hermetic .par file. Otherwise, use
         # default path.
         env = Environment.create(runtime_path="")

litert_cli/core/constants.py

@@ -20,18 +20,12 @@
 import os
 import types
 
-# Check if we are in Google3 environment
-try:
-  import google3  # pylint: disable=unused-import,g-import-not-at-top
-
-  _AUTO_IN_GOOGLE3 = True
-except ImportError:
-  _AUTO_IN_GOOGLE3 = False
+# Flag to indicate if the CLI is running in internal environment
+IS_INTERNAL_ENV: bool = False
 
 # Environment variable names
 ENV_LITERT_CLI_ROOT: str = "LITERT_CLI_ROOT"
 ENV_LITERT_CLI_ANDROID_ROOT: str = "LITERT_CLI_ANDROID_ROOT"
-ENV_LITERT_CLI_FORCE_OSS: str = "LITERT_CLI_FORCE_OSS"
 ENV_LITERT_VERBOSE: str = "LITERT_VERBOSE"
 ENV_LITERT_ENABLE_MODEL_PLUGINS: str = "LITERT_ENABLE_MODEL_PLUGINS"
 
@@ -40,14 +34,6 @@
     os.environ.get(ENV_LITERT_ENABLE_MODEL_PLUGINS, "0") == "1"
 )
 
-_FORCE_OSS = os.environ.get(ENV_LITERT_CLI_FORCE_OSS, "").lower() in (
-    "1",
-    "true",
-    "yes",
-)
-
-IN_GOOGLE3: bool = _AUTO_IN_GOOGLE3 and not _FORCE_OSS
-
 # Cache directory
 LITERT_CLI_CACHE_DIR: str = os.path.join(
     os.path.expanduser("~"), ".cache", "litert-cli"

litert_cli/core/deps.py

@@ -70,7 +70,7 @@ def ensure_extra(extra_name: str, *, silent: bool = False) -> bool:
     click.Abort: If `silent` is False and the dependency cannot be ensured
       (e.g., unknown extra or installation failure).
   """
-  if constants.IN_GOOGLE3:
+  if constants.IS_INTERNAL_ENV:
     return True
 
   packages_to_check = _PACKAGE_BY_EXTRA.get(extra_name)

litert_cli/core/deps_test.py

@@ -27,7 +27,7 @@ class DepsTest(absltest.TestCase):
 
   def setUp(self):
     super().setUp()
-    self.enter_context(mock.patch.object(constants, 'IN_GOOGLE3', False))
+    self.enter_context(mock.patch.object(constants, 'IS_INTERNAL_ENV', False))
 
   @mock.patch.object(importlib.metadata, 'version', autospec=True)
   def test_ensure_extra_already_installed(self, mock_version):

litert_cli/litert.py

@@ -110,7 +110,7 @@ def main_bridge(unused_argv):
 
 if __name__ == '__main__':
   try:
-    # Try to run in google3
+    # Try to run in internal environment
     from absl import app  # pylint: disable=g-import-not-at-top
 
     app.run(main_bridge, argv=[sys.argv[0]])