|
87 | 87 | "\n", |
88 | 88 | "ELASTIC_PASSWORD = getpass(\"ELASTIC PASSWORD: \")\n", |
89 | 89 | "\n", |
90 | | - "# e.g. \n", |
| 90 | + "# e.g.\n", |
91 | 91 | "# hosts = \"http://es-xxx.elasticsearch.aliyuncs.com:9200\"\n", |
92 | | - "hosts=getpass(\"Host: \")\n", |
| 92 | + "hosts = getpass(\"Host: \")\n", |
93 | 93 | "\n", |
94 | 94 | "# Create the client instance\n", |
95 | 95 | "client = Elasticsearch(\n", |
96 | 96 | " # For local development\n", |
97 | 97 | " # hosts=[\"http://localhost:9200\"]\n", |
98 | 98 | " hosts,\n", |
99 | | - " basic_auth=(ELASTIC_USER, ELASTIC_PASSWORD)\n", |
| 99 | + " basic_auth=(ELASTIC_USER, ELASTIC_PASSWORD),\n", |
100 | 100 | ")" |
101 | 101 | ] |
102 | 102 | }, |
|
146 | 146 | "API_KEY = getpass(\"Enter AlibabaCloud AI Search API key: \")\n", |
147 | 147 | "\n", |
148 | 148 | "# Please enter your AlibabaCloud AI Search host here\n", |
149 | | - "# e.g. \n", |
| 149 | + "# e.g.\n", |
150 | 150 | "# HOST = \"default-xxx.opensearch.aliyuncs.com\"\n", |
151 | 151 | "HOST = \"\"\n", |
152 | 152 | "\n", |
153 | 153 | "# Please enter your AlibabaCloud AI Search workspace name here\n", |
154 | | - "# e.g. \n", |
| 154 | + "# e.g.\n", |
155 | 155 | "# WORKSPACE = \"default\"\n", |
156 | 156 | "WORKSPACE = \"\"\n", |
157 | 157 | "\n", |
|
163 | 163 | " \"service_settings\": {\n", |
164 | 164 | " \"api_key\": API_KEY,\n", |
165 | 165 | " \"service_id\": \"ops-text-embedding-001\",\n", |
166 | | - " \"host\" : HOST,\n", |
167 | | - " \"workspace\" : WORKSPACE\n", |
168 | | - " }\n", |
| 166 | + " \"host\": HOST,\n", |
| 167 | + " \"workspace\": WORKSPACE,\n", |
| 168 | + " },\n", |
169 | 169 | " },\n", |
170 | 170 | ")\n", |
171 | 171 | "\n", |
|
177 | 177 | " \"service_settings\": {\n", |
178 | 178 | " \"api_key\": API_KEY,\n", |
179 | 179 | " \"service_id\": \"ops-text-sparse-embedding-001\",\n", |
180 | | - " \"host\" : HOST,\n", |
181 | | - " \"workspace\" : WORKSPACE\n", |
182 | | - " }\n", |
| 180 | + " \"host\": HOST,\n", |
| 181 | + " \"workspace\": WORKSPACE,\n", |
| 182 | + " },\n", |
183 | 183 | " },\n", |
184 | 184 | ")" |
185 | 185 | ] |
|
231 | 231 | " \"output_field\": \"plot_sparse_embedding\",\n", |
232 | 232 | " },\n", |
233 | 233 | " }\n", |
234 | | - " }\n", |
| 234 | + " },\n", |
235 | 235 | " ],\n", |
236 | 236 | ")" |
237 | 237 | ] |
|
277 | 277 | } |
278 | 278 | ], |
279 | 279 | "source": [ |
280 | | - "client.indices.delete(index=\"alibaba-text-embeddings-notebook-test\", ignore_unavailable=True)\n", |
| 280 | + "client.indices.delete(\n", |
| 281 | + " index=\"alibaba-text-embeddings-notebook-test\", ignore_unavailable=True\n", |
| 282 | + ")\n", |
281 | 283 | "client.indices.create(\n", |
282 | 284 | " index=\"alibaba-text-embeddings-notebook-test\",\n", |
283 | 285 | " settings={\"index\": {\"default_pipeline\": \"alibaba-embeddings-notebook-test\"}},\n", |
284 | 286 | " mappings={\n", |
285 | 287 | " \"properties\": {\n", |
286 | 288 | " \"plot\": {\"type\": \"text\"},\n", |
287 | | - " \"plot_embedding\": {\n", |
288 | | - " \"type\": \"dense_vector\",\n", |
289 | | - " \"dims\": 1536,\n", |
290 | | - " \"index\": \"true\"\n", |
291 | | - " },\n", |
292 | | - " \"plot_sparse_embedding\": {\n", |
293 | | - " \"type\":\"sparse_vector\"\n", |
294 | | - " }\n", |
| 289 | + " \"plot_embedding\": {\"type\": \"dense_vector\", \"dims\": 1536, \"index\": \"true\"},\n", |
| 290 | + " \"plot_sparse_embedding\": {\"type\": \"sparse_vector\"},\n", |
295 | 291 | " }\n", |
296 | 292 | " },\n", |
297 | 293 | ")" |
|
437 | 433 | " query={\n", |
438 | 434 | " \"sparse_vector\": {\n", |
439 | 435 | " \"field\": \"plot_sparse_embedding\",\n", |
440 | | - " \"inference_id\":\"os-sparse-embeddings-notebook-test\",\n", |
441 | | - " \"query\":\"crime dynasty\"\n", |
| 436 | + " \"inference_id\": \"os-sparse-embeddings-notebook-test\",\n", |
| 437 | + " \"query\": \"crime dynasty\",\n", |
442 | 438 | " }\n", |
443 | | - " }\n", |
| 439 | + " },\n", |
444 | 440 | ")\n", |
445 | 441 | "\n", |
446 | 442 | "for hit in response[\"hits\"][\"hits\"]:\n", |
|
0 commit comments