make embedding configurable and add gpt-4-turbo.

This commit is contained in:
seehi 2024-04-10 14:52:48 +08:00
parent c494844c02
commit caa1300163
2 changed files with 23 additions and 5 deletions

View file

@ -14,7 +14,25 @@ class EmbeddingType(Enum):
class EmbeddingConfig(YamlModel):
"""Config for Embedding."""
"""Config for Embedding.
Examples:
---------
api_type: "openai"
api_key: "YOU_API_KEY"
api_type: "azure"
api_key: "YOU_API_KEY"
base_url: "YOU_BASE_URL"
api_version: "YOU_API_VERSION"
api_type: "gemini"
api_key: "YOU_API_KEY"
api_type: "ollama"
base_url: "YOU_BASE_URL"
model: "YOU_MODEL"
"""
api_type: Optional[EmbeddingType] = None
api_key: Optional[str] = None