Add test case for POST /v1/embeddings (#63)

Signed-off-by: Tyson Dai <[email protected]>
SkywardAI · Aug 23, 2024 · e3c7ec7 · e3c7ec7
1 parent 5cdb5f4
commit e3c7ec7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/example/test_apis.py b/example/test_apis.py
@@ -83,6 +83,13 @@ def test_inference_by_request_stream(self):
         res=requests.post(url=self.base_url+"/v1/chat/completions", json=data,headers={"Content-Type": "application/json", "Authorization":"Bearer no-key"},)
         self.assertEqual(res.status_code, 200)
 
+    def test_embedding_api(self):
+        data = {
+                "input": "Hello, world!",
+                "model": "all-MiniLM-L6-v2"
+            }
+        res = requests.post(url=self.base_url+"/v1/embeddings", json=data, headers={"Content-Type": "application/json", "Authorization":"Bearer no-key"})
+        self.assertEqual(res.status_code, 200)
 
     def test_api_key(self):
         res=requests.get(url=self.base_url+"/v1/token/api-key")