nucliadb-sdk 0.0.0.post2__py3-none-any.whl

nucliadb_sdk/tests/test_search_resource.py

@@ -0,0 +1,346 @@
+# Copyright (C) 2021 Bosutech XXI S.L.
+#
+# nucliadb is offered under the AGPL v3.0 and as commercial software.
+# For commercial licensing, contact us at info@nuclia.com.
+#
+# AGPL:
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+from typing import Any, Dict
+
+import pytest
+
+from nucliadb_sdk.knowledgebox import KnowledgeBox
+from nucliadb_sdk.labels import Label
+from nucliadb_sdk.vectors import Vector
+
+DATA: Dict[str, Any] = {
+    "text": [
+        "Sunday afternoon walking through Venice in the sun with @user ️ ️ ️ @ Abbot Kinney, Venice",
+        "Time for some BBQ and whiskey libations. Chomp, belch, chomp! (@ Lucille's Smokehouse Bar-B-Que)",
+        "Love love love all these people ️ ️ ️ #friends #bff #celebrate #blessed #sundayfunday @ San…",
+        '️ ️ ️ ️ @ Toys"R"Us',
+        "Man these are the funniest kids ever!! That face! #HappyBirthdayBubb @ FLIPnOUT Xtreme",
+        "#sandiego @ San Diego, California",
+        "My little ️ ️ ️ ️ ️ #ObsessedWithMyDog @ Cafe Solstice Capitol Hill",
+        "More #tinyepic things #tinyepicwestern, this one is crazy @user I may be one of your…",
+        "Last night ️ @ Omnia Night Club At Caesars Palace",
+        "friendship at its finest. ....#pixar #toystory #buzz #woody #friends #friendship #bff…",
+        "I L VE working for a cause! Yesterday's balloon decor for SNN 11th Annual Back 2 School Health…",
+        "Dem shoes tho! Lol! ...I'm getting some! Get ready copleypriceymca…",
+        "I love weekends with you, calebrancourt @ Cannery Row At Monterey Bay, CA",
+        "Yesss! August is full of love and birthdays ️ @ Los Angeles,…",
+        "I took a pic with Eminem! *still not over that performance vibe from #parrisproject!…",
+        "I L VE working for a cause! Yesterday's balloon decor for SNN 11th Annual Back 2 School Health…",
+        "Sundays are all about the cute babies and dogs! #Ballard #sundaymarket #littelestmodel…",
+        "it was too hot .. trying to find shade behind @user at the roadraceengineering booth…",
+        "Sometimes you just have to be a kid Skyspace, Los Angeles. #CRStylesTravel #FunDay #LALife…",
+        "had an amazing night, with some pretty amazing people @ Pandora Summer",
+        "Thank you @user for an incredible night last night @ Shrine Auditorium &amp; Expo Hall",
+        "thanks for the amazing night @ Pandora Summer",
+        "@user draw the line ~",
+        "This is why I hate when Taylor visits #Bunny #NotCaturday @ Our House",
+        "thank you to my boys for performing so well last night, I love you all very much @ Pandora…",
+        "2nd theater and 3rd movie of the day a new record #badmoms #suicidesquad #sausageparty (at @user",
+        "In honor of #happygayunclesday ️ #happygunclesday #guncle #guncles @ Chula Vista, California",
+        "I think it's nap time for us @ Los Angeles, California",
+        "I L VE working for a cause! Yesterday's balloon decor for SNN 11th Annual Back 2 School Health…",
+        "Chasing #mclarenp1 in Malibu today.________________________________________ : credit: @user",
+        "Sunday afternoon at home! Drinking good wine and enjoying this incredible view to the sound of…",
+        "#Lunch @user best #Ribs in #SanDiego (at @user in San Diego, CA)",
+        "Here's to the start of a great adventure. Niners today, Alaska tomorrow. #grams @ Levi's Stadium",
+        "Really horrible seats for a really awful Sunday afternoon! Life is really bad #soundersfc.…",
+        "Enjoy your last days of summer! Clearly, mauricethewhale is enjoying his! #ladykiller…",
+        "Surprised everyone by actually surviving a weekend in the wild (however still came close to…",
+        "In the zone | : @user #colorsworldwide #RnBOnly @ The Regent",
+        "“Me &amp; Catherine Sutherland!!!” #PowerMorphicon2016 #PowerRangersZeo #PowerRangersTurbo…",
+        "Taped up and dripped out. Know what I'm talking bout #paint #painting #diy #homeimprovement…",
+        "princesscruises @ Port of Seattle, Pier 91",
+        "Stay classy #SanDiego (@ San Diego International Airport - @user in San Diego, CA)",
+        "I ️ the people I work with. Enjoying our Sunday watching the first…",
+        "Another big happy 18th birthday to my partner in crime ️ I love u and all the crazy times we've…",
+        "Day is DONE! Thank u INSTA-LIKERS! Rollicking day at FRINGE! C U THURSDAY~~~and then some .…",
+        "SnakeShred recording guitars for the new SnakeSkin record #guitar #recording #api #apipegged…",
+        "My sisters, my best friends #hamessisters #andlewisandclark…",
+        "my absolute favorite place. @ La Jolla Cove Seal Beach",
+        "To live is an awfully big adventure #Seattle #spaceneedle @ Space Needle",
+        "Finally met Snow White ! She was my favorite when i was little :) #SnowWhite #Disneyland @user",
+        "@user loved your coverage at TI as always, now need to get you on some of those Sounders matches...",
+        "Saturdayzzz w/ Lionel Richie @ Harvey's Outdoor Ampitheater",
+        "All the beautiful ladies that helped make last night #APositiveExperience @ Upscale Salon",
+        "Sunday funday indeed #sundayfunday #igers #igdaily #instalike #enjoylife #asian #LA…",
+        "Lights, Camera, Action @ Hollywood Walk of Fame",
+        "Found your Unicorn @user it's a pillow/stuffed toy and every pride weekend it…",
+        "Harry Potter ️ @ TCL Chinese Theatres",
+        "My key was still here...thank the Lawd #littlehoneyvee #doodles #doodleoftheday #illustration…",
+        "️ ️ ️ @ Los Angeles, California",
+        "Absolutely in love with this ethereal glow on holytolidoitseric. Who would rock this look?…",
+        "Getting up that ice wall one ax at a time : pechoi11 @ Mount Shasta, California",
+        "Sephora is my favorite place @ SEPHORA",
+        "Tutorial time w/ @user @user @ Stern Grove Festival",
+        "I love my people! @ Paso Robles, California",
+        "Rad Fashion at the midway SF on a Sunday ️ ️ ️ @ The Midway SF",
+        "Just like the movie... Radiator Springs ️ #cars #inlove…",
+        "Cheers to Bruce on his birthday. ️#westhollywood…",
+        "My bears! #cityfest ️ ️ ️ @ Babycakes San Diego",
+        "gonna miss you guys so much. thank you for an amazing experience ️#ballyhoocarrie…",
+        "MATTRESS HELPER - new again SAGGING #BED #FIRMER #fix #guaranteed #sleepless #needsleep…",
+        "Just that sound alone makes the world stop ️ @ Sausalito, California",
+        "#ExclusiveBaseballPicks are KILLING the #Bookie like always. #WeeklyRecap 8/8-8/14. 96-3…",
+        "My lady love beautiful friend #doubleLtrouble #leilavie #beachour @ Stinson Beach, California",
+        "My date for tonight @ Hotel Las Rocas",
+        "This weekend love was truly in the air but out of all the celeb weddings I have to say…",
+        "@user hi from Tijuana B.C. See you soon",
+        "️ #gramma @ Living Way Christian Fellowship",
+        "Acquired this stunner today at my new favorite spot Sharks Tooth #crassula.…",
+        "Birthday cheers with my favorite people!!#26again #justafewofmygirls #justafewofmygirls…",
+        "today was my first day of work at Victoria Secret SO happy and SO…",
+        "Taking bae to all my favorite #LA spots. #gettycenter @ Getty Center",
+        "So much fun this weekend celebrating the bride to be. #MalliesLastRally @ Las Vegas, Nevada",
+        "Love this picture of me and baby Eve at Disneyland earlier today …",
+        "The one photo I've taken with my Mother that I like for once. -08/13/16, Sat. @ El Dorado…",
+        "Love my bed @ Downey, California",
+        "So cute!! #KnottsBerryFarm #Charlie #Snoopy #Woodstock @ Knott's Berry Farm",
+        "I ️ Faure Requiem @user oboyddd @ Walt Disney Concert Hall",
+        "️ @ Columbia River",
+        "No I don't have any carrots!! #seattlepoloclub #horse #horses @ Seattle Polo &amp; Equestrian Club",
+        "My favourite place in California #YosemiteNationalPark #California @ Yosemite Wilderness",
+        "#sundayfunday #live at allure... @user @user @user",
+        "My babies ️ ️ ️ #cat #dog #catsofinstagram #dogsofinstagram #lovemyfurbabies @ Port Hueneme,…",
+        "Photo taken Thursday at SDCC2016 by MannyLLanura Photography. #widowmaker #overwatch…",
+        "Hello there, my king Jareth #bowie #goblinking #labyrinth @ Los Angeles, California",
+        "These Teens @ ARIA Resort &amp; Casino",
+        "Sunday views #hike #waterfall #waterfallhike #latourellfalls #hikeoregon #hikepnw #whyihike…",
+        '"Beauty begins the moment you decide to be yourself" -Coco Chanel love this!Skin is the largest…',
+        "Caught by the lens of Yc Wong. photo by 8888tiger8888 suit by virginblak #ljifff…",
+        "Weldone champ #babrain #alikhamis #rio #rio2016 #olympics @ Los…",
+        "Took a lil trip to Tahoe #iputhehoeintahoe @ Incline Village At…",
+        "About to hang out with a bunch of #Armenians and #Russians...trying to fit in Sporting my…",
+    ],
+    "label": [
+        12,
+        19,
+        0,
+        0,
+        2,
+        11,
+        0,
+        19,
+        0,
+        7,
+        1,
+        2,
+        9,
+        0,
+        1,
+        1,
+        1,
+        4,
+        19,
+        8,
+        7,
+        8,
+        5,
+        2,
+        8,
+        2,
+        0,
+        1,
+        1,
+        10,
+        8,
+        14,
+        1,
+        16,
+        2,
+        14,
+        18,
+        6,
+        6,
+        8,
+        14,
+        0,
+        0,
+        9,
+        19,
+        13,
+        3,
+        8,
+        0,
+        16,
+        7,
+        7,
+        6,
+        6,
+        2,
+        0,
+        0,
+        0,
+        1,
+        18,
+        7,
+        2,
+        8,
+        0,
+        0,
+        0,
+        0,
+        0,
+        7,
+        0,
+        19,
+        1,
+        1,
+        0,
+        14,
+        0,
+        1,
+        14,
+        1,
+        8,
+        3,
+        1,
+        5,
+        1,
+        1,
+        0,
+        12,
+        2,
+        1,
+        6,
+        0,
+        10,
+        0,
+        14,
+        12,
+        13,
+        18,
+        7,
+        8,
+        19,
+    ],
+}
+
+
+def test_search_resource(knowledgebox: KnowledgeBox):
+    # Lets create a bunch of resources
+
+    text: str
+    for index, text in enumerate(DATA["text"]):
+        if index == 50:
+            break
+        label = DATA["label"][index]
+        knowledgebox.upload(
+            text=text,
+            labels=[f"emoji/{label}"],
+            vectors={"all-MiniLM-L6-v2": [1.0, 2.0, 3.0, 2.0]},
+        )
+
+    assert len(knowledgebox) == 50
+    labels = knowledgebox.get_uploaded_labels()
+
+    assert labels["emoji"].count == 50
+    assert labels["emoji"].labels["0"] == 9
+
+    resources = knowledgebox.search(text="love")
+    assert resources.fulltext.total == 5
+    assert len(resources.resources) == 5
+
+    resources = knowledgebox.search(filter=[Label(labelset="emoji", label="0")])
+
+    assert resources.fulltext.total == 9
+
+    resources = knowledgebox.search(filter=["emoji/0"])
+
+    assert resources.fulltext.total == 9
+
+    vector_q = [1.0, 2.0, 3.0, 2.0]
+    resources = knowledgebox.search(
+        vector=vector_q, vectorset="all-MiniLM-L6-v2", min_score=0.70
+    )
+    assert len(resources.sentences.results) == 11
+
+
+def _test_search_resource_simple_label(knowledgebox: KnowledgeBox):
+    # Lets create a bunch of resources
+
+    text: str
+    for index, text in enumerate(DATA["text"]):
+        if index == 50:
+            break
+        label = DATA["label"][index]
+        knowledgebox.upload(
+            text=text,
+            labels=[str(label)],
+            vectors={"all-MiniLM-L6-v2": [1.0, 2.0, 3.0, 2.0]},
+        )
+
+    assert len(knowledgebox) == 50
+    labels = knowledgebox.get_uploaded_labels()
+
+    assert labels["default"].count == 50
+    assert labels["default"].labels["0"] == 9
+
+    resources = knowledgebox.search(text="love")
+    assert resources.fulltext.total == 5
+    assert len(resources.resources) == 5
+
+    resources = knowledgebox.search(filter=["12"])
+
+    vector_q = [1.0, 2.0, 3.0, 2.0]
+    resources = knowledgebox.search(
+        vector=vector_q,
+        vectorset="all-MiniLM-L6-v2",
+    )
+
+
+@pytest.mark.asyncio
+async def _test_search_resource_async(knowledgebox: KnowledgeBox):
+    knowledgebox.new_vectorset("all-MiniLM-L6-v2", 4)
+
+    text: str
+    for index, text in enumerate(DATA["text"]):
+        if index == 50:
+            break
+        label = DATA["label"][index]
+        await knowledgebox.async_upload(
+            text=text,
+            labels=[Label(label=str(label), labelset="emoji")],
+            vectors=[
+                Vector(
+                    value=[1.0, 2.0, 3.0, 2.0],
+                    vectorset="all-MiniLM-L6-v2",
+                )
+            ],
+        )
+
+    assert len(knowledgebox) == 50
+    labels = await knowledgebox.async_get_uploaded_labels()
+
+    assert labels["emoji"].count == 50
+    assert labels["emoji"].labels["0"] == 9
+
+    resources = await knowledgebox.async_search(text="love")
+    assert resources.fulltext.total == 5
+    assert len(resources.resources) == 5
+
+    resources = await knowledgebox.async_search(
+        filter=[Label(labelset="emoji", label="12")]
+    )
+
+    vector_q = [1.0, 2.0, 3.0, 2.0]
+    resources = await knowledgebox.async_search(
+        vector=vector_q,
+        vectorset="all-MiniLM-L6-v2",
+    )

