Move v1 routes out into separate file/wrapper class

chroma-core · Oct 16, 2024 · 1c8f562 · 1c8f562
1 parent 983f11f
commit 1c8f562
Show file tree

Hide file tree

Showing 4 changed files with 870 additions and 816 deletions.
diff --git a/chromadb/api/fastapi.py b/chromadb/api/fastapi.py
@@ -122,7 +122,6 @@ def get_database(
  resp_json = self._make_request(
  "get",
  f"/tenants/{tenant}/databases/{name}",
- params={"tenant": tenant},
  )
  return Database(
  id=resp_json["id"], name=resp_json["name"], tenant=resp_json["tenant"]

diff --git a/chromadb/app.py b/chromadb/app.py
@@ -1,7 +1,7 @@
 import chromadb
 import chromadb.config
-from chromadb.server.fastapi import FastAPI
+from chromadb.server.fastapi.v1 import FastAPIWithV1
 
 settings = chromadb.config.Settings()
-server = FastAPI(settings)
+server = FastAPIWithV1(settings)
 app = server.app()