Semantic search

response1 = client.files.upload(path="./well_report.pdf")
document_id = response1.id
print(document_id)

import time

status_path = f"/api/v1/projects/{client.config.project}/documents/status"

body = {
    "items": [
        {
            "id": document_id
        }
    ]
}

while True:
    response2 = client.post(status_path, json=body, headers={"cdf-version": "beta"}).json()

    status = response2["items"][0]["passages"]["status"]
    print(f"status: {status}")

    if status in {"waiting", "running"}:
        time.sleep(5)
        continue

    break

import json

search_path = f"/api/v1/projects/{client.config.project}/documents/passages/search"

body = {
    "limit": 3,
    "filter": {
        "and": [
            {
                "equals": {
                    "property": ["document", "id"],
                    "value": document_id
                }
            },
            {
                "semanticSearch": {
                    "property": ["content"],
                    "value": "Where is the Volve field located?"
                }
            }
        ]
    }
}

response3 = client.post(search_path, json=body).json()

print(json.dumps(response3, indent=2))

{
  "items": [
    {
      "document": {
        "id": 8482943812996165,
      },
    "text": " 1. Introduction\n Well: 15/9-F-14\n 1.1. Purpose of the project\n Volve is an oil field located in the southern part of the North Sea approximately eight kilometres north of Sleipner \u00d8st. The sea depth is approximately 90 metres. The development concept is a jack-up processing and drilling facility and a vessel for storing stabilized oil.\n The reservoir contains oil in a combined stratigraphic and structural trap with Jurassic and Triassic sandstones in the Hugin formation. The western part of the structure is heavily faulted and it is uncertain if there is communication across the faults.\n Volve will be recovered by water injection.\n The rich gas will be transported to Sleipner A and exported onwards from there.\n Well F-14 is designed as an oil producer in the Volve development drilling program. The F-14 well is the second oil producer on the Volve Field. The well is located in a structural high position on the crest of the structure 800m up flank of the NO 15/9-19A discovery well.\n Main objectives:\n To establish an oil production well draining the northern and northwest flank of the Volve structure, the reservoir being the Hugin Formation.\n The upper fault block will be pressure supported by water injection in the F-4 and the F-5 Injection wells, whereas the lower fault block will receive support only from F-5.\n Other objectives:\n Collect data for planning of future wells and optimalization of production and well design.\n \ufffd \ufffd Collect pressure data.\n \ufffd \ufffd Logs, inclusive velocity.\n \ufffd \ufffd Stratigraphy, lithology, reservoir data.\n 3",
    "locations": [
      {
        "pageNumber": 3,
        "left": 57.59,
        "right": 60.66,
        "top": 43.58,
        "bottom": 53.54
      }
    ]
  },
  ...
  ]
}