Skip to content

Commit 63e5b82

Browse files
committed
Fix more tests
1 parent cb56176 commit 63e5b82

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

ansible_ai_connect/ai/api/model_pipelines/wca/pipelines_onprem.py

+19
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
PlaybookExplanationResponse,
3737
PlaybookGenerationParameters,
3838
PlaybookGenerationResponse,
39+
RoleExplanationParameters,
40+
RoleExplanationResponse,
3941
RoleGenerationParameters,
4042
RoleGenerationResponse,
4143
)
@@ -51,6 +53,7 @@
5153
WCABasePipeline,
5254
WCABasePlaybookExplanationPipeline,
5355
WCABasePlaybookGenerationPipeline,
56+
WCABaseRoleExplanationPipeline,
5457
WCABaseRoleGenerationPipeline,
5558
WcaModelRequestException,
5659
)
@@ -212,6 +215,22 @@ def self_test(self) -> Optional[HealthCheckSummary]:
212215
raise NotImplementedError
213216

214217

218+
@Register(api_type="wca-onprem")
219+
class WCAOnPremRoleExplanationPipeline(
220+
WCAOnPremPipeline[RoleExplanationParameters, RoleExplanationResponse],
221+
WCABaseRoleExplanationPipeline[WCAOnPremConfiguration],
222+
):
223+
224+
def __init__(self, config: WCAOnPremConfiguration):
225+
super().__init__(config=config)
226+
227+
def invoke(self, params: RoleExplanationParameters) -> RoleExplanationResponse:
228+
raise NotImplementedError
229+
230+
def self_test(self) -> Optional[HealthCheckSummary]:
231+
raise NotImplementedError
232+
233+
215234
@Register(api_type="wca-onprem")
216235
class WCAOnPremPlaybookExplanationPipeline(
217236
WCAOnPremPipeline[PlaybookExplanationParameters, PlaybookExplanationResponse],

0 commit comments

Comments
 (0)