AGBenchmark v0.0.10 compatible?

autogpts/autogpt/pyproject.toml

@@ -23,6 +23,8 @@ serve = "autogpt.app.cli:serve"
 
 [tool.poetry.dependencies]
 python = "^3.10"
+# autogpt-forge = { path = "../forge" }
+autogpt-forge = {git = "https://github.com/Significant-Gravitas/AutoGPT.git", subdirectory = "autogpts/forge", rev = "10aecec"}
 beautifulsoup4 = "^4.12.2"
 charset-normalizer = "^3.1.0"
 click = "*"
@@ -31,8 +33,7 @@ distro = "^1.8.0"
 docker = "*"
 duckduckgo-search = "^3.0.2"
 en-core-web-sm = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.5.0/en_core_web_sm-3.5.0-py3-none-any.whl"}
-# autogpt-forge = { path = "../forge" }
-autogpt-forge = {git = "https://github.com/Significant-Gravitas/AutoGPT.git", subdirectory = "autogpts/forge", rev = "10aecec"}
+fastapi = "*"
 ftfy = "^6.1.1"
 google-api-python-client = "*"
 gTTS = "^2.3.1"
@@ -61,17 +62,18 @@ spacy = "^3.0.0"
 tiktoken = "^0.5.0"
 webdriver-manager = "*"
 
-# web server
-fastapi = "*"
-uvicorn = "*"
-
 # OpenAI and Generic plugins import
 openapi-python-client = "^0.14.0"
 
+# Benchmarking
+agbenchmark = { path = "../../benchmark", optional = true }
+
+[tool.poetry.extras]
+benchmark = ["agbenchmark"]
+
 [tool.poetry.group.dev.dependencies]
 auto-gpt-plugin-template = {git = "https://github.com/Significant-Gravitas/Auto-GPT-Plugin-Template", rev = "0.1.0"}
 black = "*"
-coverage = "*"
 flake8 = "*"
 gitpython = "^3.1.32"
 isort = "*"
@@ -82,8 +84,9 @@ types-colorama = "*"
 types-Markdown = "*"
 types-Pillow = "*"
 
-# Testing dependencies
+# Testing
 asynctest = "*"
+coverage = "*"
 pytest = "*"
 pytest-asyncio = "*"
 pytest-benchmark = "*"

autogpts/autogpt/run

@@ -0,0 +1,10 @@
+#!/bin/sh
+
+kill $(lsof -t -i :8000)
+
+if [ ! -f .env ]; then
+  cp .env.example .env
+  echo "Please add your api keys to the .env file." >&2
+  # exit 1
+fi
+poetry run serve

autogpts/autogpt/run_benchmark

@@ -0,0 +1,9 @@
+#!/bin/sh
+
+# Kill processes using port 8080 if any.
+if lsof -t -i :8080; then
+    kill $(lsof -t -i :8080)
+fi
+# This is the cli entry point for the benchmarking tool.
+# To run this in server mode pass in `serve` as the first argument.
+poetry run agbenchmark "$@"

autogpts/autogpt/setup

@@ -0,0 +1,4 @@
+#!/bin/sh
+
+poetry install --no-interaction --extras benchmark
+echo "Setup completed successfully."