phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-16 22:43:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update action config

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

3
.github/workflows/integration-vector-io-tests.yml vendored
View file

@ -137,8 +137,7 @@ jobs:
PGVECTOR_USER: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }} PGVECTOR_USER: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
PGVECTOR_PASSWORD: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }} PGVECTOR_PASSWORD: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
ENABLE_WEAVIATE: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'true' || '' }} ENABLE_WEAVIATE: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'true' || '' }}
WEAVIATE_API_KEY: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'llamastack' || '' }} WEAVIATE_CLUSTER_URL: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'localhost:8080' || '' }}
WEAVIATE_CLUSTER_URL: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'http://localhost:8080' || '' }}
run: | run: |
uv run pytest -sv --stack-config="inference=inline::sentence-transformers,vector_io=${{ matrix.vector-io-provider }}" \ uv run pytest -sv --stack-config="inference=inline::sentence-transformers,vector_io=${{ matrix.vector-io-provider }}" \

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

@ -167,6 +167,7 @@ class WeaviateVectorIOAdapter(
def _get_client(self) -> weaviate.Client: def _get_client(self) -> weaviate.Client:
if self.config.weaviate_cluster_url == "localhost:8080": if self.config.weaviate_cluster_url == "localhost:8080":
log.info("using Weaviate locally in container")
host, port = self.config.weaviate_cluster_url.split(":") host, port = self.config.weaviate_cluster_url.split(":")
key = "local_test" key = "local_test"
client = weaviate.connect_to_local( client = weaviate.connect_to_local(
@ -174,6 +175,7 @@ class WeaviateVectorIOAdapter(
port=port, port=port,
) )
else: else:
log.info("Using Weaviate remote cluster with URL")
key = f"{self.config.weaviate_cluster_url}::{self.config.weaviate_api_key}" key = f"{self.config.weaviate_cluster_url}::{self.config.weaviate_api_key}"
if key in self.client_cache: if key in self.client_cache:
return self.client_cache[key] return self.client_cache[key]