@superblocksteam/types 0.1859.0

package/src/ai/v1/metadata_pb.ts

@@ -0,0 +1,134 @@
+// @generated by protoc-gen-es v1.2.0 with parameter "target=ts"
+// @generated from file ai/v1/metadata.proto (package ai.v1, syntax proto3)
+/* eslint-disable */
+// @ts-nocheck
+
+import type { BinaryReadOptions, FieldList, JsonReadOptions, JsonValue, PartialMessage, PlainMessage } from "@bufbuild/protobuf";
+import { Message, proto3 } from "@bufbuild/protobuf";
+import { SQLMetadata_Minified } from "../../plugins/common/v1/metadata_pb";
+import { Metadata_Minified } from "../../plugins/kafka/v1/plugin_pb";
+
+/**
+ * This represents the metadata that the "AI stack" cares about. It is
+ * not guaranteed to be the same as the metadata for the "plugin stack".
+ * This is more than likely a minified representation of it.
+ *
+ * @generated from message ai.v1.Metadata
+ */
+export class Metadata extends Message<Metadata> {
+  /**
+   * @generated from oneof ai.v1.Metadata.config
+   */
+  config: {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified mariadb = 1;
+     */
+    value: SQLMetadata_Minified;
+    case: "mariadb";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified mssql = 2;
+     */
+    value: SQLMetadata_Minified;
+    case: "mssql";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified mysql = 3;
+     */
+    value: SQLMetadata_Minified;
+    case: "mysql";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified postgres = 4;
+     */
+    value: SQLMetadata_Minified;
+    case: "postgres";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified rockset = 5;
+     */
+    value: SQLMetadata_Minified;
+    case: "rockset";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified snowflake = 6;
+     */
+    value: SQLMetadata_Minified;
+    case: "snowflake";
+  } | {
+    /**
+     * @generated from field: plugins.common.v1.SQLMetadata.Minified cockroachdb = 7;
+     */
+    value: SQLMetadata_Minified;
+    case: "cockroachdb";
+  } | {
+    /**
+     * @generated from field: plugins.kafka.v1.Metadata.Minified kafka = 8;
+     */
+    value: Metadata_Minified;
+    case: "kafka";
+  } | {
+    /**
+     * @generated from field: plugins.kafka.v1.Metadata.Minified confluent = 9;
+     */
+    value: Metadata_Minified;
+    case: "confluent";
+  } | {
+    /**
+     * @generated from field: plugins.kafka.v1.Metadata.Minified msk = 10;
+     */
+    value: Metadata_Minified;
+    case: "msk";
+  } | {
+    /**
+     * @generated from field: plugins.kafka.v1.Metadata.Minified redpanda = 11;
+     */
+    value: Metadata_Minified;
+    case: "redpanda";
+  } | {
+    /**
+     * @generated from field: plugins.kafka.v1.Metadata.Minified aivenkafka = 12;
+     */
+    value: Metadata_Minified;
+    case: "aivenkafka";
+  } | { case: undefined; value?: undefined } = { case: undefined };
+
+  constructor(data?: PartialMessage<Metadata>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "ai.v1.Metadata";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "mariadb", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 2, name: "mssql", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 3, name: "mysql", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 4, name: "postgres", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 5, name: "rockset", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 6, name: "snowflake", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 7, name: "cockroachdb", kind: "message", T: SQLMetadata_Minified, oneof: "config" },
+    { no: 8, name: "kafka", kind: "message", T: Metadata_Minified, oneof: "config" },
+    { no: 9, name: "confluent", kind: "message", T: Metadata_Minified, oneof: "config" },
+    { no: 10, name: "msk", kind: "message", T: Metadata_Minified, oneof: "config" },
+    { no: 11, name: "redpanda", kind: "message", T: Metadata_Minified, oneof: "config" },
+    { no: 12, name: "aivenkafka", kind: "message", T: Metadata_Minified, oneof: "config" },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): Metadata {
+    return new Metadata().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): Metadata {
+    return new Metadata().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): Metadata {
+    return new Metadata().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: Metadata | PlainMessage<Metadata> | undefined, b: Metadata | PlainMessage<Metadata> | undefined): boolean {
+    return proto3.util.equals(Metadata, a, b);
+  }
+}
+

package/src/ai/v1/service_pb.ts

