Skip to content

Commit 87f5fb6

Browse files
committed
switched auth to azd cli
1 parent fe1613e commit 87f5fb6

File tree

2 files changed

+19
-16
lines changed
  • app/indexer

2 files changed

+19
-16
lines changed

app/indexer/cli/src/main/java/com/microsoft/openai/samples/indexer/CLI.java

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import java.nio.file.Path;
55
import java.util.concurrent.Callable;
66

7+
import com.azure.identity.AzureDeveloperCliCredentialBuilder;
78
import com.microsoft.openai.samples.indexer.embeddings.AzureOpenAIEmbeddingService;
89
import com.microsoft.openai.samples.indexer.embeddings.TextEmbeddingsService;
910
import com.microsoft.openai.samples.indexer.index.AzureSearchClientFactory;
@@ -74,10 +75,12 @@ public Integer call() throws Exception {
7475

7576
@Command(name = "add")
7677
public void addCommand() {
77-
TokenCredential tokenCredential = new AzureCliCredentialBuilder().build();
78-
TextEmbeddingsService textEmbeddingsService = new AzureOpenAIEmbeddingService(openaiServiceName, openaiEmbdeployment, tokenCredential, verbose);
79-
AzureSearchClientFactory azureSearchClientFactory = new AzureSearchClientFactory(searchservice, tokenCredential, index, verbose);
80-
SearchIndexManager searchIndexManager = new SearchIndexManager(azureSearchClientFactory,searchanalyzername,textEmbeddingsService);
78+
TokenCredential tokenCredential = new AzureDeveloperCliCredentialBuilder().build();
79+
80+
SearchIndexManager searchIndexManager = new SearchIndexManager(
81+
new AzureSearchClientFactory(searchservice, tokenCredential, index, verbose),
82+
searchanalyzername,
83+
new AzureOpenAIEmbeddingService(openaiServiceName, openaiEmbdeployment, tokenCredential, verbose));
8184

8285
searchIndexManager.createIndex();
8386

app/indexer/core/src/main/java/com/microsoft/openai/samples/indexer/index/SearchIndexManager.java

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -32,35 +32,35 @@
3232

3333

3434
public class SearchIndexManager {
35-
private AzureSearchClientFactory searchInfo;
35+
private AzureSearchClientFactory azureSearchClientFactory;
3636
private String searchAnalyzerName;
3737
private boolean useAcls;
3838
private TextEmbeddingsService embeddingsService;
3939
private static final Logger logger = LoggerFactory.getLogger(SearchIndexManager.class);
4040

41-
public SearchIndexManager(AzureSearchClientFactory searchInfo, String searchAnalyzerName, TextEmbeddingsService embeddingsService) {
42-
this.searchInfo = searchInfo;
41+
public SearchIndexManager(AzureSearchClientFactory azureSearchClientFactory, String searchAnalyzerName, TextEmbeddingsService embeddingsService) {
42+
this.azureSearchClientFactory = azureSearchClientFactory;
4343
this.searchAnalyzerName = searchAnalyzerName;
4444
this.embeddingsService = embeddingsService;
4545
}
4646

4747
public void createIndex() {
48-
if (searchInfo.isVerbose()) {
49-
logger.debug("Ensuring search index {} exists", searchInfo.getIndexName());
48+
if (azureSearchClientFactory.isVerbose()) {
49+
logger.debug("Ensuring search index {} exists", azureSearchClientFactory.getIndexName());
5050
}
5151

5252

53-
SearchIndexClient searchIndexClient = searchInfo.createSearchIndexClient();
53+
SearchIndexClient searchIndexClient = azureSearchClientFactory.createSearchIndexClient();
5454
SearchIndex index = null;
5555
try {
56-
index = searchIndexClient.getIndex(searchInfo.getIndexName());
56+
index = searchIndexClient.getIndex(azureSearchClientFactory.getIndexName());
5757
}catch(HttpResponseException httpEx) {
5858
if (httpEx.getResponse().getStatusCode() == 404)
59-
logger.info("index {} does not exist. Creating..", searchInfo.getIndexName());
59+
logger.info("index {} does not exist. Creating..", azureSearchClientFactory.getIndexName());
6060
}
6161

6262
if ( index != null) {
63-
logger.info("index {} already exists. Skipping creation", searchInfo.getIndexName());
63+
logger.info("index {} already exists. Skipping creation", azureSearchClientFactory.getIndexName());
6464
return;
6565
}
6666

@@ -109,7 +109,7 @@ public void createIndex() {
109109
.setFilterable(true));
110110
}
111111

112-
index = new SearchIndex(searchInfo.getIndexName(), fields);
112+
index = new SearchIndex(azureSearchClientFactory.getIndexName(), fields);
113113

114114
index.setSemanticSearch(new SemanticSearch().setConfigurations(Arrays.asList(new SemanticConfiguration(
115115
"default", new SemanticPrioritizedFields()
@@ -125,7 +125,7 @@ public void createIndex() {
125125

126126
searchIndexClient.createIndex(index);
127127

128-
logger.info("Created index {}", searchInfo.getIndexName());
128+
logger.info("Created index {}", azureSearchClientFactory.getIndexName());
129129
}
130130

131131
public void updateContent(List<Section> sections) {
@@ -135,7 +135,7 @@ public void updateContent(List<Section> sections) {
135135
sectionBatches.add(sections.subList(i, Math.min(i + MAX_BATCH_SIZE, sections.size())));
136136
}
137137

138-
SearchClient searchClient = searchInfo.createSearchClient();
138+
SearchClient searchClient = azureSearchClientFactory.createSearchClient();
139139
for (int batchIndex = 0; batchIndex < sectionBatches.size(); batchIndex++) {
140140
List<Section> sectionBatch = sectionBatches.get(batchIndex);
141141
List<Map<String, Object>> documents = new ArrayList<>();

0 commit comments

Comments
 (0)