feat(google-aistudio): set task type to RETRIEVAL_QUERY or RETRIEVAL_DOCUMENT based on context.forQueries()

Author: BlueTree242

google-aistudio/src/main/java/dev/bluetree242/saaiaddons/aistudio/GoogleAiStudioEmbeddingProvider.java

@@ -27,9 +27,15 @@ public class GoogleAiStudioEmbeddingProvider implements EmbeddingModelProvider<G
                 .apiKey(apiKey)
                 .modelName(model)
                 .timeout(Duration.ofSeconds(options.getLong("timeout")))
+                .taskType(context.forQueries() ? GoogleAiEmbeddingModel.TaskType.RETRIEVAL_QUERY : GoogleAiEmbeddingModel.TaskType.RETRIEVAL_DOCUMENT)
                 .build();
     }
 
+    @Override
+    public boolean supportsQueryOptimization() {
+        return true;
+    }
+
     @NotNull
     @Override
     public Map<String, Object> getDefaultOptions() {
@@ -44,4 +50,4 @@ public Map<String, Object> getDefaultOptions() {
     public String getDisplayName(@Nullable EmbeddingContext context) {
         return "Google AI Studio";
     }
-}
+}