jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: baichuan max chunks (#1990)

Browse source
This commit is contained in:
Yeuoly 2024-01-10 23:13:35 +08:00 • committed by GitHub
commit ebd11e7482
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 85 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

37
api/tests/integration_tests/model_runtime/baichuan/test_text_embedding.py
View file

@ -59,3 +59,40 @@ def test_get_num_tokens():
)
assert num_tokens == 2
def test_max_chunks():
model = BaichuanTextEmbeddingModel()
result = model.invoke(
model='baichuan-text-embedding',
credentials={
'api_key': os.environ.get('BAICHUAN_API_KEY'),
},
texts=[
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
"hello",
"world",
]
)
assert isinstance(result, TextEmbeddingResult)
assert len(result.embeddings) == 22