agentscope / test_models.py
ghostdrive1's picture
Upload folder using huggingface_hub (part 3)
df3937f verified
Raw
History Blame Contribute Delete
491 Bytes
import asyncio
from agentscope.credential._openai_compat import NvidiaNimCredential
from agentscope.credential import CredentialFactory
import logging
logging.basicConfig(level=logging.WARNING)
async def test():
cls = CredentialFactory.get_credential_class("nvidia_nim_credential")
print("credential class:", cls)
model_cls = cls.get_chat_model_class()
print("model class:", model_cls)
models = model_cls.list_models()
print("models:", models)
asyncio.run(test())