Upload folder using huggingface_hub

src/agentscope/credential/__init__.py

@@ -9,6 +9,11 @@ from ._gemini import GeminiCredential
 from ._moonshot import MoonshotCredential
 from ._ollama import OllamaCredential
 from ._openai import OpenAICredential
+from ._openai_compat import (
+    GroqCredential,
+    NvidiaNimCredential,
+    OpenRouterCredential,
+)
 from ._xai import XAICredential
 from ._factory import CredentialFactory
 
@@ -19,9 +24,12 @@ __all__ = [
     "DashScopeCredential",
     "DeepSeekCredential",
     "GeminiCredential",
+    "GroqCredential",
     "MoonshotCredential",
+    "NvidiaNimCredential",
     "OllamaCredential",
     "OpenAICredential",
+    "OpenRouterCredential",
     "XAICredential",
     "CredentialFactory",
 ]

src/agentscope/credential/_factory.py

@@ -11,6 +11,11 @@ from ._gemini import GeminiCredential
 from ._moonshot import MoonshotCredential
 from ._ollama import OllamaCredential
 from ._openai import OpenAICredential
+from ._openai_compat import (
+    GroqCredential,
+    NvidiaNimCredential,
+    OpenRouterCredential,
+)
 from ._xai import XAICredential
 from ._base import CredentialBase
 
@@ -38,9 +43,12 @@ class CredentialFactory:
         DashScopeCredential,
         DeepSeekCredential,
         GeminiCredential,
+        GroqCredential,
         MoonshotCredential,
+        NvidiaNimCredential,
         OllamaCredential,
         OpenAICredential,
+        OpenRouterCredential,
         XAICredential,
     ]
     _adapter: TypeAdapter[CredentialBase] | None = None

src/agentscope/credential/_openai_compat.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+"""OpenAI-compatible credentials (Groq, OpenRouter, NVIDIA NIM)."""
+from typing import Literal
+
+from pydantic import ConfigDict, Field
+
+from ._openai import OpenAICredential
+
+
+class GroqCredential(OpenAICredential):
+    """The Groq credential model."""
+
+    model_config = ConfigDict(
+        title="Groq API",
+    )
+
+    type: Literal["groq_credential"] = "groq_credential"
+    """The credential type."""
+
+    base_url: str | None = Field(
+        default="https://api.groq.com/openai/v1",
+        description="The base URL for the Groq API.",
+    )
+    """Custom base URL for the Groq API."""
+
+
+class OpenRouterCredential(OpenAICredential):
+    """The OpenRouter credential model."""
+
+    model_config = ConfigDict(
+        title="OpenRouter API",
+    )
+
+    type: Literal["openrouter_credential"] = "openrouter_credential"
+    """The credential type."""
+
+    base_url: str | None = Field(
+        default="https://openrouter.ai/api/v1",
+        description="The base URL for the OpenRouter API.",
+    )
+    """Custom base URL for the OpenRouter API."""
+
+
+class NvidiaNimCredential(OpenAICredential):
+    """The NVIDIA NIM credential model."""
+
+    model_config = ConfigDict(
+        title="NVIDIA NIM API",
+    )
+
+    type: Literal["nvidia_nim_credential"] = "nvidia_nim_credential"
+    """The credential type."""
+
+    base_url: str | None = Field(
+        default="https://integrate.api.nvidia.com/v1",
+        description="The base URL for the NVIDIA NIM API.",
+    )
+    """Custom base URL for the NVIDIA NIM API."""