🎉 New Destination: Weaviate

airbyte-config/init/src/main/resources/seed/destination_definitions.yaml

@@ -436,3 +436,9 @@
   icon: teradata.svg
   documentationUrl: https://docs.airbyte.io/integrations/destinations/teradata
   releaseStage: alpha
+- name: Weaviate
+  destinationDefinitionId: 7b7d7a0d-954c-45a0-bcfc-39a634b97736
+  dockerRepository: airbyte/destination-weaviate
+  dockerImageTag: 0.1.0
+  documentationUrl: https://docs.airbyte.com/integrations/destinations/weaviate
+  releaseStage: alpha

airbyte-config/init/src/main/resources/seed/destination_specs.yaml

@@ -7297,3 +7297,53 @@
     supported_destination_sync_modes:
     - "overwrite"
     - "append"
+- dockerImage: "airbyte/destination-weaviate:0.1.0"
+  spec:
+    documentationUrl: "https://docs.airbyte.com/integrations/destinations/weaviate"
+    connectionSpecification:
+      $schema: "http://json-schema.org/draft-07/schema#"
+      title: "Destination Weaviate"
+      type: "object"
+      required:
+      - "url"
+      additionalProperties: false
+      properties:
+        url:
+          type: "string"
+          description: "The URL to the weaviate instance"
+          examples:
+          - "http://localhost:8080"
+          - "https://your-instance.semi.network"
+        username:
+          type: "string"
+          description: "Username used with OIDC authentication"
+          examples:
+          - "xyz@weaviate.io"
+        password:
+          type: "string"
+          description: "Password used with OIDC authentication"
+          airbyte_secret: true
+        batch_size:
+          type: "integer"
+          description: "Batch size for writing to Weaviate"
+          default: 100
+        vectors:
+          type: "string"
+          description: "Comma separated list of strings of `stream_name.vector_column_name`\
+            \ to specify which field holds the vectors."
+          examples:
+          - "my_table.my_vector_column, another_table.vector"
+          - "mytable.vector"
+        id_schema:
+          type: "string"
+          description: "Comma separated list of strings of `stream_name.id_column_name`\
+            \ to specify which field holds the ID of the record."
+          examples:
+          - "my_table.my_id_column, another_table.id"
+          - "users.user_id"
+    supportsIncremental: true
+    supportsNormalization: false
+    supportsDBT: false
+    supported_destination_sync_modes:
+    - "append"
+    - "overwrite"

airbyte-integrations/connectors/destination-weaviate/.dockerignore

@@ -0,0 +1,5 @@
+*
+!Dockerfile
+!main.py
+!destination_weaviate
+!setup.py

airbyte-integrations/connectors/destination-weaviate/Dockerfile

@@ -0,0 +1,38 @@
+FROM python:3.9.11-alpine3.15 as base
+
+# build and load all requirements
+FROM base as builder
+WORKDIR /airbyte/integration_code
+
+# upgrade pip to the latest version
+RUN apk --no-cache upgrade \
+    && pip install --upgrade pip \
+    && apk --no-cache add tzdata build-base
+
+
+COPY setup.py ./
+# install necessary packages to a temporary folder
+RUN pip install --prefix=/install .
+
+# build a clean environment
+FROM base
+WORKDIR /airbyte/integration_code
+
+# copy all loaded and built libraries to a pure basic image
+COPY --from=builder /install /usr/local
+# add default timezone settings
+COPY --from=builder /usr/share/zoneinfo/Etc/UTC /etc/localtime
+RUN echo "Etc/UTC" > /etc/timezone
+
+# bash is installed for more convenient debugging.
+RUN apk --no-cache add bash
+
+# copy payload code only
+COPY main.py ./
+COPY destination_weaviate ./destination_weaviate
+
+ENV AIRBYTE_ENTRYPOINT "python /airbyte/integration_code/main.py"
+ENTRYPOINT ["python", "/airbyte/integration_code/main.py"]
+
+LABEL io.airbyte.version=0.1.0
+LABEL io.airbyte.name=airbyte/destination-weaviate

airbyte-integrations/connectors/destination-weaviate/README.md

