added support for string in embeddings

ebd-all-minilm/build.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 set -e
-export VERSION=1.0.2
+export VERSION=1.0.3
 source "$(dirname "${BASH_SOURCE[0]}")/../utils.sh"
 
 build_cpu ghcr.io/premai-io/embeddings-all-minilm-l6-v2-cpu all-MiniLM-L6-v2 ${@:1}

ebd-all-minilm/routes.py

@@ -2,13 +2,14 @@
 
 import tiktoken
 from fastapi import APIRouter
-from models import SentenceTransformerBasedModel as model
 from pydantic import BaseModel
 
+from models import SentenceTransformerBasedModel as model
+
 
 class EmbeddingsInput(BaseModel):
     model: str = None
-    input: Union[List[str], List[List[int]]]
+    input: Union[List[str], List[List[int]], str]
     user: str = ""
 
 
@@ -44,7 +45,10 @@ async def health():
 
 @router.post("/embeddings", response_model=EmbeddingsResponse)
 async def embeddings(body: EmbeddingsInput):
-    values = model.embeddings(texts=body.input)
+    if isinstance(body.input, str):
+        values = model.embeddings(texts=[body.input])
+    else:
+        values = model.embeddings(texts=body.input)
     return EmbeddingsResponse(
         object="list",
         data=[EmbeddingObject(embedding=value) for value in values],