add chromadb dependency to requirements

requirements.in

@@ -1,5 +1,6 @@
 aiohttp
 cashews
+chromadb
 datasets
 einops
 fastapi

requirements.txt

@@ -15,35 +15,66 @@ anyio==4.8.0
     # via
     #   httpx
     #   starlette
+    #   watchfiles
+asgiref==3.8.1
+    # via opentelemetry-instrumentation-asgi
 attrs==25.1.0
     # via aiohttp
+backoff==2.2.1
+    # via posthog
+bcrypt==4.2.1
+    # via chromadb
+build==1.2.2.post1
+    # via chromadb
+cachetools==5.5.2
+    # via google-auth
 cashews==7.4.0
     # via -r requirements.in
 certifi==2025.1.31
     # via
     #   httpcore
     #   httpx
+    #   kubernetes
     #   requests
 charset-normalizer==3.4.1
     # via requests
+chroma-hnswlib==0.7.6
+    # via chromadb
+chromadb==0.6.3
+    # via -r requirements.in
 click==8.1.8
-    # via uvicorn
+    # via
+    #   typer
+    #   uvicorn
+coloredlogs==15.0.1
+    # via onnxruntime
 datasets==3.3.2
     # via -r requirements.in
+deprecated==1.2.18
+    # via
+    #   opentelemetry-api
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-semantic-conventions
 dill==0.3.8
     # via
     #   datasets
     #   multiprocess
+durationpy==0.9
+    # via kubernetes
 einops==0.8.1
     # via -r requirements.in
 fastapi==0.115.8
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   chromadb
 filelock==3.17.0
     # via
     #   datasets
     #   huggingface-hub
     #   torch
     #   transformers
+flatbuffers==25.2.10
+    # via onnxruntime
 frozenlist==1.5.0
     # via
     #   aiohttp
@@ -53,6 +84,14 @@ fsspec==2024.12.0
     #   datasets
     #   huggingface-hub
     #   torch
+google-auth==2.38.0
+    # via kubernetes
+googleapis-common-protos==1.68.0
+    # via opentelemetry-exporter-otlp-proto-grpc
+grpcio==1.70.0
+    # via
+    #   chromadb
+    #   opentelemetry-exporter-otlp-proto-grpc
 h11==0.14.0
     # via
     #   httpcore
@@ -65,8 +104,12 @@ hpack==4.1.0
     # via h2
 httpcore==1.0.7
     # via httpx
+httptools==0.6.4
+    # via uvicorn
 httpx==0.28.1
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   chromadb
 huggingface-hub==0.29.1
     # via
     #   -r requirements.in
@@ -74,6 +117,8 @@ huggingface-hub==0.29.1
     #   sentence-transformers
     #   tokenizers
     #   transformers
+humanfriendly==10.0
+    # via coloredlogs
 hyperframe==6.1.0
     # via h2
 idna==3.10
@@ -82,12 +127,26 @@ idna==3.10
     #   httpx
     #   requests
     #   yarl
+importlib-metadata==8.5.0
+    # via opentelemetry-api
+importlib-resources==6.5.2
+    # via chromadb
 jinja2==3.1.5
     # via torch
 joblib==1.4.2
     # via scikit-learn
+kubernetes==32.0.1
+    # via chromadb
+markdown-it-py==3.0.0
+    # via rich
 markupsafe==3.0.2
     # via jinja2
+mdurl==0.1.2
+    # via markdown-it-py
+mmh3==5.1.0
+    # via chromadb
+monotonic==1.6
+    # via posthog
 mpmath==1.3.0
     # via sympy
 multidict==6.1.0
@@ -100,15 +159,70 @@ networkx==3.4.2
     # via torch
 numpy==2.2.3
     # via
+    #   chroma-hnswlib
+    #   chromadb
     #   datasets
+    #   onnxruntime
     #   pandas
     #   scikit-learn
     #   scipy
     #   transformers
+oauthlib==3.2.2
+    # via
+    #   kubernetes
+    #   requests-oauthlib
+onnxruntime==1.20.1
+    # via chromadb
+opentelemetry-api==1.30.0
+    # via
+    #   chromadb
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-asgi
+    #   opentelemetry-instrumentation-fastapi
+    #   opentelemetry-sdk
+    #   opentelemetry-semantic-conventions
+opentelemetry-exporter-otlp-proto-common==1.30.0
+    # via opentelemetry-exporter-otlp-proto-grpc
+opentelemetry-exporter-otlp-proto-grpc==1.30.0
+    # via chromadb
+opentelemetry-instrumentation==0.51b0
+    # via
+    #   opentelemetry-instrumentation-asgi
+    #   opentelemetry-instrumentation-fastapi
+opentelemetry-instrumentation-asgi==0.51b0
+    # via opentelemetry-instrumentation-fastapi
+opentelemetry-instrumentation-fastapi==0.51b0
+    # via chromadb
+opentelemetry-proto==1.30.0
+    # via
+    #   opentelemetry-exporter-otlp-proto-common
+    #   opentelemetry-exporter-otlp-proto-grpc
+opentelemetry-sdk==1.30.0
+    # via
+    #   chromadb
+    #   opentelemetry-exporter-otlp-proto-grpc
+opentelemetry-semantic-conventions==0.51b0
+    # via
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-asgi
+    #   opentelemetry-instrumentation-fastapi
+    #   opentelemetry-sdk
+opentelemetry-util-http==0.51b0
+    # via
+    #   opentelemetry-instrumentation-asgi
+    #   opentelemetry-instrumentation-fastapi
+orjson==3.10.15
+    # via chromadb
+overrides==7.7.0
+    # via chromadb
 packaging==24.2
     # via
