Skip to content

Commit d9fafc2

Browse files
committed
Text-to-SQL: Rename Hugging Face provider identifier
1 parent 58549fb commit d9fafc2

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

cratedb_toolkit/query/nlsql/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ class ModelProvider(Enum):
1515
ANTHROPIC = "anthropic"
1616
AZURE = "azure"
1717
GOOGLE = "google"
18-
HUGGINGFACE_API = "huggingface_api"
18+
HUGGINGFACE_SERVERLESS = "huggingface_serverless"
1919
LLAMAFILE = "llamafile"
2020
MISTRAL = "mistral"
2121
OLLAMA = "ollama"

cratedb_toolkit/query/nlsql/util.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def disable_embeddings():
6868
ModelProvider.ANTHROPIC: "claude-haiku-4-5",
6969
ModelProvider.AZURE: "gpt-4.1", # TODO: Not validated yet.
7070
ModelProvider.GOOGLE: "gemini-2.5-flash", # TODO: Not validated yet.
71-
ModelProvider.HUGGINGFACE_API: "HuggingFaceH4/zephyr-7b-alpha", # TODO: Not validated yet.
71+
ModelProvider.HUGGINGFACE_SERVERLESS: "HuggingFaceH4/zephyr-7b-alpha", # TODO: Not validated yet.
7272
ModelProvider.OLLAMA: "gemma3:1b",
7373
ModelProvider.OPENAI: "gpt-4o-mini",
7474
ModelProvider.OPENROUTER: "gryphe/mythomax-l2-13b",
@@ -132,7 +132,7 @@ def read_llm_options(
132132
raise ValueError(
133133
"LLM API key not defined. Use either CLI/API parameter or GOOGLE_API_KEY environment variable."
134134
)
135-
elif provider is ModelProvider.HUGGINGFACE_API:
135+
elif provider is ModelProvider.HUGGINGFACE_SERVERLESS:
136136
llm_api_key = llm_api_key or os.getenv("HF_TOKEN")
137137
if not llm_api_key:
138138
raise ValueError(
@@ -252,7 +252,7 @@ def configure_llm(info: ModelInfo, debug: bool = False) -> "LLM":
252252
base_url=info.endpoint,
253253
api_key=info.api_key,
254254
)
255-
elif info.provider is ModelProvider.HUGGINGFACE_API:
255+
elif info.provider is ModelProvider.HUGGINGFACE_SERVERLESS:
256256
from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
257257

258258
llm = HuggingFaceInferenceAPI(

0 commit comments

Comments
 (0)