fix: bump ai-resource dependencies

BREAKING CHANGE: requires gravitee-resource-ai-model-api@2.1

Author: remisultan

pom.xml

@@ -35,7 +35,11 @@
     <properties>
         <!-- Gravitee dependencies version -->
         <gravitee-apim.version>4.8.1</gravitee-apim.version>
-        <gravitee-resource-ai-model-api.version>1.0.0</gravitee-resource-ai-model-api.version>
+        <gravitee-inference-service.version>1.2.0</gravitee-inference-service.version>
+        <gravitee-resource-ai-model-api.version>2.1.0</gravitee-resource-ai-model-api.version>
+
+        <!-- Tests -->
+        <gravitee-resource-ai-model-text-classification.version>2.0.0-bump-ai-model-resource-api-SNAPSHOT</gravitee-resource-ai-model-text-classification.version>
 
         <!-- Maven plugins -->
         <maven-plugin-assembly.version>3.7.1</maven-plugin-assembly.version>
@@ -106,6 +110,18 @@
             <artifactId>assertj-core</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>io.gravitee.resource</groupId>
+            <artifactId>gravitee-resource-ai-model-text-classification</artifactId>
+            <version>${gravitee-resource-ai-model-text-classification.version}</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>io.gravitee.inference.service</groupId>
+            <artifactId>gravitee-inference-service</artifactId>
+            <version>${gravitee-inference-service.version}</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
     <build>

src/main/java/io/gravitee/policy/ai/prompt/guard/rails/AiPromptGuardRailsPolicy.java

@@ -22,9 +22,8 @@
 import io.gravitee.policy.ai.prompt.guard.rails.configuration.RequestPolicy;
 import io.gravitee.policy.ai.prompt.guard.rails.model.AiModelResourceProvider;
 import io.gravitee.policy.api.annotations.RequireResource;
-import io.gravitee.reporter.api.v4.metric.Metrics;
-import io.gravitee.resource.ai_model.api.ClassifierResults;
 import io.gravitee.resource.ai_model.api.model.PromptInput;
+import io.gravitee.resource.ai_model.api.result.ClassifierResults;
 import io.reactivex.rxjava3.core.Completable;
 import io.reactivex.rxjava3.core.CompletableSource;
 import java.util.Set;

src/main/java/io/gravitee/policy/ai/prompt/guard/rails/model/AiModelResourceProvider.java

@@ -17,7 +17,8 @@
 
 import io.gravitee.gateway.reactive.api.context.http.HttpPlainExecutionContext;
 import io.gravitee.policy.ai.prompt.guard.rails.configuration.AiPromptGuardRailsConfiguration;
-import io.gravitee.resource.ai_model.api.AiTextClassificationModelResource;
+import io.gravitee.resource.ai_model.api.AiTextModelResource;
+import io.gravitee.resource.ai_model.api.result.ClassifierResults;
 import io.gravitee.resource.api.ResourceManager;
 
 public class AiModelResourceProvider {
@@ -28,7 +29,7 @@ public AiModelResourceProvider(AiPromptGuardRailsConfiguration configuration) {
         this.configuration = configuration;
     }
 
-    public AiTextClassificationModelResource<?> get(HttpPlainExecutionContext ctx) {
-        return ctx.getComponent(ResourceManager.class).getResource(configuration.resourceName(), AiTextClassificationModelResource.class);
+    public AiTextModelResource<?, ?, ClassifierResults> get(HttpPlainExecutionContext ctx) {
+        return ctx.getComponent(ResourceManager.class).getResource(configuration.resourceName(), AiTextModelResource.class);
     }
 }