in tools/aisearch.py [0:0]
def __init__(self):
self.search_service_name = os.getenv("AZURE_SEARCH_SERVICE")
if not self.search_service_name:
logging.error("[aisearch] AZURE_SEARCH_SERVICE environment variable not set.")
raise ValueError("AZURE_SEARCH_SERVICE environment variable not set.")
self.endpoint = f"https://{self.search_service_name}.search.windows.net"
# Initialize the ChainedTokenCredential
try:
self.credential = ChainedTokenCredential(
ManagedIdentityCredential(),
AzureCliCredential()
)
logging.debug("[aisearch] Initialized ChainedTokenCredential with ManagedIdentity and AzureCliCredential.")
except Exception as e:
logging.error(f"[aisearch] Failed to initialize credentials: {e}")
raise
self.clients = {} # Cache SearchClient instances per index