Updating to chromadb server's fastAPI

main.py

@@ -22,6 +22,21 @@ from chromadb.api.types import (
     QueryResult,
     CollectionMetadata,
 )
+from chromadb.errors import (
+    ChromaError,
+    InvalidUUIDError,
+    InvalidDimensionException,
+)
+from chromadb.server.fastapi.types import (
+    AddEmbedding,
+    DeleteEmbedding,
+    GetEmbedding,
+    QueryEmbedding,
+    RawSql,  # Results,
+    CreateCollection,
+    UpdateCollection,
+    UpdateEmbedding,
+)
 from chromadb.api import API
 from chromadb.config import System
 import chromadb.utils.embedding_functions as ef
@@ -60,22 +75,17 @@ def list_collections():
 @app.post(api_base+"/collections")
 def create_collection(
         self,
-        name: str,
-        metadata: Optional[CollectionMetadata] = None,
-        embedding_function: Optional[EmbeddingFunction] = ef.DefaultEmbeddingFunction(),
-        get_or_create: bool = False,
+        collection: CreateCollection 
     ) -> Collection:
     print("Received request to create_collection")
-    return bkend.create_collection(name,metadata=metadata,embedding_function=embedding_function,get_or_create=get_or_create)
+    return bkend.create_collection(name=collection.name,metadata=collection.metadata,get_or_create=collection.get_or_create)
 
 @app.get(api_base+"/collections")
 def get_collection(
         self,
-        name: str,
-        embedding_function: Optional[EmbeddingFunction] = ef.DefaultEmbeddingFunction(),
+        collection_name: str,
     ) -> Collection:
-    return bkend.get_collection(name,embedding_function=embedding_function
-    )
+    return bkend.get_collection(collection_name)
 
 if __name__ == "__main__":
   uvicorn.run(app, host="0.0.0.0", port=8000)