pantry/projects/vanna.ai/package.yml

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