Add mthds

main.py

@@ -67,6 +67,31 @@ def heartbeat():
     print("Received heartbeat request")
     return bkend.heartbeat()
 
+@app.post(api_base+"/reset")
+def reset():
+    print("Received reset request")
+    return bkend.reset()
+
+@app.get(api_base+"/version")
+def version():
+    print("Received version request")
+    return bkend.version()
+
+@app.post(api_base+"/persist")
+def persist():
+    print("Received persist request")
+    return bkend.persist()
+
+@app.post(api_base+"/raw_sql")
+def raw_sql(raw_sql: RawSql):
+    print("Received raw_sql request")
+    return bkend.raw_sql(raw_sql)
+
+@app.get(api_base+"/heartbeat")
+def heartbeat():
+    print("Received heartbeat request")
+    return bkend.heartbeat()
+
 @app.get(api_base+"/collections")
 def list_collections():
     print("Received list_collections request")

server.py

@@ -1,8 +1,7 @@
 from chromadb.config import Settings
 import chromadb
-rom pydantic import BaseModel
+from pydantic import BaseModel
 from typing import List, Dict, Any, Generator, Optional, cast
-from server import client
 from chromadb.api.types import (
     Documents,
     Embeddings,
@@ -79,4 +78,15 @@ class client():
         col=self.db.get_collection(name,embedding_function=embedding_function)
         print(col)
         return col
-    
+
+    def reset():
+        return self.db.reset()
+
+    def version():
+        return self.db.get_version()
+
+    def persist():
+        return self.db.persist()
+
+    def raw_sql(raw_sql: RawSql):
+        return self.db.raw_sql(raw_sql.raw_sql)