repositories
loading repo index
repositories
loading repo index
repository
loading code, commits, and activity
public Clawd ADK gateway launch mirror
stars
latest
clone command
git clone gitlawb://did:key:z6Mkq5mY...iFZ5/my-project-publ...git clone gitlawb://did:key:z6Mkq5mY.../my-project-publ...2fa351d6docs: add automaton and perps launch sources16d ago| #1 | from unittest.mock import Mock, patch |
| #2 | |
| #3 | import pytest |
| #4 | import numpy as np |
| #5 | from mem0.configs.embeddings.base import BaseEmbedderConfig |
| #6 | |
| #7 | try: |
| #8 | from mem0.embeddings.fastembed import FastEmbedEmbedding |
| #9 | except ImportError: |
| #10 | pytest.skip("fastembed not installed", allow_module_level=True) |
| #11 | |
| #12 | |
| #13 | @pytest.fixture |
| #14 | def mock_fastembed_client(): |
| #15 | with patch("mem0.embeddings.fastembed.TextEmbedding") as mock_fastembed: |
| #16 | mock_client = Mock() |
| #17 | mock_fastembed.return_value = mock_client |
| #18 | yield mock_client |
| #19 | |
| #20 | |
| #21 | def test_embed_with_jina_model(mock_fastembed_client): |
| #22 | config = BaseEmbedderConfig(model="jinaai/jina-embeddings-v2-base-en", embedding_dims=768) |
| #23 | embedder = FastEmbedEmbedding(config) |
| #24 | |
| #25 | mock_embedding = np.array([0.1, 0.2, 0.3, 0.4, 0.5]) |
| #26 | mock_fastembed_client.embed.return_value = iter([mock_embedding]) |
| #27 | |
| #28 | text = "Sample text to embed." |
| #29 | embedding = embedder.embed(text) |
| #30 | |
| #31 | mock_fastembed_client.embed.assert_called_once_with(text) |
| #32 | assert list(embedding) == [0.1, 0.2, 0.3, 0.4, 0.5] |
| #33 | |
| #34 | |
| #35 | def test_embed_removes_newlines(mock_fastembed_client): |
| #36 | config = BaseEmbedderConfig(model="jinaai/jina-embeddings-v2-base-en", embedding_dims=768) |
| #37 | embedder = FastEmbedEmbedding(config) |
| #38 | |
| #39 | mock_embedding = np.array([0.7, 0.8, 0.9]) |
| #40 | mock_fastembed_client.embed.return_value = iter([mock_embedding]) |
| #41 | |
| #42 | text_with_newlines = "Hello\nworld" |
| #43 | embedding = embedder.embed(text_with_newlines) |
| #44 | |
| #45 | mock_fastembed_client.embed.assert_called_once_with("Hello world") |
| #46 | assert list(embedding) == [0.7, 0.8, 0.9] |