Fix collection model

Author: Sicheng Pan

Diff for: chromadb/api/models/AsyncCollection.py

@@ -269,12 +269,18 @@ async def fork(
         Returns:
             Collection: A new collection with the specified name and containing identical data to the current collection.
         """
-        return await self._client._fork(
+        model = await self._client._fork(
             collection_id=self.id,
             new_name=new_name,
             tenant=self.tenant,
             database=self.database,
         )
+        return AsyncCollection(
+            client=self._client,
+            model=model,
+            embedding_function=self._embedding_function,
+            data_loader=self._data_loader
+        )
 
     async def update(
         self,

Diff for: chromadb/api/models/Collection.py

@@ -275,12 +275,18 @@ def fork(
         Returns:
             Collection: A new collection with the specified name and containing identical data to the current collection.
         """
-        return self._client._fork(
+        model = self._client._fork(
             collection_id=self.id,
             new_name=new_name,
             tenant=self.tenant,
             database=self.database,
         )
+        return Collection(
+            client=self._client,
+            model=model,
+            embedding_function=self._embedding_function,
+            data_loader=self._data_loader
+        )
 
     def update(
         self,