Skip to content

Commit

Permalink
chore: add pip-audit
Browse files Browse the repository at this point in the history
  • Loading branch information
tonykchen committed Apr 12, 2024
1 parent 45a8c37 commit fc25322
Show file tree
Hide file tree
Showing 4 changed files with 26 additions and 29 deletions.
4 changes: 3 additions & 1 deletion .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,4 +25,6 @@ jobs:
- name: Unit test
run: python -m pytest --junitxml=junit/test-results.xml --cov=src --cov-report=xml --cov-report=html
- name: Bandit
run: bandit -r src/
run: bandit -r src/
- name: pip-audit
run: pip-audit -r requirements.txt
12 changes: 12 additions & 0 deletions requirements-dev.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
flake8
black
isort
pytest
pytest-cov
pytest-mock
mkdocs
mkdocs-material
mkdocstrings[python]
mkdocs-click
bandit
pip-audit
7 changes: 7 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
pyyaml~=6.0
boto3>=1.34.20,<2.0
click~=8.0
pydantic>=2.1.0,<3.0
rich>=13.7.0,<14.0
jinja2>=3.1.3,<4.0
jsonpath-ng>=1.6.1,<2.0
32 changes: 4 additions & 28 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,37 +4,12 @@
from setuptools import find_packages

VERSION = "0.1.0"

DESCRIPTION = "A LLM-powered framework for testing virtual agents."
AUTHOR = "Amazon Web Services"
EMAIL = "[email protected]"
URL = "https://github.com/awslabs/agent-evaluation"

PACKAGE_DIR = "src"
REQUIRES_PYTHON = ">=3.9"
REQUIREMENTS = [
"pyyaml~=6.0",
"boto3>=1.34.20,<2.0",
"click~=8.0",
"pydantic>=2.1.0,<3.0",
"rich>=13.7.0,<14.0",
"jinja2>=3.1.3,<4.0",
"jsonpath-ng>=1.6.1,<2.0",
]
DEV_REQUIREMENTS = [
"flake8",
"black",
"isort",
"pytest",
"pytest-cov",
"pytest-mock",
"mkdocs",
"mkdocs-material",
"mkdocstrings[python]",
"mkdocs-click",
"bandit",
]

PACKAGE_DATA = {
"": [
"templates/**/*",
Expand All @@ -43,7 +18,8 @@


def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
with open(os.path.join(os.path.dirname(__file__), fname)) as f:
return f.read()


setup(
Expand All @@ -55,8 +31,8 @@ def read(fname):
description=DESCRIPTION,
long_description=read("README.md"),
python_requires=REQUIRES_PYTHON,
install_requires=REQUIREMENTS,
extras_require={"dev": DEV_REQUIREMENTS},
install_requires=read("requirements.txt").splitlines(),
extras_require={"dev": read("requirements-dev.txt").splitlines()},
entry_points={"console_scripts": ["agenteval=agenteval.cli:cli"]},
author=AUTHOR,
author_email=EMAIL,
Expand Down

0 comments on commit fc25322

Please sign in to comment.