Skip to content

Commit 0ac6b29

Browse files
committed
Fix more tests
1 parent 9fa396e commit 0ac6b29

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

ansible_ai_connect/ai/api/model_pipelines/langchain/pipelines.py

+3
Original file line numberDiff line numberDiff line change
@@ -459,5 +459,8 @@ class LangchainRoleExplanationPipeline(
459459
def __init__(self, config: LANGCHAIN_PIPELINE_CONFIGURATION):
460460
super().__init__(config=config)
461461

462+
def self_test(self) -> Optional[HealthCheckSummary]:
463+
raise NotImplementedError
464+
462465
def invoke(self, params: RoleExplanationParameters) -> RoleExplanationResponse:
463466
raise NotImplementedError

ansible_ai_connect/ai/api/model_pipelines/ollama/pipelines.py

+14
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
LangchainMetaData,
2222
LangchainPlaybookExplanationPipeline,
2323
LangchainPlaybookGenerationPipeline,
24+
LangchainRoleExplanationPipeline,
2425
LangchainRoleGenerationPipeline,
2526
)
2627
from ansible_ai_connect.ai.api.model_pipelines.ollama.configuration import (
@@ -80,6 +81,19 @@ def get_chat_model(self, model_id):
8081
)
8182

8283

84+
@Register(api_type="ollama")
85+
class OllamaRoleExplanationPipeline(LangchainRoleExplanationPipeline[OllamaConfiguration]):
86+
87+
def __init__(self, config: OllamaConfiguration):
88+
super().__init__(config=config)
89+
90+
def get_chat_model(self, model_id):
91+
return OllamaLLM(
92+
base_url=self.config.inference_url,
93+
model=model_id,
94+
)
95+
96+
8397
@Register(api_type="ollama")
8498
class OllamaPlaybookExplanationPipeline(LangchainPlaybookExplanationPipeline[OllamaConfiguration]):
8599

0 commit comments

Comments
 (0)