22 lines
669 B
Python
22 lines
669 B
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
|
|
class EmbedderConfig(BaseModel):
|
|
provider: str = Field(
|
|
description="Provider of the embedding model (e.g., 'ollama', 'openai')",
|
|
default="openai",
|
|
)
|
|
config: Optional[dict] = Field(
|
|
description="Configuration for the specific embedding model", default=None
|
|
)
|
|
|
|
@field_validator("config")
|
|
def validate_config(cls, v, values):
|
|
provider = values.data.get("provider")
|
|
if provider in ["openai", "ollama"]:
|
|
return v
|
|
else:
|
|
raise ValueError(f"Unsupported embedding provider: {provider}")
|
|
|