pantry/projects/vanna.ai/package.yml
Jacob Heider a9c4623960 fix(vanna)
closes #6359
2024-06-07 12:41:33 -04:00

70 lines
2.2 KiB
YAML

distributable:
url: git+https://github.com/vanna-ai/vanna.git
ref: ${{version.tag}}
display-name: Vanna
versions:
github: vanna-ai/vanna
# /Users/runner/work/pantry/pantry/testbeds/vanna.ai-0.2.1.sh: line 38: 4897 Illegal instruction: 4 python test_vanna.py
platforms:
- darwin/aarch64
- linux
dependencies:
python.org: ^3.12
linux:
openmp.llvm.org: 17 # needed by chromadb
runtime:
env:
PYTHONPATH: ${{prefix}}/lib/python{{deps.python.org.version.major}}/site-packages:$PYTHONPATH
build:
script:
- python -m pip install . --prefix={{prefix}}
# additional requirements as of 0.1.0
- run: python -m pip install openai mistralai --prefix={{prefix}}
if: '>=0.1'
# additional requirements as of 0.2.0
- run: python -m pip install chromadb python-dotenv --prefix={{prefix}}
if: '>=0.2'
# additional requirements as of 0.3.0
- run: python -m pip install anthropic --prefix={{prefix}}
if: '>=0.3'
# additional requirements as of 0.4.0
- run: python -m pip install google-generativeai --prefix={{prefix}}
if: '>=0.4'
# additional requirements as of 0.4.1
- run: python -m pip install qdrant-client fastembed --prefix={{prefix}}
if: '>=0.4.1'
# protobuf ^5.27 (transient dep) currently not playing nicely with other components
- run: |
rm -rf ${{prefix}}/lib/python{{deps.python.org.version.marketing}}/site-packages/proto
python -m pip install proto-plus==1.24.0.dev1 --prefix={{prefix}}
if: 0.6.0
- run: ln -s python{{deps.python.org.version.marketing}} python{{deps.python.org.version.major}}
working-directory: ${{prefix}}/lib
env:
linux:
CC: clang
CXX: clang++
LD: clang
test:
dependencies:
curl.se: '*'
env:
VANNA_API_KEY: FAKE_API_KEY
OPENAI_API_KEY: FAKE_API_KEY
MISTRAL_API_KEY: FAKE_API_KEY
ANTHROPIC_API_KEY: FAKE_API_KEY
SNOWFLAKE_ACCOUNT: FAKE_ACCOUNT
SNOWFLAKE_USERNAME: FAKE_USERNAME
SNOWFLAKE_PASSWORD: FAKE_PASSWORD
GEMINI_API_KEY: FAKE_API_KEY
script:
- curl -L "https://raw.githubusercontent.com/vanna-ai/vanna/{{ version.tag }}/tests/test_vanna.py" -o test_vanna.py
- python test_vanna.py