updating server imports

server.py

@@ -1,8 +1,8 @@
 from chromadb.config import Settings
 import chromadb
-from typing import Optional, cast
-from chromadb.api import API
-from chromadb.config import System
+rom pydantic import BaseModel
+from typing import List, Dict, Any, Generator, Optional, cast
+from server import client
 from chromadb.api.types import (
     Documents,
     Embeddings,
@@ -16,6 +16,23 @@ 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
 import pandas as pd
 import requests
@@ -26,6 +43,8 @@ import chromadb.errors as errors
 from uuid import UUID
 from chromadb.telemetry import Telemetry
 from overrides import override
+
+
 import time
 
 class client():
@@ -48,12 +67,16 @@ class client():
         embedding_function: Optional[EmbeddingFunction] = ef.DefaultEmbeddingFunction(),
         get_or_create: bool = False,
     ) -> Collection:
-        return self.db.create_collection(name,metadata=metadata,embedding_function=embedding_function,get_or_create=get_or_create)
+        col=self.db.create_collection(name,metadata=metadata,embedding_function=embedding_function,get_or_create=get_or_create)
+        print(col)
+        return col
     
     def get_collection(
         self,
         name: str,
         embedding_function: Optional[EmbeddingFunction] = ef.DefaultEmbeddingFunction(),
     ) -> Collection:
-        return self.db.get_collection(name,embedding_function=embedding_function)
+        col=self.db.get_collection(name,embedding_function=embedding_function)
+        print(col)
+        return col