import weaviate
client = weaviate.Client("http://localhost:8080")

classObj = {
    "class": "Superhero",
    "description": "A class describing a super hero",
    "properties": [
        {
            "name": "name",
            "dataType": [
                "string"
            ],
            "description": "Name of the super hero"
        }
    ],
    "vectorizer": "text2vec-contextionary" # Tell Weaviate to vectorize the content
}
client.schema.create_class(classObj) # returns None if successful

batman = {
    "name": "Batman"
}
superman = {
    "name": "Superman"
}
batch = weaviate.ObjectsBatchRequest()
batch.add(batman, "Superhero")
batch.add(superman, "Superhero")
client.batch.create(batch)

def findAlterego(alterEgo):
    whoIsIt = client.query.get(
        "Superhero",
        ["name", "_additional {certainty, id } "]
    ).with_near_text({
        "concepts": [alterEgo] # query that gets vectorized 🪄
    }).do()

    print(
        alterEgo, "is", whoIsIt['data']['Get']['Superhero'][0]['name'],
        "with a certainy of", whoIsIt['data']['Get']['Superhero'][0]['_additional']['certainty']
    )

findAlterego("Clark Kent")  # prints something like: Clark Kent is Superman with a certainy of 0.6026741
findAlterego("Bruce Wayne") # prints something like: Bruce Wayne is Batman with a certainy of 0.6895526

def showVectorForAlterego(alterEgo):
    whoIsIt = client.query.get(
        "Superhero",
        ["_additional {id} "]
    ).with_near_text({
        "concepts": [alterEgo] # query that gets vectorized 🪄
    }).do()

    getVector = client.data_object.get_by_id(
        whoIsIt['data']['Get']['Superhero'][0]['_additional']['id'],
        additional_properties=["vector"]
    )

    print(
        "The vector for",
        alterEgo,
        "is",
        getVector['vector']
    )

showVectorForAlterego("Clark Kent") # prints something like: The vector for Clark Kent is [-0.05484624, 0.08283167, -0.3002325, ...etc...