|
283 | 283 | "\n", |
284 | 284 | "t = (Tag(\"credit_score\") == \"high\") & (Text(\"job\") % \"enginee*\") & (Num(\"age\") > 17)\n", |
285 | 285 | "\n", |
286 | | - "v = VectorQuery([0.1, 0.1, 0.5],\n", |
287 | | - " \"user_embedding\",\n", |
288 | | - " return_fields=[\"user\", \"credit_score\", \"age\", \"job\", \"office_location\"],\n", |
289 | | - " filter_expression=t)\n", |
| 286 | + "v = VectorQuery(\n", |
| 287 | + " vector=[0.1, 0.1, 0.5],\n", |
| 288 | + " vector_field_name=\"user_embedding\",\n", |
| 289 | + " return_fields=[\"user\", \"credit_score\", \"age\", \"job\", \"office_location\"],\n", |
| 290 | + " filter_expression=t\n", |
| 291 | + ")\n", |
290 | 292 | "\n", |
291 | 293 | "\n", |
292 | 294 | "results = hindex.query(v)\n", |
|
395 | 397 | "metadata": {}, |
396 | 398 | "outputs": [], |
397 | 399 | "source": [ |
398 | | - "import numpy as np\n", |
399 | | - "\n", |
400 | 400 | "json_data = data.copy()\n", |
401 | 401 | "\n", |
402 | 402 | "for d in json_data:\n", |
|
593 | 593 | "outputs": [], |
594 | 594 | "source": [ |
595 | 595 | "# construct a search index from the json schema\n", |
596 | | - "bike_index = SearchIndex.from_dict(bike_schema)\n", |
597 | | - "\n", |
598 | | - "# connect to local redis instance\n", |
599 | | - "bike_index.connect(\"redis://localhost:6379\")\n", |
| 596 | + "bike_index = SearchIndex.from_dict(bike_schema, redis_url=\"redis://localhost:6379\")\n", |
600 | 597 | "\n", |
601 | 598 | "# create the index (no data yet)\n", |
602 | 599 | "bike_index.create(overwrite=True)" |
|
633 | 630 | "\n", |
634 | 631 | "vec = emb_model.embed(\"I'd like a bike for aggressive riding\")\n", |
635 | 632 | "\n", |
636 | | - "v = VectorQuery(vector=vec,\n", |
637 | | - " vector_field_name=\"bike_embedding\",\n", |
638 | | - " return_fields=[\n", |
639 | | - " \"brand\",\n", |
640 | | - " \"name\",\n", |
641 | | - " \"$.metadata.type\"\n", |
642 | | - " ]\n", |
643 | | - " )\n", |
| 633 | + "v = VectorQuery(\n", |
| 634 | + " vector=vec,\n", |
| 635 | + " vector_field_name=\"bike_embedding\",\n", |
| 636 | + " return_fields=[\n", |
| 637 | + " \"brand\",\n", |
| 638 | + " \"name\",\n", |
| 639 | + " \"$.metadata.type\"\n", |
| 640 | + " ]\n", |
| 641 | + ")\n", |
644 | 642 | "\n", |
645 | 643 | "\n", |
646 | 644 | "results = bike_index.query(v)" |
|
0 commit comments