Skip to content

Commit a5b3388

Browse files
committed
fix: list_foundation_models() fetch function
1 parent d2d7e40 commit a5b3388

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

adalflow/adalflow/components/model_client/bedrock_client.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -220,12 +220,14 @@ def list_models(self):
220220

221221
try:
222222
response = self._client.list_foundation_models()
223-
models = response.get("models", [])
223+
models = response.get("modelSummaries", [])
224224
for model in models:
225225
print(f"Model ID: {model['modelId']}")
226-
print(f" Name: {model['name']}")
227-
print(f" Description: {model['description']}")
228-
print(f" Provider: {model['provider']}")
226+
print(f" Name: {model['modelName']}")
227+
print(f" Provider: {model['providerName']}")
228+
print(f" Input: {model['inputModalities']}")
229+
print(f" Output: {model['outputModalities']}")
230+
print(f" InferenceTypesSupported: {model['inferenceTypesSupported']}")
229231
print("")
230232

231233
except Exception as e:

0 commit comments

Comments
 (0)