Skip to content

Commit bf55899

Browse files
committed
Now reads elser model and index from envar
1 parent 654f842 commit bf55899

File tree

1 file changed

+5
-3
lines changed
  • example-apps/relevance-workbench/app-api

1 file changed

+5
-3
lines changed

example-apps/relevance-workbench/app-api/app.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
CLOUD_ID = os.environ["CLOUD_ID"]
66
ES_USER = os.environ["ELASTICSEARCH_USERNAME"]
77
ES_PASSWORD = os.environ["ELASTICSEARCH_PASSWORD"]
8+
ELSER_MODEL = os.getenv("ELSER_MODEL", ".elser_model_2")
9+
INDEX = os.getenv("ES_INDEX", "search-movies")
810

911
datasets = {
1012
"movies": {
@@ -84,7 +86,7 @@ def get_text_expansion_request_body(query, size=10, **options):
8486
te = {"text_expansion": {}}
8587
te["text_expansion"][field] = {
8688
"model_text": query,
87-
"model_id": ".elser_model_1",
89+
"model_id": ELSER_MODEL,
8890
"boost": boost,
8991
}
9092
text_expansions.append(te)
@@ -114,7 +116,7 @@ def get_text_expansion_request_body(query, size=10, **options):
114116
te = {"text_expansion": {}}
115117
te["text_expansion"][field] = {
116118
"model_text": query,
117-
"model_id": ".elser_model_1",
119+
"model_id": ELSER_MODEL,
118120
"boost": boost,
119121
}
120122
text_expansions.append(te)
@@ -159,7 +161,7 @@ def get_hybrid_search_rrf_request_body(query, size=10, **options):
159161
te = {"text_expansion": {}}
160162
te["text_expansion"][field] = {
161163
"model_text": query,
162-
"model_id": ".elser_model_1",
164+
"model_id": ELSER_MODEL,
163165
"boost": boost,
164166
}
165167
text_expansions.append(te)

0 commit comments

Comments
 (0)