requirements.yaml

# SPDX-FileCopyrightText: Copyright (c) 2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

channels:
  - conda-forge
  - huggingface
  - pytorch
dependencies:
  - aiohttp-client-cache=0.11
  - aiohttp=3.9
  - beautifulsoup4=4.12
  # - faiss-gpu=1.7 # Uncomment this when the package supports CUDA 12. See: https://github.com/conda-forge/faiss-split-feedstock/pull/72
  - faiss=1.7
  - openai=1.13
  - pytorch=*=*cuda*
  - sentence-transformers>=2.0.0,<3.0.0
  - tiktoken>=0.3.2,<0.6.0
  - transformers

  ####### Pip Transitive Dependencies (keep sorted!) #######
  # These are dependencies that are available on conda, but are required by the pip packages listed below. Its much
  # better to install them with conda than pip to allow for better dependency resolution.
  - pydantic=2.6

  ####### Pip Dependencies (keep sorted!) #######
  - pip
  - pip:
    - google-search-results==2.4
    - langchain-nvidia-ai-endpoints==0.0.3
    - langchain==0.1.9
    - nemollm==0.3.5
    - pydpkg==1.9.2