Skip to content

Commit ead370d

Browse files
committed
fix: list_foundation_models() fetch function
1 parent 4d80d26 commit ead370d

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
@@ -180,12 +180,14 @@ def list_models(self):
180180

181181
try:
182182
response = self._client.list_foundation_models()
183-
models = response.get("models", [])
183+
models = response.get("modelSummaries", [])
184184
for model in models:
185185
print(f"Model ID: {model['modelId']}")
186-
print(f" Name: {model['name']}")
187-
print(f" Description: {model['description']}")
188-
print(f" Provider: {model['provider']}")
186+
print(f" Name: {model['modelName']}")
187+
print(f" Provider: {model['providerName']}")
188+
print(f" Input: {model['inputModalities']}")
189+
print(f" Output: {model['outputModalities']}")
190+
print(f" InferenceTypesSupported: {model['inferenceTypesSupported']}")
189191
print("")
190192
except Exception as e:
191193
print(f"Error listing models: {e}")

0 commit comments

Comments
 (0)