From 399461dee290aa2df432008fd8d51953d4696f96 Mon Sep 17 00:00:00 2001
From: raspawar <raspawar@nvidia.com>
Date: Mon, 24 Mar 2025 15:23:04 +0530
Subject: [PATCH] add aiohttp as unit, integration dependency

---
 pyproject.toml | 3 ++-
 uv.lock        | 6 ++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 999b2fd72..0b34a601c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -56,7 +56,7 @@ dev = [
     "ruamel.yaml",      # needed for openapi generator
 ]
 # These are the dependencies required for running unit tests.
-unit = ["sqlite-vec", "openai", "aiosqlite", "pypdf", "chardet", "qdrant-client"]
+unit = ["sqlite-vec", "openai", "aiosqlite", "aiohttp", "pypdf", "chardet", "qdrant-client"]
 # These are the core dependencies required for running integration tests. They are shared across all
 # providers. If a provider requires additional dependencies, please add them to your environment
 # separately. If you are using "uv" to execute your tests, you can use the "--with" flag to specify extra
@@ -64,6 +64,7 @@ unit = ["sqlite-vec", "openai", "aiosqlite", "pypdf", "chardet", "qdrant-client"
 test = [
     "openai",
     "aiosqlite",
+    "aiohttp",
     "torch>=2.6.0",
     "torchvision>=0.21.0",
     "opentelemetry-sdk",
diff --git a/uv.lock b/uv.lock
index 7d5e250cb..b1c95c511 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1,4 +1,5 @@
 version = 1
+revision = 1
 requires-python = ">=3.10"
 resolution-markers = [
     "(python_full_version < '3.11' and platform_machine != 'aarch64' and sys_platform == 'linux') or (python_full_version < '3.11' and sys_platform != 'darwin' and sys_platform != 'linux')",
@@ -1370,6 +1371,7 @@ docs = [
     { name = "tomli" },
 ]
 test = [
+    { name = "aiohttp" },
     { name = "aiosqlite" },
     { name = "autoevals" },
     { name = "chardet" },
@@ -1385,6 +1387,7 @@ test = [
     { name = "torchvision", version = "0.21.0+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine != 'aarch64' and sys_platform == 'linux') or (sys_platform != 'darwin' and sys_platform != 'linux')" },
 ]
 unit = [
+    { name = "aiohttp" },
     { name = "aiosqlite" },
     { name = "chardet" },
     { name = "openai" },
@@ -1395,6 +1398,8 @@ unit = [
 
 [package.metadata]
 requires-dist = [
+    { name = "aiohttp", marker = "extra == 'test'" },
+    { name = "aiohttp", marker = "extra == 'unit'" },
     { name = "aiosqlite", marker = "extra == 'test'" },
     { name = "aiosqlite", marker = "extra == 'unit'" },
     { name = "autoevals", marker = "extra == 'test'" },
@@ -1455,6 +1460,7 @@ requires-dist = [
     { name = "types-setuptools", marker = "extra == 'dev'" },
     { name = "uvicorn", marker = "extra == 'dev'" },
 ]
+provides-extras = ["dev", "unit", "test", "docs", "codegen"]
 
 [[package]]
 name = "llama-stack-client"