tested for embeddings/embed

This commit is contained in:
EvensXia 2024-10-30 09:47:10 +08:00
parent 187e512547
commit f2aa4e3f9d
3 changed files with 49 additions and 14 deletions

View file

@ -28,7 +28,8 @@ class LLMType(Enum):
AZURE = "azure"
OLLAMA = "ollama" # /chat at ollama api
OLLAMA_GENERATE = "ollama.generate" # /generate at ollama api
OLLAMA_EMBEDDING = "ollama.embeddings" # /embeddings at ollama api
OLLAMA_EMBEDDINGS = "ollama.embeddings" # /embeddings at ollama api
OLLAMA_EMBED = "ollama.embed" # /embeddings at ollama api
QIANFAN = "qianfan" # Baidu BCE
DASHSCOPE = "dashscope" # Aliyun LingJi DashScope
MOONSHOT = "moonshot"