hf -> hf-embed

This commit is contained in:
Paul Gauthier 2024-07-10 17:26:56 +01:00
parent f688c18b65
commit ea89ca6eb5
5 changed files with 11 additions and 10 deletions

View file

@ -92,7 +92,7 @@ class PipInstallHF(Exception):
pip_install_cmd = [
"aider[hf]",
"aider-chat[hf-embed]",
"--extra-index-url",
"https://download.pytorch.org/whl/cpu",
]
@ -102,7 +102,7 @@ To use interactive /help you need to install HuggingFace embeddings:
pip install {' '.join(pip_install_cmd)}
"""
""" # noqa: E231
class Help:
@ -131,7 +131,7 @@ class Help:
# Relevant docs:
"""
""" # noqa: E231
for node in nodes:
url = node.metadata.get("url", "")

View file

@ -5,5 +5,5 @@ set -e
pip-compile requirements.in $1
pip-compile --output-file=requirements-dev.txt requirements-dev.in $1
pip-compile --output-file=requirements-hf.txt requirements-hf.in $1
pip-compile --output-file=requirements-hf-embed.txt requirements-hf-embed.in $1

View file

@ -5,16 +5,17 @@ from setuptools import find_packages, setup
from aider import __version__
from aider.help_pats import exclude_website_pats
def get_requirements(suffix=''):
fname = "requirements" + suffix + '.txt'
def get_requirements(suffix=""):
fname = "requirements" + suffix + ".txt"
with open(fname) as f:
requirements = f.read().splitlines()
return requirements
requirements = get_requirements()
dev_requirements = get_requirements('-dev')
hf_requirements = get_requirements('-hf')
dev_requirements = get_requirements("-dev")
hf_requirements = get_requirements("-hf-embed")
# README
with open("README.md", "r", encoding="utf-8") as f:
@ -38,8 +39,8 @@ setup(
exclude_package_data={"aider.website": exclude_website_pats},
install_requires=requirements,
extras_require={
'dev': dev_requirements,
'hf': hf_requirements,
"dev": dev_requirements,
"hf-embed": hf_requirements,
},
python_requires=">=3.9,<3.13",
entry_points={