ghostdrive1 commited on
Commit
df3937f
·
verified ·
1 Parent(s): eb5bdef

Upload folder using huggingface_hub (part 3)

Browse files
src/agentscope/model/_openrouter/_models/claude-3-haiku.yaml CHANGED
@@ -1,7 +1,9 @@
1
- name: anthropic/claude-3-haiku
2
- api_type: openrouter_chat
3
- description: Claude 3 Haiku via OpenRouter.
4
- max_context: 200000
5
- support_vision: true
6
- support_function_calling: true
7
- support_structured_output: true
 
 
 
1
+ name: anthropic/claude-3-haiku
2
+ label: Claude 3 Haiku
3
+ status: active
4
+ input_types:
5
+ - text/plain
6
+ output_types:
7
+ - text/plain
8
+ context_size: 200000
9
+ output_size: 4096
test_models.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import asyncio
2
+ from agentscope.credential._openai_compat import NvidiaNimCredential
3
+ from agentscope.credential import CredentialFactory
4
+ import logging
5
+
6
+ logging.basicConfig(level=logging.WARNING)
7
+
8
+ async def test():
9
+ cls = CredentialFactory.get_credential_class("nvidia_nim_credential")
10
+ print("credential class:", cls)
11
+ model_cls = cls.get_chat_model_class()
12
+ print("model class:", model_cls)
13
+ models = model_cls.list_models()
14
+ print("models:", models)
15
+
16
+ asyncio.run(test())