feat(api): support wenxin text embedding (#7377)

This commit is contained in:
Chengyu Yan
2024-08-19 09:15:19 +08:00
committed by GitHub
parent a0a67873aa
commit bfd905602f
14 changed files with 553 additions and 228 deletions

View File

@@ -0,0 +1,24 @@
import os
from time import sleep
from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
from core.model_runtime.model_providers.wenxin.text_embedding.text_embedding import WenxinTextEmbeddingModel
def test_invoke_embedding_model():
sleep(3)
model = WenxinTextEmbeddingModel()
response = model.invoke(
model='embedding-v1',
credentials={
'api_key': os.environ.get('WENXIN_API_KEY'),
'secret_key': os.environ.get('WENXIN_SECRET_KEY')
},
texts=['hello', '你好', 'xxxxx'],
user="abc-123"
)
assert isinstance(response, TextEmbeddingResult)
assert len(response.embeddings) == 3
assert isinstance(response.embeddings[0], list)