Skip to content

Commit f57e63f

Browse files
committed
chore: change default embedding model and default LLM (from 4o to 5.1)
1 parent 60b8934 commit f57e63f

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

src/sparql_llm/config.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -131,12 +131,8 @@ class Settings(BaseSettings):
131131
# vectordb_url: str = "http://vectordb:6334/"
132132
vectordb_url: str = "data/vectordb"
133133
# https://qdrant.github.io/fastembed/examples/Supported_Models/#supported-text-embedding-models
134-
# embedding_model: str = "BAAI/bge-large-en-v1.5"
135-
# embedding_dimensions: int = 1024
136-
# embedding_model: str = "BAAI/bge-base-en-v1.5"
137-
# embedding_dimensions: int = 768
138-
embedding_model: str = "BAAI/bge-small-en-v1.5"
139-
embedding_dimensions: int = 384
134+
embedding_model: str = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
135+
embedding_dimensions: int = 768
140136
force_index: bool = False
141137
# Automatically initialize the vector store client, should be False when deploying in prod with multiple workers
142138
auto_init: bool = True
@@ -148,7 +144,7 @@ class Settings(BaseSettings):
148144
entities_collection_name: str = "entities"
149145

150146
# Default settings for the agent that can be changed at runtime
151-
default_llm_model: str = "openrouter/openai/gpt-4o"
147+
default_llm_model: str = "openrouter/openai/gpt-5.1"
152148
# default_llm_model: str = "openai/gpt-4o"
153149
# TODO: default_llm_model_cheap: str = "openai/gpt-4o-mini"
154150

src/sparql_llm/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,10 +225,10 @@ def _ensure_loaded(self) -> None:
225225

226226
logger.info(f"Fetching metadata for {len(self._endpoints)} endpoints...")
227227
for endpoint in self._endpoints:
228+
logger.info(f"Fetching {endpoint['endpoint_url']} metadata...")
228229
self._void_dict[endpoint["endpoint_url"]] = get_schema_for_endpoint(
229230
endpoint["endpoint_url"], endpoint.get("void_file")
230231
)
231-
logger.info(f"Fetching {endpoint['endpoint_url']} metadata...")
232232
self._prefixes_map = get_prefixes_for_endpoint(
233233
endpoint["endpoint_url"], endpoint.get("examples_file"), self._prefixes_map
234234
)

0 commit comments

Comments
 (0)