phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 00:52:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove host and port and use cluster url instead

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-07-30 20:38:39 -04:00
parent 87780534b0
commit edf4514ba2
3 changed files with 8 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/remote/vector_io/weaviate/config.py
View file

@ -17,25 +17,19 @@ from llama_stack.schema_utils import json_schema_type
@json_schema_type
class WeaviateVectorIOConfig(BaseModel):
host: str | None = Field(default="localhost")
port: int | None = Field(default=8080)
weaviate_api_key: str | None = Field(description="The API key for the Weaviate instance", default=None)
weaviate_cluster_url: str | None = Field(description="The URL of the Weaviate cluster", default=None)
weaviate_cluster_url: str | None = Field(description="The URL of the Weaviate cluster", default="localhost:8080")
kvstore: KVStoreConfig | None = Field(description="Config for KV store backend (SQLite only for now)", default=None)
@classmethod
def sample_run_config(
cls,
__distro_dir__: str,
host: str = "${env.WEAVIATE_HOST:=localhost}",
port: int = "${env.WEAVIATE_PORT:=8080}",
**kwargs: Any,
) -> dict[str, Any]:
return {
"host": "${env.WEAVIATE_HOST:=localhost}",
"port": "${env.WEAVIATE_PORT:=8080}",
"weaviate_api_key": None,
"weaviate_cluster_url": None,
"weaviate_cluster_url": "${env.WEAVIATE_CLUSTER_URL:=localhost:8080}",
"kvstore": SqliteKVStoreConfig.sample_run_config(
__distro_dir__=__distro_dir__,
db_name="weaviate_registry.db",

7
llama_stack/providers/remote/vector_io/weaviate/weaviate.py
View file

@ -166,11 +166,12 @@ class WeaviateVectorIOAdapter(
self.metadata_collection_name = "openai_vector_stores_metadata"
def _get_client(self) -> weaviate.Client:
if self.config.weaviate_cluster_url is None:
if self.config.weaviate_cluster_url == "localhost:8080":
host, port = self.config.weaviate_cluster_url.split(":")
key = "local_test"
client = weaviate.connect_to_local(
host=self.config.host,
port=self.config.port,
host=host,
port=port,
)
else:
key = f"{self.config.weaviate_cluster_url}::{self.config.weaviate_api_key}"