mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
29 lines
883 B
Python
29 lines
883 B
Python
import unittest
|
|
from memory_scope.models.base_embedding_model import LLIEmbedding
|
|
|
|
class TestLLIEmbedding(unittest.TestCase):
|
|
"""Tests for LLIEmbedding"""
|
|
|
|
def setUp(self):
|
|
config = {
|
|
"method_type": "DashScopeEmbedding",
|
|
"model_name": "text-embedding-v2",
|
|
"clazz": "models.base_embedding_model"
|
|
}
|
|
self.emb = LLIEmbedding(**config)
|
|
|
|
def test_single_embedding(self):
|
|
text = "您吃了吗？"
|
|
embs = self.emb.call(text=text)
|
|
|
|
def test_batch_embedding(self):
|
|
texts = ["您吃了吗？",
|
|
"吃了吗您？"]
|
|
embs = self.emb.call(text=texts)
|
|
|
|
async def test_async_embedding(self):
|
|
texts = ["您吃了吗？",
|
|
"吃了吗您？"]
|
|
# 调用异步函数并等待其结果
|
|
embs = await self.emb.async_call(texts)
|
|
|