Convert old COCOINDEX_CODE_EMBEDDING_MODEL to (provider, model).
(env_model: str)
| 167 | |
| 168 | |
| 169 | def _convert_embedding_model(env_model: str) -> tuple[str, str]: |
| 170 | """Convert old COCOINDEX_CODE_EMBEDDING_MODEL to (provider, model).""" |
| 171 | sbert_prefix = "sbert/" |
| 172 | if env_model.startswith(sbert_prefix): |
| 173 | return "sentence-transformers", env_model[len(sbert_prefix) :] |
| 174 | return "litellm", env_model |
| 175 | |
| 176 | |
| 177 | def main() -> None: |
no outgoing calls