nucliadb_sdk/tests/test_set_resource.py

@@ -0,0 +1,126 @@
+# Copyright (C) 2021 Bosutech XXI S.L.
+#
+# nucliadb is offered under the AGPL v3.0 and as commercial software.
+# For commercial licensing, contact us at info@nuclia.com.
+#
+# AGPL:
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+import pytest
+
+from nucliadb_sdk.entities import Entity
+from nucliadb_sdk.file import File
+from nucliadb_sdk.knowledgebox import KnowledgeBox
+from nucliadb_sdk.vectors import Vector
+
+
+def test_create_resource(knowledgebox: KnowledgeBox):
+    assert knowledgebox.get("mykey1") is None
+
+    resource_id = knowledgebox.upload(
+        key="mykey1",
+        binary=File(data=b"asd", filename="data"),
+        text="I'm Ramon",
+        labels=["labelset/positive"],
+        entities=[Entity(type="NAME", value="Ramon", positions=[(5, 9)])],
+        vectors=[Vector(value=[1.0, 0.2], vectorset="base")],
+    )
+    resource2 = knowledgebox[resource_id]
+    assert (
+        resource2.data is not None
+        and resource2.data.texts is not None
+        and "text" in resource2.data.texts
+    )
+    assert (
+        resource2.data is not None
+        and resource2.data.files is not None
+        and "file" in resource2.data.files
+    )
+
+    resource3 = knowledgebox["mykey1"]
+    assert resource3.id == resource2.id
+
+    assert len(knowledgebox) == 1
+
+    del knowledgebox["mykey1"]
+
+    assert len(knowledgebox) == 0
+
+
+def test_create_resource_dict(knowledgebox: KnowledgeBox):
+    assert knowledgebox.get("mykey1") is None
+
+    resource_id = knowledgebox.upload(
+        key="mykey1",
+        binary=File(data=b"asd", filename="data"),
+        text="I'm Ramon",
+        labels=["labelset/positive"],
+        entities=[Entity(type="NAME", value="Ramon", positions=[(5, 9)])],
+        vectors={"base": [1.0, 2.0]},
+    )
+    resource2 = knowledgebox[resource_id]
+    assert (
+        resource2.data is not None
+        and resource2.data.texts is not None
+        and "text" in resource2.data.texts
+    )
+    assert (
+        resource2.data is not None
+        and resource2.data.files is not None
+        and "file" in resource2.data.files
+    )
+
+    resource3 = knowledgebox["mykey1"]
+    assert resource3.id == resource2.id
+
+    assert len(knowledgebox) == 1
+
+    del knowledgebox["mykey1"]
+
+    assert len(knowledgebox) == 0
+
+
+@pytest.mark.asyncio
+async def test_create_resource_async(knowledgebox: KnowledgeBox):
+    assert await knowledgebox.async_get("mykey1") is None
+
+    resource_id = await knowledgebox.async_upload(
+        key="mykey1",
+        binary=File(data=b"asd", filename="data"),
+        text="asd",
+        labels=["labelset/positive"],
+        entities=[Entity(type="NAME", value="Ramon", positions=[(5, 9)])],
+        vectors=[Vector(value=[1.0, 0.2], vectorset="base")],
+    )
+
+    resource2 = await knowledgebox.async_get(resource_id)
+    assert (
+        resource2.data is not None
+        and resource2.data.texts is not None
+        and "text" in resource2.data.texts
+    )
+    assert (
+        resource2.data is not None
+        and resource2.data.files is not None
+        and "file" in resource2.data.files
+    )
+
+    resource3 = await knowledgebox.async_get("mykey1")
+    assert resource3.id == resource2.id
+
+    assert await knowledgebox.async_len() == 1
+
+    await knowledgebox.async_del("mykey1")
+
+    assert await knowledgebox.async_len() == 0