@@ -0,0 +1,123 @@
+# Weaviate Destination
+
+This is the repository for the Weaviate destination connector, written in Python.
+For information about how to use this connector within Airbyte, see [the documentation](https://docs.airbyte.io/integrations/destinations/weaviate).
+
+## Local development
+
+### Prerequisites
+**To iterate on this connector, make sure to complete this prerequisites section.**
+
+#### Minimum Python version required `= 3.7.0`
+
+#### Build & Activate Virtual Environment and install dependencies
+From this connector directory, create a virtual environment:
+```
+python -m venv .venv
+```
+
+This will generate a virtualenv for this module in `.venv/`. Make sure this venv is active in your
+development environment of choice. To activate it from the terminal, run:
+```
+source .venv/bin/activate
+pip install -r requirements.txt
+```
+If you are in an IDE, follow your IDE's instructions to activate the virtualenv.
+
+Note that while we are installing dependencies from `requirements.txt`, you should only edit `setup.py` for your dependencies. `requirements.txt` is
+used for editable installs (`pip install -e`) to pull in Python dependencies from the monorepo and will call `setup.py`.
+If this is mumbo jumbo to you, don't worry about it, just put your deps in `setup.py` but install using `pip install -r requirements.txt` and everything
+should work as you expect.
+
+#### Building via Gradle
+From the Airbyte repository root, run:
+```
+./gradlew :airbyte-integrations:connectors:destination-weaviate:build
+```
+
+#### Create credentials
+**If you are a community contributor**, follow the instructions in the [documentation](https://docs.airbyte.io/integrations/destinations/weaviate)
+to generate the necessary credentials. Then create a file `secrets/config.json` conforming to the `destination_weaviate/spec.json` file.
+Note that the `secrets` directory is gitignored by default, so there is no danger of accidentally checking in sensitive information.
+See `integration_tests/sample_config.json` for a sample config file.
+
+**If you are an Airbyte core member**, copy the credentials in Lastpass under the secret name `destination weaviate test creds`
+and place them into `secrets/config.json`.
+
+### Locally running the connector
+```
+python main.py spec
+python main.py check --config secrets/config.json
+python main.py discover --config secrets/config.json
+python main.py read --config secrets/config.json --catalog integration_tests/configured_catalog.json
+```
+
+### Locally running the connector docker image
+
+#### Build
+First, make sure you build the latest Docker image:
+```
+docker build . -t airbyte/destination-weaviate:dev
+```
+
+You can also build the connector image via Gradle:
+```
+./gradlew :airbyte-integrations:connectors:destination-weaviate:airbyteDocker
+```
+When building via Gradle, the docker image name and tag, respectively, are the values of the `io.airbyte.name` and `io.airbyte.version` `LABEL`s in
+the Dockerfile.
+
+#### Run
+Then run any of the connector commands as follows:
+```
+docker run --rm airbyte/destination-weaviate:dev spec
+docker run --rm -v $(pwd)/secrets:/secrets airbyte/destination-weaviate:dev check --config /secrets/config.json
+# messages.jsonl is a file containing line-separated JSON representing AirbyteMessages
+cat messages.jsonl | docker run --rm -v $(pwd)/secrets:/secrets -v $(pwd)/integration_tests:/integration_tests airbyte/destination-weaviate:dev write --config /secrets/config.json --catalog /integration_tests/configured_catalog.json
+```
+## Testing
+   Make sure to familiarize yourself with [pytest test discovery](https://docs.pytest.org/en/latest/goodpractices.html#test-discovery) to know how your test files and methods should be named.
+First install test dependencies into your virtual environment:
+```
+pip install .[tests]
+```
+### Unit Tests
+To run unit tests locally, from the connector directory run:
+```
+python -m pytest unit_tests
+```
+
+### Integration Tests
+There are two types of integration tests: Acceptance Tests (Airbyte's test suite for all destination connectors) and custom integration tests (which are specific to this connector).
+#### Custom Integration tests
+Place custom tests inside `integration_tests/` folder, then, from the connector root, run
+```
+python -m pytest integration_tests
+```
+#### Acceptance Tests
+Coming soon: 
+
+### Using gradle to run tests
+All commands should be run from airbyte project root.
+To run unit tests:
+```
+./gradlew :airbyte-integrations:connectors:destination-weaviate:unitTest
+```
+To run acceptance and custom integration tests:
+```
+./gradlew :airbyte-integrations:connectors:destination-weaviate:integrationTest
+```
+
+## Dependency Management
+All of your dependencies should go in `setup.py`, NOT `requirements.txt`. The requirements file is only used to connect internal Airbyte dependencies in the monorepo for local development.
+We split dependencies between two groups, dependencies that are:
+* required for your connector to work need to go to `MAIN_REQUIREMENTS` list.
+* required for the testing need to go to `TEST_REQUIREMENTS` list
+
+### Publishing a new version of the connector
+You've checked out the repo, implemented a million dollar feature, and you're ready to share your changes with the world. Now what?
+1. Make sure your changes are passing unit and integration tests.
+1. Bump the connector version in `Dockerfile` -- just increment the value of the `LABEL io.airbyte.version` appropriately (we use [SemVer](https://semver.org/)).
+1. Create a Pull Request.
+1. Pat yourself on the back for being an awesome contributor.
+1. Someone from Airbyte will take a look at your PR and iterate with you to merge it into master.

airbyte-integrations/connectors/destination-weaviate/build.gradle

@@ -0,0 +1,8 @@
+plugins {
+    id 'airbyte-python'
+    id 'airbyte-docker'
+}
+
+airbytePython {
+    moduleDirectory 'destination_weaviate'
+}

airbyte-integrations/connectors/destination-weaviate/destination_weaviate/__init__.py

@@ -0,0 +1,8 @@
+#
+# Copyright (c) 2022 Airbyte, Inc., all rights reserved.
+#
+
+
+from .destination import DestinationWeaviate
+
+__all__ = ["DestinationWeaviate"]

airbyte-integrations/connectors/destination-weaviate/destination_weaviate/client.py

@@ -0,0 +1,130 @@
+#
+# Copyright (c) 2022 Airbyte, Inc., all rights reserved.
+#
+
+import json
+import logging
+import time
+import uuid
+from dataclasses import dataclass
+from typing import Any, List, Mapping, MutableMapping
+
+import weaviate
+
+from .utils import generate_id, parse_id_schema, parse_vectors, stream_to_class_name
+
+
+@dataclass
+class BufferedObject:
+    id: str
+    properties: Mapping[str, Any]
+    vector: List[Any]
+    class_name: str
+
+
+class WeaviatePartialBatchError(Exception):
+    pass
+
+
+class Client:
+    def __init__(self, config: Mapping[str, Any], schema: Mapping[str, str]):
+        self.client = self.get_weaviate_client(config)
+        self.config = config
+        self.batch_size = int(config.get("batch_size", 100))
+        self.schema = schema
+        self.vectors = parse_vectors(config.get("vectors"))
+        self.id_schema = parse_id_schema(config.get("id_schema"))
+        self.buffered_objects: MutableMapping[str, BufferedObject] = {}
+        self.objects_with_error: MutableMapping[str, BufferedObject] = {}
+
+    def buffered_write_operation(self, stream_name: str, record: MutableMapping):
+        if self.id_schema.get(stream_name, "") in record:
+            id_field_name = self.id_schema.get(stream_name, "")
+            record_id = generate_id(record.get(id_field_name))
+            del record[id_field_name]
+        else:
+            if "id" in record:
+                record_id = generate_id(record.get("id"))
+                del record["id"]
+            # Weaviate will throw an error if you try to store a field with name _id
+            elif "_id" in record:
+                record_id = generate_id(record.get("_id"))
+                del record["_id"]
+            else:
+                record_id = uuid.uuid4()
+        record_id = str(record_id)
+
+        # TODO support nested objects instead of converting to json string when weaviate supports this
+        for k, v in record.items():
+            if self.schema[stream_name].get(k, "") == "jsonify":
+                record[k] = json.dumps(v)
+            # Handling of empty list that's not part of defined schema otherwise Weaviate throws invalid string property
+            if isinstance(v, list) and len(v) == 0 and k not in self.schema[stream_name]:
+                record[k] = ""
+
+        # Property names in Weaviate have to start with lowercase letter
+        record = {k[0].lower() + k[1:]: v for k, v in record.items()}
+        vector = None
+        if stream_name in self.vectors:
+            vector_column_name = self.vectors.get(stream_name)
+            vector = record.get(vector_column_name)
+            del record[vector_column_name]
+        class_name = stream_to_class_name(stream_name)
+        self.client.batch.add_data_object(record, class_name, record_id, vector=vector)
+        self.buffered_objects[record_id] = BufferedObject(record_id, record, vector, class_name)
+        if self.client.batch.num_objects() >= self.batch_size:
+            self.flush()
+
+    def flush(self, retries: int = 3):
+        if len(self.objects_with_error) > 0 and retries == 0:
+            error_msg = f"Objects had errors and retries failed as well. Object IDs: {self.objects_with_error.keys}"
+            raise WeaviatePartialBatchError(error_msg)
+
+        results = self.client.batch.create_objects()
+        self.objects_with_error.clear()
+        for result in results:
+            errors = result.get("result", {}).get("errors", [])
+            if errors:
+                obj_id = result.get("id")
+                self.objects_with_error[obj_id] = self.buffered_objects.get(obj_id)
+                logging.info(f"Object {obj_id} had errors: {errors}. Going to retry.")
+
+        for buffered_object in self.objects_with_error.values():
+            self.client.batch.add_data_object(
+                buffered_object.properties, buffered_object.class_name, buffered_object.id, buffered_object.vector
+            )
+
+        if len(self.objects_with_error) > 0 and retries > 0:
+            logging.info("sleeping 2 seconds before retrying batch again")
+            time.sleep(2)
+            self.flush(retries - 1)
+
+        self.buffered_objects.clear()
+
+    def delete_stream_entries(self, stream_name: str):
+        class_name = stream_to_class_name(stream_name)
+        try:
+            original_schema = self.client.schema.get(class_name=class_name)
+            self.client.schema.delete_class(class_name=class_name)
+            logging.info(f"Deleted class {class_name}")
+            self.client.schema.create_class(original_schema)
+            logging.info(f"Recreated class {class_name}")
+        except weaviate.exceptions.UnexpectedStatusCodeException as e:
+            if e.message.startswith("Get schema! Unexpected status code: 404"):
+                logging.info(f"Class {class_name} did not exist.")
+            else:
+                raise e
+
+    @staticmethod
+    def get_weaviate_client(config: Mapping[str, Any]) -> weaviate.Client:
+        url, username, password = config.get("url"), config.get("username"), config.get("password")
+
+        if username and not password:
+            raise Exception("Password is required when username is set")
+        if password and not username:
+            raise Exception("Username is required when password is set")
+
+        if username and password:
+            credentials = weaviate.auth.AuthClientPassword(username, password)
+            return weaviate.Client(url=url, auth_client_secret=credentials)
+        return weaviate.Client(url=url, timeout_config=(2, 2))