Skip to content

Commit 6030d62

Browse files
chore: Update opensearch-rag stack and demo for SDP 26.3
1 parent 6f62558 commit 6030d62

3 files changed

Lines changed: 20 additions & 18 deletions

File tree

demos/opensearch-rag/load-embeddings-from-git.yaml

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,10 @@ data:
1818
from requests.auth import HTTPBasicAuth
1919
2020
# Configuration
21-
OPENSEARCH_HOST = os.getenv('OPENSEARCH_HOST', 'opensearch')
22-
OPENSEARCH_PORT = int(os.getenv('OPENSEARCH_PORT', '9200'))
21+
OPENSEARCH_HOSTS = os.getenv('OPENSEARCH_HOSTS', 'https://opensearch:9200')
22+
OPENSEARCH_HOSTNAME = os.getenv('OPENSEARCH_HOSTNAME', 'opensearch')
23+
OPENSEARCH_PORT = os.getenv('OPENSEARCH_PORT', '9200')
24+
OPENSEARCH_PROTOCOL = os.getenv('OPENSEARCH_PROTOCOL', 'https')
2325
OPENSEARCH_USER = os.getenv('OPENSEARCH_USER', 'admin')
2426
OPENSEARCH_PASSWORD = os.getenv('OPENSEARCH_PASSWORD', 'adminadmin')
2527
INDEX_NAME = os.getenv('INDEX_NAME', 'rag-documents')
@@ -31,7 +33,7 @@ data:
3133
for i in range(60):
3234
try:
3335
response = requests.get(
34-
f'https://{OPENSEARCH_HOST}:{OPENSEARCH_PORT}',
36+
OPENSEARCH_HOSTS,
3537
auth=HTTPBasicAuth(OPENSEARCH_USER, OPENSEARCH_PASSWORD),
3638
verify=False,
3739
timeout=5
@@ -142,7 +144,7 @@ data:
142144
auth = HTTPBasicAuth(OPENSEARCH_USER, OPENSEARCH_PASSWORD)
143145
144146
# Create index pattern
145-
response = requests.put(
147+
response = requests.post(
146148
f'{base_url}/api/saved_objects/index-pattern/{INDEX_NAME}?overwrite=true',
147149
json={"attributes": {"title": INDEX_NAME, "timeFieldName": "timestamp"}},
148150
auth=auth, headers=headers, timeout=10
@@ -202,9 +204,9 @@ data:
202204
return 1
203205
204206
client = OpenSearch(
205-
hosts=[{'host': OPENSEARCH_HOST, 'port': OPENSEARCH_PORT}],
207+
hosts=[{'host': OPENSEARCH_HOSTNAME, 'port': OPENSEARCH_PORT}],
206208
http_auth=(OPENSEARCH_USER, OPENSEARCH_PASSWORD),
207-
use_ssl=True,
209+
use_ssl=OPENSEARCH_PROTOCOL == 'https',
208210
verify_certs=False,
209211
ssl_show_warn=False
210212
)
@@ -277,10 +279,6 @@ spec:
277279
pip install -q opensearch-py requests urllib3
278280
python -u /scripts/load.py
279281
env:
280-
- name: OPENSEARCH_HOST
281-
value: "opensearch"
282-
- name: OPENSEARCH_PORT
283-
value: "9200"
284282
- name: OPENSEARCH_USER
285283
value: "admin"
286284
- name: OPENSEARCH_PASSWORD
@@ -292,6 +290,9 @@ spec:
292290
value: "rag-documents"
293291
- name: EMBEDDINGS_FILE
294292
value: "/data/stackable-docs-embeddings.json"
293+
envFrom:
294+
- configMapRef:
295+
name: opensearch
295296
volumeMounts:
296297
- name: script
297298
mountPath: /scripts

stacks/opensearch-rag/jupyterlab.yaml

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,6 @@ spec:
3131
env:
3232
- name: JUPYTER_PORT
3333
value: "8888"
34-
- name: OPENSEARCH_HOST
35-
value: "opensearch"
36-
- name: OPENSEARCH_PORT
37-
value: "9200"
3834
- name: OPENSEARCH_USER
3935
value: "admin"
4036
- name: OPENSEARCH_PASSWORD
@@ -48,6 +44,9 @@ spec:
4844
value: "11434"
4945
- name: OLLAMA_LLM_MODEL
5046
value: "{{ ollamaLlmModel }}"
47+
envFrom:
48+
- configMapRef:
49+
name: opensearch
5150
ports:
5251
- name: http
5352
containerPort: 8888

stacks/opensearch-rag/opensearch-rag.ipynb

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,16 +68,18 @@
6868
"warnings.filterwarnings('ignore')\n",
6969
"\n",
7070
"# Configuration from environment variables\n",
71-
"OPENSEARCH_HOST = os.getenv('OPENSEARCH_HOST')\n",
71+
"OPENSEARCH_HOSTS = os.getenv('OPENSEARCH_HOSTS')\n",
72+
"OPENSEARCH_HOSTNAME = os.getenv('OPENSEARCH_HOSTNAME')\n",
7273
"OPENSEARCH_PORT = int(os.getenv('OPENSEARCH_PORT'))\n",
74+
"OPENSEARCH_PROTOCOL = os.getenv('OPENSEARCH_PROTOCOL')\n",
7375
"OPENSEARCH_USER = os.getenv('OPENSEARCH_USER')\n",
7476
"OPENSEARCH_PASSWORD = os.getenv('OPENSEARCH_PASSWORD')\n",
7577
"OLLAMA_HOST = os.getenv('OLLAMA_HOST')\n",
7678
"OLLAMA_PORT = int(os.getenv('OLLAMA_PORT'))\n",
7779
"OLLAMA_LLM_MODEL = os.getenv('OLLAMA_LLM_MODEL')\n",
7880
"INDEX_NAME = 'rag-documents'\n",
7981
"\n",
80-
"print(f\"OpenSearch: {OPENSEARCH_HOST}:{OPENSEARCH_PORT}\")\n",
82+
"print(f\"OpenSearch: {OPENSEARCH_HOSTS}\")\n",
8183
"print(f\"Ollama: {OLLAMA_HOST}:{OLLAMA_PORT} using {OLLAMA_LLM_MODEL} to generate responses\")\n",
8284
"print(f\"Index: {INDEX_NAME}\")"
8385
]
@@ -101,9 +103,9 @@
101103
"source": [
102104
"# Initialize OpenSearch client\n",
103105
"opensearch_client = OpenSearch(\n",
104-
" hosts=[{'host': OPENSEARCH_HOST, 'port': OPENSEARCH_PORT}],\n",
106+
" hosts=[{'host': OPENSEARCH_HOSTNAME, 'port': OPENSEARCH_PORT}],\n",
105107
" http_auth=(OPENSEARCH_USER, OPENSEARCH_PASSWORD),\n",
106-
" use_ssl=True,\n",
108+
" use_ssl=OPENSEARCH_PROTOCOL == 'https',\n",
107109
" verify_certs=False,\n",
108110
" ssl_show_warn=False\n",
109111
")\n",

0 commit comments

Comments
 (0)