autogenerated code (#802)

comet-ml · Dec 3, 2024 · 1430133 · 1430133
1 parent 2b59659
commit 1430133
Show file tree

Hide file tree

Showing 37 changed files with 668 additions and 45 deletions.
diff --git a/...ackend/src/test/java/com/comet/opik/api/resources/v1/priv/AuthenticationResourceTest.java b/...ackend/src/test/java/com/comet/opik/api/resources/v1/priv/AuthenticationResourceTest.java
@@ -173,16 +173,14 @@ private Stream<Arguments> useInvalidWorkspace__thenReturnForbiddenResponse() {
             return Stream.of(
                     arguments(""),
                     arguments(UNAUTHORISED_WORKSPACE_NAME),
-                    arguments(DEFAULT_WORKSPACE_NAME)
-            );
+                    arguments(DEFAULT_WORKSPACE_NAME));
         }
 
         private void checkProjectAccess(String apiKey,
-                                        String workspaceName,
-                                        int expectedStatus) {
+                String workspaceName,
+                int expectedStatus) {
             var request = AuthDetailsHolder.builder().build();
 
-
             try (var actualResponse = client.target(URL_TEMPLATE.formatted(baseURI))
                     .request()
                     .accept(MediaType.APPLICATION_JSON_TYPE)

diff --git a/.../opik-backend/src/test/java/com/comet/opik/infrastructure/auth/RemoveAuthServiceTest.java b/.../opik-backend/src/test/java/com/comet/opik/infrastructure/auth/RemoveAuthServiceTest.java
@@ -9,7 +9,6 @@
 import jakarta.ws.rs.InternalServerErrorException;
 import jakarta.ws.rs.client.Client;
 import jakarta.ws.rs.core.HttpHeaders;
-import jakarta.ws.rs.core.Response;
 import org.apache.commons.lang3.RandomStringUtils;
 import org.apache.hc.core5.http.HttpStatus;
 import org.junit.jupiter.api.AfterAll;

diff --git a/apps/opik-documentation/documentation/rest_api/opik.yaml b/apps/opik-documentation/documentation/rest_api/opik.yaml
@@ -37,6 +37,8 @@ servers:
 tags:
 - name: System usage
   description: System usage related resource
+- name: Check
+  description: Access check resources
 - name: Datasets
   description: Dataset resources
 - name: Experiments
@@ -108,6 +110,33 @@ paths:
             application/json:
               schema:
                 $ref: '#/components/schemas/TraceCountResponse'
+  /v1/private/auth:
+    post:
+      tags:
+      - Check
+      summary: Check user access to workspace
+      description: Check user access to workspace
+      operationId: checkAccess
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/AuthDetailsHolder'
+      responses:
+        "204":
+          description: No Content
+        "401":
+          description: Bad Request
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorMessage'
+        "403":
+          description: Access forbidden
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorMessage'
   /v1/private/datasets:
     get:
       tags:
@@ -1932,6 +1961,18 @@ components:
         trace_count:
           type: integer
           format: int32
+    ErrorMessage:
+      type: object
+      properties:
+        code:
+          type: integer
+          format: int32
+        message:
+          type: string
+        details:
+          type: string
+    AuthDetailsHolder:
+      type: object
     Dataset:
       required:
       - name
@@ -2578,16 +2619,6 @@ components:
           type: array
           items:
             $ref: '#/components/schemas/Column_Public'
-    ErrorMessage:
-      type: object
-      properties:
-        code:
-          type: integer
-          format: int32
-        message:
-          type: string
-        details:
-          type: string
     ChunkedOutputJsonNode:
       type: object
       properties:

diff --git a/sdks/code_generation/fern/openapi/openapi.yaml b/sdks/code_generation/fern/openapi/openapi.yaml
@@ -37,6 +37,8 @@ servers:
 tags:
 - name: System usage
   description: System usage related resource
+- name: Check
+  description: Access check resources
 - name: Datasets
   description: Dataset resources
 - name: Experiments
@@ -108,6 +110,33 @@ paths:
             application/json:
               schema:
                 $ref: '#/components/schemas/TraceCountResponse'
+  /v1/private/auth:
+    post:
+      tags:
+      - Check
+      summary: Check user access to workspace
+      description: Check user access to workspace
+      operationId: checkAccess
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/AuthDetailsHolder'
+      responses:
+        "204":
+          description: No Content
+        "401":
+          description: Bad Request
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorMessage'
+        "403":
+          description: Access forbidden
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorMessage'
   /v1/private/datasets:
     get:
       tags:
@@ -1932,6 +1961,18 @@ components:
         trace_count:
           type: integer
           format: int32
+    ErrorMessage:
+      type: object
+      properties:
+        code:
+          type: integer
+          format: int32
+        message:
+          type: string
+        details:
+          type: string
+    AuthDetailsHolder:
+      type: object
     Dataset:
       required:
       - name
@@ -2578,16 +2619,6 @@ components:
           type: array
           items:
             $ref: '#/components/schemas/Column_Public'
-    ErrorMessage:
-      type: object
-      properties:
-        code:
-          type: integer
-          format: int32
-        message:
-          type: string
-        details:
-          type: string
     ChunkedOutputJsonNode:
       type: object
       properties:

diff --git a/sdks/python/src/opik/rest_api/__init__.py b/sdks/python/src/opik/rest_api/__init__.py
@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AuthDetailsHolder,
     AvgValueStatPublic,
     BatchDelete,
     BiInformation,
@@ -128,11 +129,14 @@
 from .errors import (
     BadRequestError,
     ConflictError,
+    ForbiddenError,
     NotFoundError,
     NotImplementedError,
+    UnauthorizedError,
     UnprocessableEntityError,
 )
 from . import (
+    check,
     datasets,
     experiments,
     feedback_definitions,
@@ -157,6 +161,7 @@
 
 __all__ = [
     "AsyncOpikApi",
+    "AuthDetailsHolder",
     "AvgValueStatPublic",
     "BadRequestError",
     "BatchDelete",
@@ -232,6 +237,7 @@
     "Feedback_Numerical",
     "FindFeedbackDefinitionsRequestType",
     "FindFeedbackScoreNames1RequestType",
+    "ForbiddenError",
     "GetSpanStatsRequestType",
     "GetSpansByProjectRequestType",
     "JsonNode",
@@ -291,8 +297,10 @@
     "TracePagePublic",
     "TracePublic",
     "TraceWrite",
+    "UnauthorizedError",
     "UnprocessableEntityError",
     "WorkspaceTraceCount",
+    "check",
     "datasets",
     "experiments",
     "feedback_definitions",

diff --git a/sdks/python/src/opik/rest_api/check/__init__.py b/sdks/python/src/opik/rest_api/check/__init__.py
@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		# This file was auto-generated by Fern from our API Definition.