diff --git a/python/python/lancedb/embeddings/fine_tuner/llm.py b/python/python/lancedb/embeddings/fine_tuner/llm.py index 1bdf2d97..510c1fea 100644 --- a/python/python/lancedb/embeddings/fine_tuner/llm.py +++ b/python/python/lancedb/embeddings/fine_tuner/llm.py @@ -3,7 +3,7 @@ import re from functools import cached_property from typing import Optional -from pydantic import BaseModel +from pydantic import BaseModel, config from ...util import attempt_import_or_raise from ..utils import api_key_not_found_help @@ -15,10 +15,12 @@ class BaseLLM(BaseModel): Base class for Language Model based Embedding Functions. This class is loosely desined rn, and will be updated as the usage gets clearer. """ + class Config: + protected_namespaces = () # Disable protected namespace check model_name: str model_kwargs: dict = {} - + @cached_property def _client(): """