# # environment_setup.py
# import subprocess
# import sys
# import os

# def install_packages():
#     packages = [
#         "einops==0.7.0",
#         "protobuf==5.26.1",
#         "llama-index==0.11.20",
#         "llama-index-llms-replicate==0.3.0",
#         "llama-index-llms-openai-like==0.2.0",
#         "llama-index-embeddings-huggingface==0.3.1",
#         "llama-index-embeddings-instructor==0.2.1",
#         "torch==2.5.0",
#         "torchvision==0.20.0",
#         "torchaudio==2.5.0"
#     ]
#     index_url = "https://download.pytorch.org/whl/cu121"  # 根据你的 CUDA 版本调整

#     for package in packages:
#         try:
#             if package.startswith("torch") or package.startswith("torchvision") or package.startswith("torchaudio"):
#                 subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}", "--index-url", index_url, "-q"])
#             else:
#                 subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}", "-q"])
#             print(f"Successfully installed {package}")
#         except subprocess.CalledProcessError as e:
#             print(f"Error installing {package}: {e}")

# if __name__ == "__main__":
#     print("Starting to install required packages...")
#     install_packages()
#     print("Package installation completed.")