theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-29 18:51:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Embeddings: Add string input as an option

Browse Source 
Used in OAI's API

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-11-16 23:48:31 -05:00
parent f9fffd42e0
commit b94c646210
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
endpoints/OAI/types/embedding.py
View File

@@ -1,4 +1,4 @@
from typing import List, Optional from typing import List, Optional, Union
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
@@ -10,7 +10,7 @@ class UsageInfo(BaseModel):
class EmbeddingsRequest(BaseModel): class EmbeddingsRequest(BaseModel):
input: List[str] = Field( input: Union[str, List[str]] = Field(
..., description="List of input texts to generate embeddings for." ..., description="List of input texts to generate embeddings for."
) )
encoding_format: str = Field( encoding_format: str = Field(