def test_delete_skillset(self, api_key, endpoint, index_name, **kwargs):
client = SearchServiceClient(endpoint, AzureKeyCredential(api_key)).get_skillsets_client()
s = EntityRecognitionSkill(inputs=[InputFieldMappingEntry(name="text", source="/document/content")],
outputs=[OutputFieldMappingEntry(name="organizations", target_name="organizations")])
result = client.create_skillset(name='test-ss', skills=[s], description="desc")
assert len(client.get_skillsets()) == 1
client.delete_skillset("test-ss")
assert len(client.get_skillsets()) == 0
async def test_get_skillsets(self, api_key, endpoint, index_name, **kwargs):
client = SearchServiceClient(endpoint, AzureKeyCredential(api_key)).get_skillsets_client()
s = EntityRecognitionSkill(inputs=[InputFieldMappingEntry(name="text", source="/document/content")],
outputs=[OutputFieldMappingEntry(name="organizations", target_name="organizations")])
await client.create_skillset(name='test-ss-1', skills=[s], description="desc1")
await client.create_skillset(name='test-ss-2', skills=[s], description="desc2")
result = await client.get_skillsets()
assert isinstance(result, list)
assert all(isinstance(x, Skillset) for x in result)
assert set(x.name for x in result) == {"test-ss-1", "test-ss-2"}
async def test_create_or_update_skillset_inplace(self, api_key, endpoint, index_name, **kwargs):
client = SearchServiceClient(endpoint, AzureKeyCredential(api_key)).get_skillsets_client()
s = EntityRecognitionSkill(inputs=[InputFieldMappingEntry(name="text", source="/document/content")],
outputs=[OutputFieldMappingEntry(name="organizations", target_name="organizations")])
ss = await client.create_or_update_skillset(name='test-ss', skills=[s], description="desc1")
await client.create_or_update_skillset(name='test-ss', skills=[s], description="desc2", skillset=ss)
assert len(await client.get_skillsets()) == 1
result = await client.get_skillset("test-ss")
assert isinstance(result, Skillset)
assert result.name == "test-ss"
assert result.description == "desc2"
async def test_delete_skillset_if_unchanged(self, api_key, endpoint, index_name, **kwargs):
client = SearchServiceClient(endpoint, AzureKeyCredential(api_key)).get_skillsets_client()
s = EntityRecognitionSkill(inputs=[InputFieldMappingEntry(name="text", source="/document/content")],
outputs=[OutputFieldMappingEntry(name="organizations", target_name="organizations")])
result = await client.create_skillset(name='test-ss', skills=[s], description="desc")
etag = result.e_tag
updated = await client.create_or_update_skillset(name='test-ss', skills=[s], description="updated")
updated.e_tag = etag
with pytest.raises(HttpResponseError):
await client.delete_skillset(updated, match_condition=MatchConditions.IfNotModified)
def test_create_or_update_skillset_if_unchanged(self, api_key, endpoint, index_name, **kwargs):
client = SearchServiceClient(endpoint, AzureKeyCredential(api_key)).get_skillsets_client()
s = EntityRecognitionSkill(inputs=[InputFieldMappingEntry(name="text", source="/document/content")],
outputs=[OutputFieldMappingEntry(name="organizations", target_name="organizations")])
ss = client.create_or_update_skillset(name='test-ss', skills=[s], description="desc1")
etag = ss.e_tag
client.create_or_update_skillset(name='test-ss', skills=[s], description="desc2", skillset=ss)
assert len(client.get_skillsets()) == 1
ss.e_tag = etag
with pytest.raises(HttpResponseError):
client.create_or_update_skillset(name='test-ss', skills=[s], skillset=ss, match_condition=MatchConditions.IfNotModified)