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 | |
| #5 | from mem0.configs.embeddings.base import BaseEmbedderConfig |
| #6 | from mem0.embeddings.ollama import OllamaEmbedding |
| #7 | |
| #8 | |
| #9 | @pytest.fixture |
| #10 | def mock_ollama_client(): |
| #11 | with patch("mem0.embeddings.ollama.Client") as mock_ollama: |
| #12 | mock_client = Mock() |
| #13 | mock_client.list.return_value = {"models": [{"name": "nomic-embed-text"}]} |
| #14 | mock_ollama.return_value = mock_client |
| #15 | yield mock_client |
| #16 | |
| #17 | |
| #18 | def test_embed_text(mock_ollama_client): |
| #19 | config = BaseEmbedderConfig(model="nomic-embed-text", embedding_dims=512) |
| #20 | embedder = OllamaEmbedding(config) |
| #21 | |
| #22 | mock_response = {"embedding": [0.1, 0.2, 0.3, 0.4, 0.5]} |
| #23 | mock_ollama_client.embeddings.return_value = mock_response |
| #24 | |
| #25 | text = "Sample text to embed." |
| #26 | embedding = embedder.embed(text) |
| #27 | |
| #28 | mock_ollama_client.embeddings.assert_called_once_with(model="nomic-embed-text", prompt=text) |
| #29 | |
| #30 | assert embedding == [0.1, 0.2, 0.3, 0.4, 0.5] |
| #31 | |
| #32 | |
| #33 | def test_ensure_model_exists(mock_ollama_client): |
| #34 | config = BaseEmbedderConfig(model="nomic-embed-text", embedding_dims=512) |
| #35 | embedder = OllamaEmbedding(config) |
| #36 | |
| #37 | mock_ollama_client.pull.assert_not_called() |
| #38 | |
| #39 | mock_ollama_client.list.return_value = {"models": []} |
| #40 | |
| #41 | embedder._ensure_model_exists() |
| #42 | |
| #43 | mock_ollama_client.pull.assert_called_once_with("nomic-embed-text") |
| #44 |