Skip to content

Commit ba0c4e4

Browse files
committed
fix: add missing has_model method to FakeModelStore in test
The test_basic_chat_completion test was failing because FakeModelStore was missing the has_model() method required by OpenAIMixin.
1 parent 8d1fded commit ba0c4e4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/unit/providers/test_bedrock.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@ async def test_basic_chat_completion():
3838
adapter = BedrockInferenceAdapter(config=config)
3939

4040
class FakeModelStore:
41+
async def has_model(self, model_id):
42+
return True
43+
4144
async def get_model(self, model_id):
4245
return SimpleNamespace(provider_resource_id="meta.llama3-1-8b-instruct-v1:0")
4346

0 commit comments

Comments
 (0)