nucliadb_sdk/utils.py

@@ -0,0 +1,133 @@
+# Copyright (C) 2021 Bosutech XXI S.L.
+#
+# nucliadb is offered under the AGPL v3.0 and as commercial software.
+# For commercial licensing, contact us at info@nuclia.com.
+#
+# AGPL:
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+# europe-1.nuclia.cloud
+# localhost:8080
+
+from typing import List, Optional
+from urllib.parse import urlparse
+from uuid import uuid4
+
+import requests
+
+from nucliadb_models.resource import KnowledgeBoxList, KnowledgeBoxObj
+from nucliadb_models.utils import SlugString
+from nucliadb_sdk.client import Environment, NucliaDBClient
+from nucliadb_sdk.knowledgebox import KnowledgeBox
+
+
+class InvalidHost(Exception):
+    pass
+
+
+class KnowledgeBoxAlreadyExists(Exception):
+    pass
+
+
+def create_knowledge_box(
+    slug: Optional[str] = None,
+    nucliadb_base_url: Optional[str] = "http://localhost:8080",
+):
+    url_obj = urlparse(nucliadb_base_url)
+    if url_obj.hostname and url_obj.hostname.endswith("nuclia.cloud"):  # type: ignore
+        raise InvalidHost(
+            "You can not create a Knowledge Box via API, please use https://nuclia.cloud interface"
+        )
+
+    if slug is None:
+        slug = uuid4().hex
+
+    api_path = f"{nucliadb_base_url}/api/v1"
+    response = requests.post(
+        f"{api_path}/kbs",
+        json={"slug": slug},
+        headers={"X-NUCLIADB-ROLES": "MANAGER"},
+    )
+    if response.status_code == 419:
+        raise KnowledgeBoxAlreadyExists()
+
+    assert response.status_code == 201
+
+    kb = KnowledgeBoxObj.parse_raw(response.content)
+    kbid = kb.uuid
+
+    url = f"{api_path}/kb/{kbid}"
+    client = NucliaDBClient(environment=Environment.OSS, url=url)
+
+    return KnowledgeBox(client)
+
+
+def get_kb(
+    slug: str,
+    nucliadb_base_url: Optional[str] = "http://localhost:8080",
+) -> Optional[KnowledgeBox]:
+    url_obj = urlparse(nucliadb_base_url)
+    if url_obj.hostname and url_obj.hostname.endswith("nuclia.cloud"):  # type: ignore
+        raise InvalidHost(
+            "You can not create a Knowledge Box via API, please use https://nuclia.cloud interface"
+        )
+
+    api_path = f"{nucliadb_base_url}/api/v1"
+    response = requests.get(
+        f"{api_path}/kb/s/{slug}",
+        headers={"X-NUCLIADB-ROLES": "READER"},
+    )
+
+    if response.status_code == 404:
+        return None
+
+    assert response.status_code == 200
+
+    kb = KnowledgeBoxObj.parse_raw(response.content)
+    kbid = kb.uuid
+
+    url = f"{api_path}/kb/{kbid}"
+    client = NucliaDBClient(environment=Environment.OSS, url=url)
+
+    return KnowledgeBox(client)
+
+
+def get_or_create(
+    slug: str, nucliadb_base_url: Optional[str] = "http://localhost:8080"
+):
+    kb = get_kb(slug, nucliadb_base_url)
+    if kb is None:
+        kb = create_knowledge_box(slug, nucliadb_base_url)
+    return kb
+
+
+def delete_kb(slug: str, nucliadb_base_url: Optional[str] = "http://localhost:8080"):
+    kb = get_kb(slug, nucliadb_base_url)
+    if kb is None or kb.client.url is None:
+        raise AttributeError("URL should not be none")
+    response = requests.delete(kb.client.url, headers={"X-NUCLIADB-ROLES": f"MANAGER"})
+    assert response.status_code == 200
+
+
+def list_kbs(
+    nucliadb_base_url: Optional[str] = "http://localhost:8080",
+) -> List[Optional[SlugString]]:
+    response = requests.get(
+        f"{nucliadb_base_url}/kbs",
+        headers={"X-NUCLIADB-ROLES": "MANAGER"},
+    )
+
+    assert response.status_code == 200
+    kbs = KnowledgeBoxList.parse_raw(response.content)
+    return [kb.slug for kb in kbs.kbs]

