Migrates ai/docbot to uv to manage dependencies (#832)

google · Aug 24, 2024 · 4087683 · 4087683
1 parent 2cb9343
commit 4087683
Show file tree

Hide file tree

Showing 4 changed files with 2,168 additions and 9 deletions.
diff --git a/ai/docbot/deploy_to_hf.sh b/ai/docbot/deploy_to_hf.sh
@@ -22,8 +22,11 @@ if [ ! -d "$DEST_PATH" ]; then
     mkdir -p "$DEST_PATH"
 fi
 
+# Install uv
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
 # Build the docs index
-cd ai/docbot && python docs_index.py --build-index && cd -
+cd ai/docbot && uv run docs_index.py --build-index && cd -
 
 # Get the path of this script which is the demo dir.
 DEMO_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

diff --git a/ai/docbot/pyproject.toml b/ai/docbot/pyproject.toml
@@ -0,0 +1,21 @@
+[project]
+name = "docbot"
+version = "0.1.0"
+description = "A chatbot for docs"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "gunicorn>=23.0.0",
+    "nest-asyncio>=1.6.0",
+    "llama-index==0.10.68",
+    "google-generativeai>=0.5.4",
+    "llama-index-llms-gemini==0.2.0",
+    "llama-index-embeddings-google==0.1.6",
+    "llama-index-retrievers-bm25==0.2.2",
+    "mesop>=0.12.2",
+]
+
+[tool.uv]
+dev-dependencies = []
+
+[tool.uv.workspace]
diff --git a/ai/docbot/requirements.txt b/ai/docbot/requirements.txt