feat(specs): add transformation copilot to ingestion (#3479)

algolia · Aug 6, 2024 · ac0cadb · ac0cadb
1 parent 1034d8e
commit ac0cadb
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 1 deletion.
diff --git a/scripts/cts/runCts.ts b/scripts/cts/runCts.ts
@@ -123,7 +123,7 @@ async function runCtsOne(language: Language, suites: Record<CTSType, boolean>):
       break;
     case 'swift':
       await run(
-        `swift test -Xswiftc -suppress-warnings --parallel ${filter((f) => `--filter ${f}.*`)}`,
+        `swift test -Xswiftc -suppress-warnings --parallel ${filter((f) => `--filter "${f}.*"`)}`,
         {
           cwd,
           language,

diff --git a/specs/ingestion/common/schemas/transformation.yml b/specs/ingestion/common/schemas/transformation.yml
@@ -119,3 +119,24 @@ TransformationTryResponse:
           type: string
   required:
     - payloads
+
+Model:
+  type: object
+  additionalProperties: false
+  properties:
+    fullname:
+      type: string
+    modelName:
+      type: string
+    systemPrompt:
+      type: string
+    id:
+      type: string
+    provider:
+      type: string
+  required:
+    - fullname
+    - modelName
+    - systemPrompt
+    - id
+    - provider
diff --git a/specs/ingestion/paths/transformations/transformationsModels.yml b/specs/ingestion/paths/transformations/transformationsModels.yml
@@ -0,0 +1,29 @@
+get:
+  tags:
+    - transformations
+  summary: Retrieve existing LLM transformation helpers
+  description: Retrieves a list of existing LLM transformation helpers.
+  operationId: listTransformationModels
+  x-acl:
+    - addObject
+    - deleteIndex
+    - editSettings
+  responses:
+    '200':
+      description: OK
+      content:
+        application/json:
+          schema:
+            title: transformationModels
+            type: object
+            description: List of available AI models for transformation purposes.
+            additionalProperties: false
+            properties:
+              llms:
+                type: array
+                items:
+                  $ref: '../../common/schemas/transformation.yml#/Model'
+            required:
+              - llms
+    '400':
+      $ref: '../../../common/responses/BadRequest.yml'
diff --git a/specs/ingestion/spec.yml b/specs/ingestion/spec.yml
@@ -175,6 +175,8 @@ paths:
     $ref: 'paths/transformations/transformationsTry.yml'
   /1/transformations/search:
     $ref: 'paths/transformations/transformationsSearch.yml'
+  /1/transformations/copilot:
+    $ref: 'paths/transformations/transformationsModels.yml'
   /1/transformations/{transformationID}:
     $ref: 'paths/transformations/transformationID.yml'
 

diff --git a/tests/CTS/requests/ingestion/listTransformationModels.json b/tests/CTS/requests/ingestion/listTransformationModels.json
@@ -0,0 +1,9 @@
+[
+  {
+    "parameters": {},
+    "request": {
+      "path": "/1/transformations/copilot",
+      "method": "GET"
+    }
+  }
+]