Feature/datasets

composer.json

@@ -54,7 +54,7 @@
         "test:lint": "@php vendor/bin/pint --config https://raw.githubusercontent.com/DIJ-digital/pint-config/main/pint.json",
         "test:unit": "pest",
         "test:types": "phpstan",
-        "test:refactor": "rector --dry-run",
+        "test:refactor": "rector",
         "test": [
             "@test:lint",
             "@test:type-coverage",

src/Contracts/TransporterInterface.php

@@ -9,34 +9,34 @@
 interface TransporterInterface
 {
     /**
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $options
      */
     public function request(string $method, string $uri, array $options = []): ResponseInterface;
 
     /**
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $options
      */
     public function get(string $uri, array $options = []): ResponseInterface;
 
     /**
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $options
      */
     public function post(string $uri, array $options = []): ResponseInterface;
 
     /**
-     * @param array<string, mixed> $data
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $data
+     * @param  array<string, mixed>  $options
      */
     public function postJson(string $uri, array $data = [], array $options = []): ResponseInterface;
 
     /**
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $options
      */
     public function delete(string $uri, array $options = []): ResponseInterface;
 
     /**
-     * @param array<string, mixed> $data
-     * @param array<string, mixed> $options
+     * @param  array<string, mixed>  $data
+     * @param  array<string, mixed>  $options
      */
     public function patchJson(string $uri, array $data = [], array $options = []): ResponseInterface;
 }

src/Dataset.php

@@ -0,0 +1,100 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP;
+
+use DIJ\Langfuse\PHP\Contracts\TransporterInterface;
+use DIJ\Langfuse\PHP\Responses\DatasetListResponse;
+use DIJ\Langfuse\PHP\Responses\DatasetResponse;
+use DIJ\Langfuse\PHP\Responses\DatasetRunListResponse;
+use JsonException;
+
+class Dataset
+{
+    public function __construct(private readonly TransporterInterface $transporter) {}
+
+    /**
+     * @param  array<string, mixed>|null  $metadata
+     * @param  array<string, mixed>|null  $inputSchema
+     * @param  array<string, mixed>|null  $expectedOutputSchema
+     *
+     * @throws JsonException
+     */
+    public function create(
+        string $name,
+        ?string $description = null,
+        ?array $metadata = null,
+        ?array $inputSchema = null,
+        ?array $expectedOutputSchema = null,
+    ): DatasetResponse {
+        $response = $this->transporter->postJson(
+            '/api/public/v2/datasets',
+            array_filter([
+                'name' => $name,
+                'description' => $description,
+                'metadata' => $metadata,
+                'inputSchema' => $inputSchema,
+                'expectedOutputSchema' => $expectedOutputSchema,
+            ], fn ($value) => $value !== null)
+        );
+
+        /** @var array{id: string, name: string, projectId: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null, inputSchema?: array<string, mixed>|null, expectedOutputSchema?: array<string, mixed>|null} $data */
+        $data = json_decode($response->getBody()->getContents(), true, flags: JSON_THROW_ON_ERROR);
+
+        return DatasetResponse::fromArray($data);
+    }
+
+    /**
+     * @throws JsonException
+     */
+    public function get(string $datasetName): DatasetResponse
+    {
+        $response = $this->transporter->get(
+            sprintf('/api/public/v2/datasets/%s', urlencode($datasetName))
+        );
+
+        /** @var array{id: string, name: string, projectId: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null, inputSchema?: array<string, mixed>|null, expectedOutputSchema?: array<string, mixed>|null} $data */
+        $data = json_decode($response->getBody()->getContents(), true, flags: JSON_THROW_ON_ERROR);
+
+        return DatasetResponse::fromArray($data);
+    }
+
+    /**
+     * @throws JsonException
+     */
+    public function list(?int $page = null, ?int $limit = null): DatasetListResponse
+    {
+        $response = $this->transporter->get(
+            '/api/public/v2/datasets',
+            ['query' => array_filter([
+                'page' => $page,
+                'limit' => $limit,
+            ], fn ($value) => $value !== null)]
+        );
+
+        /** @var array{data: array<int, array{id: string, name: string, projectId: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null, inputSchema?: array<string, mixed>|null, expectedOutputSchema?: array<string, mixed>|null}>, meta: array{page: int, limit: int, totalPages: int, totalItems: int}} $data */
+        $data = json_decode($response->getBody()->getContents(), true, flags: JSON_THROW_ON_ERROR);
+
+        return DatasetListResponse::fromArray($data);
+    }
+
+    /**
+     * @throws JsonException
+     */
+    public function getRuns(string $datasetName, ?int $page = null, ?int $limit = null): DatasetRunListResponse
+    {
+        $response = $this->transporter->get(
+            sprintf('/api/public/datasets/%s/runs', urlencode($datasetName)),
+            ['query' => array_filter([
+                'page' => $page,
+                'limit' => $limit,
+            ], fn ($value) => $value !== null)]
+        );
+
+        /** @var array{data: array<int, array{id: string, name: string, datasetId: string, datasetName: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null}>, meta: array{page: int, limit: int, totalPages: int, totalItems: int}} $data */
+        $data = json_decode($response->getBody()->getContents(), true, flags: JSON_THROW_ON_ERROR);
+
+        return DatasetRunListResponse::fromArray($data);
+    }
+}

src/Enums/DatasetStatus.php

@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP\Enums;
+
+enum DatasetStatus: string
+{
+    case ACTIVE = 'ACTIVE';
+    case ARCHIVED = 'ARCHIVED';
+}

src/Langfuse.php

@@ -12,8 +12,7 @@ public function __construct(
         private readonly TransporterInterface $transporter,
         private readonly string $environment = 'default',
         private readonly string $label = 'latest',
-    ) {
-    }
+    ) {}
 
     public function prompt(): Prompt
     {
@@ -37,4 +36,11 @@ public function score(): Score
             transporter: $this->transporter,
         );
     }
