|
11 | 11 | from api.openrouter_client import OpenRouterClient
|
12 | 12 | from api.bedrock_client import BedrockClient
|
13 | 13 | from api.azureai_client import AzureAIClient
|
| 14 | +from api.dashscope_client import DashscopeClient |
14 | 15 | from adalflow import GoogleGenAIClient, OllamaClient
|
15 | 16 |
|
16 | 17 | # Get API keys from environment variables
|
|
53 | 54 | "OpenRouterClient": OpenRouterClient,
|
54 | 55 | "OllamaClient": OllamaClient,
|
55 | 56 | "BedrockClient": BedrockClient,
|
56 |
| - "AzureAIClient": AzureAIClient |
| 57 | + "AzureAIClient": AzureAIClient, |
| 58 | + "DashscopeClient": DashscopeClient |
57 | 59 | }
|
58 | 60 |
|
59 | 61 | def replace_env_placeholders(config: Union[Dict[str, Any], List[Any], str, Any]) -> Union[Dict[str, Any], List[Any], str, Any]:
|
@@ -121,14 +123,15 @@ def load_generator_config():
|
121 | 123 | if provider_config.get("client_class") in CLIENT_CLASSES:
|
122 | 124 | provider_config["model_client"] = CLIENT_CLASSES[provider_config["client_class"]]
|
123 | 125 | # Fall back to default mapping based on provider_id
|
124 |
| - elif provider_id in ["google", "openai", "openrouter", "ollama", "bedrock", "azure"]: |
| 126 | + elif provider_id in ["google", "openai", "openrouter", "ollama", "bedrock", "azure", "dashscope"]: |
125 | 127 | default_map = {
|
126 | 128 | "google": GoogleGenAIClient,
|
127 | 129 | "openai": OpenAIClient,
|
128 | 130 | "openrouter": OpenRouterClient,
|
129 | 131 | "ollama": OllamaClient,
|
130 | 132 | "bedrock": BedrockClient,
|
131 |
| - "azure": AzureAIClient |
| 133 | + "azure": AzureAIClient, |
| 134 | + "dashscope": DashscopeClient |
132 | 135 | }
|
133 | 136 | provider_config["model_client"] = default_map[provider_id]
|
134 | 137 | else:
|
|
0 commit comments