-
Couldn't load subscription status.
- Fork 2.8k
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
Bug Description
Running a Python script in Google Colab using litellm==1.48.1 throws ModuleNotFoundError: No module named 'enterprise' when importing LitellmModel from agents.extensions.models.litellm_model. The same script worked fine the previous night, suggesting a recent environment or dependency change.
Additional Context
- The error originates from
litellm/responses/litellm_completion_transformation/transformation.py, attempting to import fromenterprise.enterprise_hooks.session_handler. - Similar issue reported in #5909 — likely related to a missing or premium
enterprisemodule. - Colab’s runtime may have reset, causing package mismatches.
Possible Causes
- Missing Enterprise Module:
enterprisemight be a separate (undocumented) package likelitellm-proxy-enterprise. - New Dependency:
litellm==1.48.1may have introduced this dependency unexpectedly. - Colab Reset: Temporary Colab environment reset could have lost critical packages.
- Undocumented Feature:
enterprisemight belong to a premium feature and should be handled optionally.
Suggested Fix
- Clarify if
enterpriseis required for standard usage or is a separate installation. - If optional, update
litellmto handle missing imports gracefully (e.g.,try-exceptblock). - Recommend a stable, verified
litellmversion or pin dependencies to avoid such issues. - Provide clear installation guidance for any
enterprise-related modules.
Workarounds Attempted
- Reinstalled
litellm==1.48.1andagents; error persists. - Tried
!pip install litellm-proxy-enterprise; package not found on PyPI. - Downgrading to
litellm==1.40.14(which might avoid the dependency) was considered but not yet tested.
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working