Add skeletal metadata for Upstage AI: without implementation due to s…

…pec ambiguity (#1072)
stargate · May 8, 2024 · ff196c9 · ff196c9
1 parent 1622271
commit ff196c9
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 4 deletions.
diff --git a/...main/java/io/stargate/sgv2/jsonapi/service/embedding/configuration/ProviderConstants.java b/...main/java/io/stargate/sgv2/jsonapi/service/embedding/configuration/ProviderConstants.java
@@ -7,6 +7,7 @@ public final class ProviderConstants {
   public static final String VERTEXAI = "vertexai";
   public static final String COHERE = "cohere";
   public static final String NVIDIA = "nvidia";
+  public static final String UPSTAGE_AI = "upstageAI";
   public static final String VOYAGE_AI = "voyageAI";
   public static final String CUSTOM = "custom";
   public static final String MISTRAL = "mistral";

diff --git a/src/main/resources/embedding-providers-config.yaml b/src/main/resources/embedding-providers-config.yaml
@@ -287,8 +287,34 @@ stargate:
           models:
             - name: mistral-embed
               vector-dimension: 1024
-        # upstage:
-        # see https://developers.upstage.ai/docs/apis/embeddings
-        # NOTE: they have a model for storing and a diff one for reading, this is different to everyone else
-        # holding on the config / implementation until we confirm if this can change
+
+        # NOTE: UpstageAI have a model for storing and a diff one for reading: this is different
+        #  from everyone else.
+        #  Holding on the config / implementation until we confirm if this can change
+        # NOTE: no free tier available, CC required
+        upstageAI:
+          # see https://developers.upstage.ai/docs/apis/embeddings
+          # NOT enabled until
+          enabled: false
+          url: https://api.upstage.ai/v1/solar/embeddings
+          supported-authentications:
+            NONE:
+              enabled: false
+            HEADER:
+              enabled: true
+              tokens:
+                - accepted: x-embedding-api-key
+                  forwarded: Authorization
+            SHARED_SECRET:
+              enabled: false
+              tokens:
+                - accepted: providerKey
+                  forwarded: Authorization
+          parameters:
+          properties:
+          models: # NOTE: this is where weirdness exists; including model names as example
+            - name: solar-1-mini-embedding-query
+              vector-dimension: 4096
+            - name: solar-1-mini-embedding-passage
+              vector-dimension: 4096