import torch
from transformers import BertTokenizer, BertModel

tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
model = BertModel.from_pretrained("bert-base-uncased")


def get_bert_embedding(text):
    inputs = tokenizer(text, return_tensors="pt", max_length=128, truncation=True, padding="max_length")
    with torch.no_grad():
        outputs = model(**inputs)
    return outputs.last_hidden_state[:, :3, :].numpy()

def print_infos():
    docs = ["Город барбекю, занимающий площадь в 100 акров, был успешно построен в Цзыбо всего за 20 дней и теперь стал популярным местом, где тысячи людей соревнуются за "места для барбекю".",
            «Новый барбекю-город площадью 100 акров в Цзыбо был построен всего за 20 дней, что привлекло множество любителей барбекю. В настоящее время трудно найти «места для барбекю».»,
            «В Цзыбо город барбекю площадью 100 акров, построенный за 20 дней, стал центром всеобщего внимания. Тысячи людей соперничают за различные вкусные барбекю, которые трудно найти».
            «Цзыбо обычно относится к городу Цзыбо. Город Цзыбо, называемый «Цзы», является древней столицей государства Ци, городом префектурного уровня, находящимся под юрисдикцией провинции Шаньдун, и крупным городом типа II»]
    for doc in docs:
        print( f"Vector for '{doc}':", get_bert_embedding( doc ) )
    
if __name__ == '__main__':
    print_infos()

Vector for «Город барбекю», занимающий площадь в 100 акров, был успешно построен в Цзыбо всего за 20 дней и теперь стал популярным местом для тысяч людей, состязающихся за «места для барбекю». ': [[[-0.2703271   0.38279012 -0.29274252 ... -0.24937081  0.7212287
    0.0751707 ]
  [ 0.01726123  0.1450473   0.16286954 ... -0.20245396  1.1556625
   -0.112049  ]
  [ 0.51697373 -0.01454506  0.1063835  ... -0.2986216   0.69151103
    0.13124703]]]
Vector for «Новый город для барбекю площадью 100 акров в Цзыбо был построен всего за 20 дней, что привлекло множество любителей барбекю. В настоящее время трудно найти «места для барбекю». ': [[[-0.22879271  0.43286988 -0.21742335 ... -0.22972387  0.75263715
    0.03716223]
  [ 0.1252176  -0.02892866  0.17054333 ... -0.30524847  0.94903445
   -0.46865308]
  [ 0.42650488  0.34019586 -0.01442122 ... -0.17345914  0.6688627
   -0.75012964]]]

PUT /my_vector_index
{
  "mappings": {
    "properties": {
      "title": {
        "type": "text"
      },
      "content_vector": {
        "type": "dense_vector",
        "dims": 3
      }
    }
  }
}

POST my_vector_index/_bulk
{"index":{"_id":1}}
{"title":"Город барбекю, занимающий площадь в 100 акров, был успешно построен в Цзыбо всего за 20 дней и теперь стал популярным местом для тысяч людей, соревнующихся за "места для барбекю."","content_vector" :[-0.2703271, 0.38279012, -0.29274252]}
{"index":{"_id":2}}
{"title":«Новый барбекю-город площадью 100 акров в Цзыбо был построен всего за 20 дней, что привлекло множество любителей барбекю. В настоящее время трудно найти «места для барбекю».»,"content_vector":[-0.22879271, 0.43286988, -0.21742335]}
{"index":{"_id":3}}
{"title":«В Цзыбо город барбекю площадью 100 акров, построенный за 20 дней, стал центром всеобщего внимания. Тысячи людей соперничают за различные вкусные барбекю, которые трудно найти»."content_vector":[-0.24912262, 0.40769795, -0.26663426]}
{"index":{"_id":4}}
{"title":«Цзыбо обычно относится к городу Цзыбо. Город Цзыбо, называемый «Цзы», является древней столицей государства Ци, городом префектурного уровня, находящимся под юрисдикцией провинции Шаньдун, и крупным городом типа II","content_vector":["0.32247472, 0.19048998, -0.36749798]}

GET my_vector_index/_search
{
  "query": {
    "script_score": {
      "query": {
        "match_all": {}
      },
      "script": {
        "source": "cosineSimilarity(params.query_vector, 'content_vector') + 1.0", 
        "params": {
          "query_vector": [-0.2703271, 0.38279012, -0.29274252]  
        }
      }
    }
  }
}