+    #   build
     #   datasets
     #   huggingface-hub
+    #   onnxruntime
+    #   opentelemetry-instrumentation
     #   transformers
 pandas==2.2.3
     # via datasets
@@ -116,36 +230,75 @@ pillow==11.1.0
     # via sentence-transformers
 polars==1.23.0
     # via -r requirements.in
+posthog==3.15.1
+    # via chromadb
 propcache==0.3.0
     # via
     #   aiohttp
     #   yarl
+protobuf==5.29.3
+    # via
+    #   googleapis-common-protos
+    #   onnxruntime
+    #   opentelemetry-proto
 pyarrow==19.0.1
     # via datasets
+pyasn1==0.6.1
+    # via
+    #   pyasn1-modules
+    #   rsa
+pyasn1-modules==0.4.1
+    # via google-auth
 pydantic==2.10.6
     # via
     #   -r requirements.in
+    #   chromadb
     #   fastapi
 pydantic-core==2.27.2
     # via pydantic
+pygments==2.19.1
+    # via rich
+pypika==0.48.9
+    # via chromadb
+pyproject-hooks==1.2.0
+    # via build
 python-dateutil==2.9.0.post0
-    # via pandas
+    # via
+    #   kubernetes
+    #   pandas
+    #   posthog
 python-dotenv==1.0.1
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   uvicorn
 pytz==2025.1
     # via pandas
 pyyaml==6.0.2
     # via
+    #   chromadb
     #   datasets
     #   huggingface-hub
+    #   kubernetes
     #   transformers
+    #   uvicorn
 regex==2024.11.6
     # via transformers
 requests==2.32.3
     # via
     #   datasets
     #   huggingface-hub
+    #   kubernetes
+    #   posthog
+    #   requests-oauthlib
     #   transformers
+requests-oauthlib==2.0.0
+    # via kubernetes
+rich==13.9.4
+    # via
+    #   chromadb
+    #   typer
+rsa==4.9
+    # via google-auth
 safetensors==0.5.2
     # via transformers
 scikit-learn==1.6.1
@@ -158,8 +311,13 @@ sentence-transformers==3.4.1
     # via -r requirements.in
 setuptools==75.8.1
     # via torch
+shellingham==1.5.4
+    # via typer
 six==1.17.0
-    # via python-dateutil
+    # via
+    #   kubernetes
+    #   posthog
+    #   python-dateutil
 sniffio==1.3.1
     # via anyio
 stamina==24.3.0
@@ -167,39 +325,69 @@ stamina==24.3.0
 starlette==0.45.3
     # via fastapi
 sympy==1.13.1
-    # via torch
+    # via
+    #   onnxruntime
+    #   torch
 tenacity==9.0.0
-    # via stamina
+    # via
+    #   chromadb
+    #   stamina
 threadpoolctl==3.5.0
     # via scikit-learn
 tokenizers==0.21.0
-    # via transformers
+    # via
+    #   chromadb
+    #   transformers
 torch==2.6.0
     # via sentence-transformers
 tqdm==4.67.1
     # via
     #   -r requirements.in
+    #   chromadb
     #   datasets
     #   huggingface-hub
     #   sentence-transformers
     #   transformers
 transformers==4.49.0
     # via sentence-transformers
+typer==0.15.1
+    # via chromadb
 typing-extensions==4.12.2
     # via
     #   anyio
+    #   chromadb
     #   fastapi
     #   huggingface-hub
+    #   opentelemetry-sdk
     #   pydantic
     #   pydantic-core
     #   torch
+    #   typer
 tzdata==2025.1
     # via pandas
 urllib3==2.3.0
-    # via requests
+    # via
+    #   kubernetes
+    #   requests
 uvicorn==0.34.0
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   chromadb
+uvloop==0.21.0
+    # via uvicorn
+watchfiles==1.0.4
+    # via uvicorn
+websocket-client==1.8.0
+    # via kubernetes
+websockets==15.0
+    # via uvicorn
+wrapt==1.17.2
+    # via
+    #   deprecated
+    #   opentelemetry-instrumentation
 xxhash==3.5.0
     # via datasets
 yarl==1.18.3
     # via aiohttp
+zipp==3.21.0
+    # via importlib-metadata