@@ -0,0 +1,154 @@
+// @generated by protoc-gen-es v1.2.0 with parameter "target=ts"
+// @generated from file ai/v1/service.proto (package ai.v1, syntax proto3)
+/* eslint-disable */
+// @ts-nocheck
+
+import type { BinaryReadOptions, FieldList, JsonReadOptions, JsonValue, PartialMessage, PlainMessage } from "@bufbuild/protobuf";
+import { Message, proto3 } from "@bufbuild/protobuf";
+import { LLM, MODEL, Task } from "./ai_pb";
+
+/**
+ * @generated from message ai.v1.Overrides
+ */
+export class Overrides extends Message<Overrides> {
+  /**
+   * @generated from field: ai.v1.LLM llm = 1;
+   */
+  llm = LLM.LLM_UNSPECIFIED;
+
+  /**
+   * @generated from field: ai.v1.MODEL model = 2;
+   */
+  model = MODEL.MODEL_UNSPECIFIED;
+
+  constructor(data?: PartialMessage<Overrides>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "ai.v1.Overrides";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "llm", kind: "enum", T: proto3.getEnumType(LLM) },
+    { no: 2, name: "model", kind: "enum", T: proto3.getEnumType(MODEL) },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): Overrides {
+    return new Overrides().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): Overrides {
+    return new Overrides().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): Overrides {
+    return new Overrides().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: Overrides | PlainMessage<Overrides> | undefined, b: Overrides | PlainMessage<Overrides> | undefined): boolean {
+    return proto3.util.equals(Overrides, a, b);
+  }
+}
+
+/**
+ * @generated from message ai.v1.CreateTaskRequest
+ */
+export class CreateTaskRequest extends Message<CreateTaskRequest> {
+  /**
+   * @generated from field: ai.v1.Task task = 1;
+   */
+  task?: Task;
+
+  /**
+   * @generated from field: ai.v1.Overrides overrides = 2;
+   */
+  overrides?: Overrides;
+
+  constructor(data?: PartialMessage<CreateTaskRequest>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "ai.v1.CreateTaskRequest";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "task", kind: "message", T: Task },
+    { no: 2, name: "overrides", kind: "message", T: Overrides },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): CreateTaskRequest {
+    return new CreateTaskRequest().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): CreateTaskRequest {
+    return new CreateTaskRequest().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): CreateTaskRequest {
+    return new CreateTaskRequest().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: CreateTaskRequest | PlainMessage<CreateTaskRequest> | undefined, b: CreateTaskRequest | PlainMessage<CreateTaskRequest> | undefined): boolean {
+    return proto3.util.equals(CreateTaskRequest, a, b);
+  }
+}
+
+/**
+ * NOTE(frank): I'm considering having two event types. I'm not sure how I feel about sending
+ * inforamation that doesn't need to be sent every message in every message (i.e. id). Rather,
+ * we could send a "hello" event at the beginning with any metadata.
+ *
+ * @generated from message ai.v1.TaskEvent
+ */
+export class TaskEvent extends Message<TaskEvent> {
+  /**
+   * @generated from field: string chunk = 1;
+   */
+  chunk = "";
+
+  /**
+   * @generated from field: ai.v1.LLM llm = 2;
+   */
+  llm = LLM.LLM_UNSPECIFIED;
+
+  /**
+   * @generated from field: ai.v1.MODEL model = 3;
+   */
+  model = MODEL.MODEL_UNSPECIFIED;
+
+  /**
+   * @generated from field: string id = 4;
+   */
+  id = "";
+
+  constructor(data?: PartialMessage<TaskEvent>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "ai.v1.TaskEvent";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "chunk", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 2, name: "llm", kind: "enum", T: proto3.getEnumType(LLM) },
+    { no: 3, name: "model", kind: "enum", T: proto3.getEnumType(MODEL) },
+    { no: 4, name: "id", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): TaskEvent {
+    return new TaskEvent().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): TaskEvent {
+    return new TaskEvent().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): TaskEvent {
+    return new TaskEvent().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: TaskEvent | PlainMessage<TaskEvent> | undefined, b: TaskEvent | PlainMessage<TaskEvent> | undefined): boolean {
+    return proto3.util.equals(TaskEvent, a, b);
+  }
+}
+

package/src/ai/v1/vector_pb.ts

@@ -0,0 +1,63 @@
+// @generated by protoc-gen-es v1.2.0 with parameter "target=ts"
+// @generated from file ai/v1/vector.proto (package ai.v1, syntax proto3)
+/* eslint-disable */
+// @ts-nocheck
+
+import type { BinaryReadOptions, FieldList, JsonReadOptions, JsonValue, PartialMessage, PlainMessage } from "@bufbuild/protobuf";
+import { Message, proto3, Struct } from "@bufbuild/protobuf";
+
+/**
+ * @generated from message ai.v1.Vector
+ */
+export class Vector extends Message<Vector> {
+  /**
+   * @generated from field: string id = 1;
+   */
+  id = "";
+
+  /**
+   * @generated from field: repeated float values = 2;
+   */
+  values: number[] = [];
+
+  /**
+   * @generated from field: google.protobuf.Struct metadata = 3;
+   */
+  metadata?: Struct;
+
+  /**
+   * @generated from field: optional float score = 4;
+   */
+  score?: number;
+
+  constructor(data?: PartialMessage<Vector>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "ai.v1.Vector";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "id", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 2, name: "values", kind: "scalar", T: 2 /* ScalarType.FLOAT */, repeated: true },
+    { no: 3, name: "metadata", kind: "message", T: Struct },
+    { no: 4, name: "score", kind: "scalar", T: 2 /* ScalarType.FLOAT */, opt: true },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): Vector {
+    return new Vector().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): Vector {
+    return new Vector().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): Vector {
+    return new Vector().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: Vector | PlainMessage<Vector> | undefined, b: Vector | PlainMessage<Vector> | undefined): boolean {
+    return proto3.util.equals(Vector, a, b);
+  }
+}
+