+
+    public function dataset(): Dataset
+    {
+        return new Dataset(
+            transporter: $this->transporter,
+        );
+    }
 }

src/Responses/DatasetListResponse.php

@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP\Responses;
+
+use DIJ\Langfuse\PHP\ValueObjects\MetaData;
+
+readonly class DatasetListResponse
+{
+    /**
+     * @param  array<int, DatasetResponse>  $data
+     */
+    public function __construct(
+        public array $data,
+        public MetaData $meta,
+    ) {}
+
+    /**
+     * @param array{
+     *     data: array<int, array{id: string, name: string, projectId: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null, inputSchema?: array<string, mixed>|null, expectedOutputSchema?: array<string, mixed>|null}>,
+     *     meta: array{page: int, limit: int, totalPages: int, totalItems: int}
+     * } $data
+     */
+    public static function fromArray(array $data): self
+    {
+        return new self(
+            data: array_map(
+                fn (array $item): DatasetResponse => DatasetResponse::fromArray($item),
+                $data['data']
+            ),
+            meta: MetaData::fromArray($data['meta']),
+        );
+    }
+}

src/Responses/DatasetResponse.php

@@ -0,0 +1,53 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP\Responses;
+
+readonly class DatasetResponse
+{
+    /**
+     * @param  array<string, mixed>|null  $metadata
+     * @param  array<string, mixed>|null  $inputSchema
+     * @param  array<string, mixed>|null  $expectedOutputSchema
+     */
+    public function __construct(
+        public string $id,
+        public string $name,
+        public string $projectId,
+        public string $createdAt,
+        public string $updatedAt,
+        public ?string $description = null,
+        public ?array $metadata = null,
+        public ?array $inputSchema = null,
+        public ?array $expectedOutputSchema = null,
+    ) {}
+
+    /**
+     * @param array{
+     *     id: string,
+     *     name: string,
+     *     projectId: string,
+     *     createdAt: string,
+     *     updatedAt: string,
+     *     description?: string|null,
+     *     metadata?: array<string, mixed>|null,
+     *     inputSchema?: array<string, mixed>|null,
+     *     expectedOutputSchema?: array<string, mixed>|null
+     * } $data
+     */
+    public static function fromArray(array $data): self
+    {
+        return new self(
+            id: $data['id'],
+            name: $data['name'],
+            projectId: $data['projectId'],
+            createdAt: $data['createdAt'],
+            updatedAt: $data['updatedAt'],
+            description: $data['description'] ?? null,
+            metadata: $data['metadata'] ?? null,
+            inputSchema: $data['inputSchema'] ?? null,
+            expectedOutputSchema: $data['expectedOutputSchema'] ?? null,
+        );
+    }
+}

src/Responses/DatasetRunListResponse.php

@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP\Responses;
+
+use DIJ\Langfuse\PHP\ValueObjects\MetaData;
+
+readonly class DatasetRunListResponse
+{
+    /**
+     * @param  array<int, DatasetRunResponse>  $data
+     */
+    public function __construct(
+        public array $data,
+        public MetaData $meta,
+    ) {}
+
+    /**
+     * @param array{
+     *     data: array<int, array{id: string, name: string, datasetId: string, datasetName: string, createdAt: string, updatedAt: string, description?: string|null, metadata?: array<string, mixed>|null}>,
+     *     meta: array{page: int, limit: int, totalPages: int, totalItems: int}
+     * } $data
+     */
+    public static function fromArray(array $data): self
+    {
+        return new self(
+            data: array_map(
+                fn (array $item): DatasetRunResponse => DatasetRunResponse::fromArray($item),
+                $data['data']
+            ),
+            meta: MetaData::fromArray($data['meta']),
+        );
+    }
+}

src/Responses/DatasetRunResponse.php

@@ -0,0 +1,48 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DIJ\Langfuse\PHP\Responses;
+
+readonly class DatasetRunResponse
+{
+    /**
+     * @param  array<string, mixed>|null  $metadata
+     */
+    public function __construct(
+        public string $id,
+        public string $name,
+        public string $datasetId,
+        public string $datasetName,
+        public string $createdAt,
+        public string $updatedAt,
+        public ?string $description = null,
+        public ?array $metadata = null,
+    ) {}
+
+    /**
+     * @param array{
+     *     id: string,
+     *     name: string,
+     *     datasetId: string,
+     *     datasetName: string,
+     *     createdAt: string,
+     *     updatedAt: string,
+     *     description?: string|null,
+     *     metadata?: array<string, mixed>|null
+     * } $data
+     */
+    public static function fromArray(array $data): self
+    {
+        return new self(
+            id: $data['id'],
+            name: $data['name'],
+            datasetId: $data['datasetId'],
+            datasetName: $data['datasetName'],
+            createdAt: $data['createdAt'],
+            updatedAt: $data['updatedAt'],
+            description: $data['description'] ?? null,
+            metadata: $data['metadata'] ?? null,
+        );
+    }
+}