nucliadb_sdk/vectors.py

@@ -0,0 +1,51 @@
+# Copyright (C) 2021 Bosutech XXI S.L.
+#
+# nucliadb is offered under the AGPL v3.0 and as commercial software.
+# For commercial licensing, contact us at info@nuclia.com.
+#
+# AGPL:
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+from dataclasses import dataclass
+from typing import Any, List, Optional, Tuple
+
+
+@dataclass
+class Vector:
+    value: List[float]
+    vectorset: str
+    key: Optional[str] = None
+    positions: Optional[Tuple[int, int]] = None
+
+
+Vectors = List[Vector]
+
+
+def convert_vector(vector: Any) -> List[float]:
+    if (
+        vector.__class__.__module__ == "numpy"
+        and vector.__class__.__name__ == "ndarray"
+    ):
+        vector = vector.tolist()
+
+    if (
+        vector.__class__.__module__ == "tensorflow.python.framework.ops"
+        and vector.__class__.__name__ == "EagerTensor"
+    ):
+        vector = vector.numpy().tolist()
+
+    if vector.__class__.__module__ == "torch" and vector.__class__.__name__ == "Tensor":
+        vector = vector.tolist()
+
+    return vector

nucliadb_sdk-0.0.0.post2.dist-info/METADATA

@@ -0,0 +1,25 @@
+Metadata-Version: 2.1
+Name: nucliadb-sdk
+Version: 0.0.0.post2
+Summary: UNKNOWN
+Home-page: https://nucliadb.com
+Author: UNKNOWN
+Author-email: UNKNOWN
+License: BSD
+Platform: UNKNOWN
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
+Requires-Dist: httpx
+Requires-Dist: nucliadb-models (==0.0.0-post2)
+
+# NucliaDB SDK
+
+NucliaDB Cloud and NucliaDB OSS sdk
+
+Update and compile the nucliadb image locally before running the tests.
+From nucliadb root dir:
+`make build-nucliadb-local-withbinding`
+
+