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

Refactor Weaviate config to include cluster URL in memory adapter

Browse source
This commit is contained in:
Zain Hasan 2024-09-23 21:13:24 -04:00
parent 2fc9bd95a6
commit ba8044d243
3 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/adapters/memory/weaviate/weaviate.py
View file

@ -1,7 +1,6 @@
import json
import uuid
from typing import List, Optional, Dict, Any
from urllib.parse import urlparse
from numpy.typing import NDArray
import weaviate
@ -72,7 +71,6 @@ class WeaviateIndex(EmbeddingIndex):
class WeaviateMemoryAdapter(Memory):
def __init__(self, config: WeaviateConfig) -> None:
print(f"Initializing WeaviateMemoryAdapter with URL: {config.url}")
self.config = config
self.client = None
self.cache = {}
@ -85,9 +83,10 @@ class WeaviateMemoryAdapter(Memory):
assert isinstance(request_provider_data, WeaviateRequestProviderData)
print(f"WEAVIATE API KEY: {request_provider_data.weaviate_api_key}")
print(f"WEAVIATE CLUSTER URL: {request_provider_data.weaviate_cluster_url}")
# Connect to Weaviate Cloud
self.client = weaviate.connect_to_weaviate_cloud(
cluster_url = self.config.url,
cluster_url = request_provider_data.weaviate_cluster_url,
auth_credentials = Auth.api_key(request_provider_data.weaviate_api_key),
)