mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-01 11:02:37 +00:00
2fb27b63f5
- docs: for the Ollama notebooks, improve the specificity of some links, add `homebrew` install info, update some wording - tests: reduce number of local models needed to run in half from 4 → 2 (shedding 8gb of required installs) - bump deps (non-breaking) in anticipation of upcoming "thinking" PR
18 lines
447 B
Python
18 lines
447 B
Python
"""Test Ollama embeddings."""
|
|
|
|
from langchain_tests.integration_tests import EmbeddingsIntegrationTests
|
|
|
|
from langchain_ollama.embeddings import OllamaEmbeddings
|
|
|
|
MODEL_NAME = "llama3.1"
|
|
|
|
|
|
class TestOllamaEmbeddings(EmbeddingsIntegrationTests):
|
|
@property
|
|
def embeddings_class(self) -> type[OllamaEmbeddings]:
|
|
return OllamaEmbeddings
|
|
|
|
@property
|
|
def embedding_model_params(self) -> dict:
|
|
return {"model": MODEL_NAME}
|