|
5 | 5 | CLOUD_ID = os.environ["CLOUD_ID"] |
6 | 6 | ES_USER = os.environ["ELASTICSEARCH_USERNAME"] |
7 | 7 | ES_PASSWORD = os.environ["ELASTICSEARCH_PASSWORD"] |
| 8 | +ELSER_MODEL = os.getenv("ELSER_MODEL", ".elser_model_2") |
| 9 | +INDEX = os.getenv("ES_INDEX", "search-movies") |
8 | 10 |
|
9 | 11 | datasets = { |
10 | 12 | "movies": { |
@@ -84,7 +86,7 @@ def get_text_expansion_request_body(query, size=10, **options): |
84 | 86 | te = {"text_expansion": {}} |
85 | 87 | te["text_expansion"][field] = { |
86 | 88 | "model_text": query, |
87 | | - "model_id": ".elser_model_1", |
| 89 | + "model_id": ELSER_MODEL, |
88 | 90 | "boost": boost, |
89 | 91 | } |
90 | 92 | text_expansions.append(te) |
@@ -114,7 +116,7 @@ def get_text_expansion_request_body(query, size=10, **options): |
114 | 116 | te = {"text_expansion": {}} |
115 | 117 | te["text_expansion"][field] = { |
116 | 118 | "model_text": query, |
117 | | - "model_id": ".elser_model_1", |
| 119 | + "model_id": ELSER_MODEL, |
118 | 120 | "boost": boost, |
119 | 121 | } |
120 | 122 | text_expansions.append(te) |
@@ -159,7 +161,7 @@ def get_hybrid_search_rrf_request_body(query, size=10, **options): |
159 | 161 | te = {"text_expansion": {}} |
160 | 162 | te["text_expansion"][field] = { |
161 | 163 | "model_text": query, |
162 | | - "model_id": ".elser_model_1", |
| 164 | + "model_id": ELSER_MODEL, |
163 | 165 | "boost": boost, |
164 | 166 | } |
165 | 167 | text_expansions.append(te) |
|
0 commit comments