estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

enhance: use urllib join instead of fstring (#9549)

Browse Source
This commit is contained in:
Byron.wang
2024-10-21 19:04:28 +08:00
committed by GitHub
parent 31a603e905
commit 37fea072bc
4 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/tests/unit_tests/core/model_runtime/model_providers/wenxin/test_text_embedding.py
View File

@@ -1,3 +1,5 @@
import string
import numpy as np
from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
@@ -31,7 +33,7 @@ def test_max_chunks():
max_chunks = embedding_model._get_max_chunks(model, credentials)
embedding_model._create_text_embedding = _create_text_embedding
texts = ["0123456789" for i in range(0, max_chunks * 2)]
texts = [string.digits for i in range(0, max_chunks * 2)]
result: TextEmbeddingResult = embedding_model.invoke(model, credentials, texts, "test")
assert len(result.embeddings) == max_chunks * 2