Return appropriate EmbeddingModel implementation for the given name.
(model_name: str, ollama_host: str | None = None)
| 143 | return embeddings_np |
| 144 | |
| 145 | def select_embedder(model_name: str, ollama_host: str | None = None): |
| 146 | """Return appropriate EmbeddingModel implementation for the given name.""" |
| 147 | if "/" in model_name or model_name.startswith("http"): |
| 148 | # Treat as HF model path |
| 149 | return QwenEmbedder(model_name=model_name) |
| 150 | # Otherwise assume it's an Ollama tag |
| 151 | return OllamaEmbedder(model_name=model_name, host=ollama_host) |
| 152 | |
| 153 | if __name__ == '__main__': |
| 154 | print("representations.py cleaned up.") |
no test coverage detected