| # Embedding |
|
|
| In TaskWeaver, we support various embedding models to generate embeddings for auto plugin selection. |
|
|
|
|
| ## Embedding Configration |
|
|
| - `llm.embedding_api_type`: The type of the embedding API. We support the following types: |
| - openai |
| - qwen |
| - ollama |
| - sentence_transformers |
| |
| - `llm.embedding_model`: The embedding model name. The model name should be aligned with `llm.embedding_api_type`. |
| We only list some embedding models we have tested below: |
| - openai |
| - text-embedding-ada-002 |
| - qwen |
| - text-embedding-v1 |
| - ollama |
| - llama2 |
| - sentence_transformers |
| - all-mpnet-base-v2 |
| - multi-qa-mpnet-base-dot-v1 |
| - all-distilroberta-v1 |
| - all-MiniLM-L12-v2 |
| - multi-qa-MiniLM-L6-cos-v1 |
| |
| You also can use other embedding models supported by the above embedding APIs. |
