ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter1/context/tests/manual/check_provider_config.py

@@ -0,0 +1,129 @@
+#!/usr/bin/env python3
+"""
+Test script to verify provider configuration
+"""
+
+import os
+
+from _bootstrap import add_project_root
+
+add_project_root()
+
+from agent import ContextAwareAgent, ContextMode
+
+def test_providers():
+    """Test different provider configurations"""
+    
+    print("\n" + "="*60)
+    print("🧪 PROVIDER CONFIGURATION TEST")
+    print("="*60)
+
+    # Test Alibaba Cloud Model Studio / Bailian
+    dashscope_key = os.getenv("DASHSCOPE_API_KEY")
+    if dashscope_key:
+        print("\n✅ Alibaba Cloud Model Studio API key found")
+        try:
+            agent = ContextAwareAgent(
+                dashscope_key, ContextMode.FULL, provider="dashscope"
+            )
+            print(f"   Provider: {agent.provider}")
+            print(f"   Model: {agent.model}")
+            print(f"   Base URL: {agent.client.base_url}")
+        except Exception as e:
+            print(f"   ❌ Error: {str(e)}")
+    else:
+        print("\n⚠️ Alibaba Cloud Model Studio API key not found (DASHSCOPE_API_KEY)")
+    
+    # Test SiliconFlow
+    sf_key = os.getenv("SILICONFLOW_API_KEY")
+    if sf_key:
+        print("\n✅ SiliconFlow API key found")
+        try:
+            agent = ContextAwareAgent(sf_key, ContextMode.FULL, provider="siliconflow")
+            print(f"   Provider: {agent.provider}")
+            print(f"   Model: {agent.model}")
+            print(f"   Base URL: {agent.client.base_url}")
+        except Exception as e:
+            print(f"   ❌ Error: {str(e)}")
+    else:
+        print("\n⚠️ SiliconFlow API key not found (SILICONFLOW_API_KEY)")
+    
+    # Test Doubao
+    ark_key = os.getenv("ARK_API_KEY")
+    if ark_key:
+        print("\n✅ Doubao/ARK API key found")
+        try:
+            agent = ContextAwareAgent(ark_key, ContextMode.FULL, provider="doubao")
+            print(f"   Provider: {agent.provider}")
+            print(f"   Model: {agent.model}")
+            print(f"   Base URL: {agent.client.base_url}")
+        except Exception as e:
+            print(f"   ❌ Error: {str(e)}")
+    else:
+        print("\n⚠️ Doubao/ARK API key not found (ARK_API_KEY)")
+
+    # Test DeepSeek
+    deepseek_key = os.getenv("DEEPSEEK_API_KEY")
+    if deepseek_key:
+        print("\n✅ DeepSeek API key found")
+        try:
+            agent = ContextAwareAgent(deepseek_key, ContextMode.FULL, provider="deepseek")
+            print(f"   Provider: {agent.provider}")
+            print(f"   Model: {agent.model}")
+            print(f"   Base URL: {agent.client.base_url}")
+        except Exception as e:
+            print(f"   ❌ Error: {str(e)}")
+    else:
+        print("\n⚠️ DeepSeek API key not found (DEEPSEEK_API_KEY)")
+    
+    # Test custom model
+    if sf_key:
+        print("\n🔧 Testing custom model specification:")
+        try:
+            agent = ContextAwareAgent(sf_key, ContextMode.FULL, 
+                                    provider="siliconflow", 
+                                    model="Qwen/QwQ-32B")
+            print(f"   Provider: {agent.provider}")
+            print(f"   Custom Model: {agent.model}")
+        except Exception as e:
+            print(f"   ❌ Error: {str(e)}")
+    
+    print("\n" + "="*60)
+    print("Test complete!")
+    
+    # Show usage examples
+    print("\n📖 Usage Examples:")
+    print("-"*40)
+
+    if dashscope_key:
+        print("\n# Using Qwen directly through Alibaba Cloud Model Studio:")
+        print("python main.py --provider dashscope")
+        print("python main.py --provider dashscope --model qwen3.7-plus")
+    
+    if sf_key:
+        print("\n# Using SiliconFlow:")
+        print("python main.py --provider siliconflow")
+        print("python main.py --provider siliconflow --model Qwen/QwQ-32B")
+    
+    if ark_key:
+        print("\n# Using Doubao:")
+        print("python main.py --provider doubao")
+        print("python main.py --provider doubao --model doubao-seed-1-6-thinking-250715")
+
+    if deepseek_key:
+        print("\n# Using DeepSeek:")
+        print("python main.py --provider deepseek")
+        print("python main.py --provider deepseek --model deepseek-v4-pro")
+    
+    if not dashscope_key and not sf_key and not ark_key and not deepseek_key:
+        print("\n⚠️ No API keys found. Please set one of:")
+        print("   export DASHSCOPE_API_KEY=your_key")
+        print("   export SILICONFLOW_API_KEY=your_key")
+        print("   export ARK_API_KEY=your_key")
+        print("   export DEEPSEEK_API_KEY=your_key")
+    
+    print("\n" + "="*60)
+
+
+if __name__ == "__main__":
+    test_providers()