|
| 1 | +package com.datastax.astra.client.database_admin; |
| 2 | + |
| 3 | +import com.datastax.astra.client.Database; |
| 4 | +import com.datastax.astra.client.model.EmbeddingProvider; |
| 5 | +import com.datastax.astra.client.model.FindEmbeddingProvidersResult; |
| 6 | + |
| 7 | +import java.util.Map; |
| 8 | + |
| 9 | +public class FindEmbeddingProviders { |
| 10 | + public static void main(String[] args) { |
| 11 | + Database db = new Database( |
| 12 | + "https://8f915e5a-ea58-4576-bd00-9828984a1023-us-east1.apps.astra.datastax.com", |
| 13 | + "AstraCS:iLPiNPxSSIdefoRdkTWCfWXt:2b360d096e0e6cb732371925ffcc6485541ff78067759a2a1130390e231c2c7a"); |
| 14 | + FindEmbeddingProvidersResult fepr = db.getDatabaseAdmin().findEmbeddingProviders(); |
| 15 | + Map<String, EmbeddingProvider> providers = fepr.getEmbeddingProviders(); |
| 16 | + for (Map.Entry<String, EmbeddingProvider> entry : providers.entrySet()) { |
| 17 | + System.out.println("\nProvider: " + entry.getKey()); |
| 18 | + EmbeddingProvider provider = entry.getValue(); |
| 19 | + |
| 20 | + provider.getHeaderAuthentication().ifPresent(headerAuthentication -> { |
| 21 | + System.out.println("+ Authentication Header"); |
| 22 | + headerAuthentication.getTokens().forEach(token -> { |
| 23 | + System.out.println(" +" |
| 24 | + + " Token accepted=" + token.getAccepted() |
| 25 | + + ", forwarded=" + token.getForwarded()); |
| 26 | + }); |
| 27 | + }); |
| 28 | + System.out.println("+ Models:"); |
| 29 | + for(EmbeddingProvider.Model model : provider.getModels()) { |
| 30 | + System.out.println(" + Name=" + |
| 31 | + model.getName() + ((model.getVectorDimension() != null) ? ", dimension=" + model.getVectorDimension() : "")); |
| 32 | + } |
| 33 | + } |
| 34 | + } |
| 35 | +} |
0 commit comments