diff --git a/.github/workflows/install.yaml b/.github/workflows/install.yaml index dc7d14e..5932762 100644 --- a/.github/workflows/install.yaml +++ b/.github/workflows/install.yaml @@ -33,4 +33,7 @@ jobs: python -m pip check - name: Install add ons run: | - python -m pip install "unpacked_sdist/[llm]" \ No newline at end of file + python -m pip install "unpacked_sdist/[llm]" + - name: Test by importing packages + run: | + python -c "import openai" \ No newline at end of file diff --git a/trane/llm/helpers.py b/trane/llm/helpers.py index b5a82eb..999b6e1 100644 --- a/trane/llm/helpers.py +++ b/trane/llm/helpers.py @@ -6,7 +6,6 @@ openai = import_or_none("openai") tiktoken = import_or_none("tiktoken") -anthropic = import_or_none("anthropic") ipython = import_or_none("IPython")