cuv2 api upgrade (#83)

* cuv2 api upgrade * delete yay --------- Co-authored-by: Michael Ryave <[email protected]>
arcee-ai · Oct 8, 2024 · 453fa13 · 453fa13
1 parent 0b92625
commit 453fa13
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/arcee/api.py b/arcee/api.py
@@ -35,7 +35,7 @@ def upload_corpus_folder(corpus: str, s3_folder_url: str, tokenizer_name: str, b
         "block_size": block_size,
     }
 
-    return make_request("post", Route.pretraining + "/corpusUpload", data)
+    return make_request("post", Route.corpus, data)
 
 
 def upload_qa_pairs(
@@ -321,30 +321,26 @@ def merging_status(merging: str) -> Dict[str, str]:
     return make_request("get", Route.merging + "/status", data)
 
 
-def delete_corpus(corpus: str) -> Dict[str, str]:
+def delete_corpus(corpus_name: str) -> Dict[str, str]:
     """
     Delete a corpus
 
     Args:
         corpus (str): The name of the corpus to delete
     """
+    data = {"corpus_name": corpus_name}
+    return make_request("delete", Route.corpus + "/delete", data)
 
-    data = {"corpus_name": corpus}
 
-    return make_request("post", Route.pretraining + "/deleteCorpus", data)
-
-
-def corpus_status(corpus: str) -> Dict[str, str]:
+def corpus_status(corpus_id: str) -> Dict[str, str]:
     """
     Check the status of a corpus
 
     Args:
         corpus (str): The name of the corpus to check the status
     """
 
-    data = {"corpus_name": corpus}
-
-    return make_request("post", Route.pretraining + "/corpus/status", data)
+    return make_request("get", Route.corpus + f"/status/{corpus_id}")
 
 
 def start_alignment(

diff --git a/arcee/schemas/routes.py b/arcee/schemas/routes.py
@@ -15,3 +15,4 @@ class Route(StrEnum):
     deployment = "deployment"
     merging = "merging"
     retriever = "models"
+    corpus = "pretraining/corpus"