|
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