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

chapter1/context/config.py

@@ -0,0 +1,218 @@
+"""
+Configuration module for Context-Aware Agent
+"""
+
+import os
+from typing import Optional
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+
+def _reasoning_safe_temperature(model, requested=1.0):
+    """Reasoning models (Kimi K3, GPT-5, ...) only accept temperature=1.
+    Return 1 for those; otherwise the requested value so non-reasoning
+    providers (Doubao, DeepSeek, older Moonshot) are unchanged."""
+    m = str(model or "").lower().replace("/", "-")
+    return 1 if ("kimi-k3" in m or "gpt-5" in m) else requested
+
+
+# Provider resolution lives in the shared agentbook package so every chapter
+# stays consistent; see agentbook/providers.py. The fallback keeps this
+# experiment runnable from a checkout where agentbook is not installed.
+try:
+    from agentbook.providers import (
+        PROVIDERS,
+        SUPPORTED_PROVIDERS,
+        canonical_provider,
+        canonical_provider as _canonical_provider,
+        map_model_to_openrouter,
+        resolve_backend,
+        resolve_llm_backend,
+    )
+except ImportError:  # pragma: no cover - exercised only without the package
+    import sys as _sys
+
+    _sys.path.insert(
+        0, str(__import__("pathlib").Path(__file__).resolve().parents[2])
+    )
+    from agentbook.providers import (
+        PROVIDERS,
+        SUPPORTED_PROVIDERS,
+        canonical_provider,
+        canonical_provider as _canonical_provider,
+        map_model_to_openrouter,
+        resolve_backend,
+        resolve_llm_backend,
+    )
+
+
+class Config:
+    """Configuration settings for the agent"""
+    
+    # Provider Configuration
+    LLM_PROVIDER: str = os.getenv("LLM_PROVIDER", "doubao").lower()
+    
+    # API Configuration
+    DASHSCOPE_API_KEY: str = os.getenv("DASHSCOPE_API_KEY", "")
+    DASHSCOPE_BASE_URL: str = os.getenv(
+        "DASHSCOPE_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"
+    )
+
+    SILICONFLOW_API_KEY: str = os.getenv("SILICONFLOW_API_KEY", "")
+    SILICONFLOW_BASE_URL: str = "https://api.siliconflow.cn/v1"
+    
+    ARK_API_KEY: str = os.getenv("ARK_API_KEY", "")
+    ARK_BASE_URL: str = "https://ark.cn-beijing.volces.com/api/v3"
+    
+    MOONSHOT_API_KEY: str = os.getenv("MOONSHOT_API_KEY", "")
+    MOONSHOT_BASE_URL: str = "https://api.moonshot.cn/v1"
+
+    DEEPSEEK_API_KEY: str = os.getenv("DEEPSEEK_API_KEY", "")
+    DEEPSEEK_BASE_URL: str = os.getenv(
+        "DEEPSEEK_BASE_URL", "https://api.deepseek.com"
+    )
+
+    ZHIPU_API_KEY: str = os.getenv("ZHIPU_API_KEY", "")
+    ZHIPU_BASE_URL: str = "https://open.bigmodel.cn/api/paas/v4"
+    
+    # Model Configuration (defaults based on provider)
+    MODEL_NAME: str = os.getenv("MODEL_NAME", "")  # Will be set based on provider if not specified
+    MODEL_TEMPERATURE: float = float(os.getenv("MODEL_TEMPERATURE", "0.3"))
+    MODEL_MAX_TOKENS: int = int(os.getenv("MODEL_MAX_TOKENS", "1000"))
+    
+    # Agent Configuration
+    MAX_ITERATIONS: int = int(os.getenv("MAX_ITERATIONS", "10"))
+    ENABLE_REASONING: bool = os.getenv("ENABLE_REASONING", "true").lower() == "true"
+    
+    # Test Configuration
+    TEST_PDF_URL: str = os.getenv(
+        "TEST_PDF_URL",
+        "https://www.berkshirehathaway.com/qtrly/1stqtr23.pdf"
+    )
+    
+    # Currency Configuration (Example rates - in production use real API)
+    EXCHANGE_RATES = {
+        "USD": 1.0,
+        "EUR": 0.92,
+        "GBP": 0.79,
+        "JPY": 149.50,
+        "CNY": 7.24,
+        "CAD": 1.36,
+        "AUD": 1.53,
+        "CHF": 0.88,
+        "INR": 83.12,
+        "SGD": 1.34
+    }
+    
+    # Logging Configuration
+    LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
+    LOG_FILE: Optional[str] = os.getenv("LOG_FILE")
+    LOG_FORMAT: str = "%(asctime)s - %(levelname)s - %(name)s - %(message)s"
+    
+    # File paths
+    RESULTS_DIR: str = "results"
+    TEST_PDFS_DIR: str = "fixtures/pdfs"
+    
+    @classmethod
+    def get_api_key(cls, provider: str = None) -> str:
+        """
+        Get API key for the specified provider
+        
+        Args:
+            provider: Provider name (defaults to LLM_PROVIDER)
+            
+        Returns:
+            API key for the provider
+        """
+        provider = provider or cls.LLM_PROVIDER
+        # The shared registry knows every provider's key variables, so this
+        # stays correct as providers are added there.
+        try:
+            return PROVIDERS[_canonical_provider(provider)].api_key()
+        except KeyError:
+            return ""
+    
+    @classmethod
+    def get_default_model(cls, provider: str = None) -> str:
+        """
+        Get default model for the specified provider
+        
+        Args:
+            provider: Provider name (defaults to LLM_PROVIDER)
+            
+        Returns:
+            Default model name for the provider
+        """
+        provider = provider or cls.LLM_PROVIDER
+        provider = provider.lower()
+        
+        if cls.MODEL_NAME:
+            return cls.MODEL_NAME
+
+        try:
+            return PROVIDERS[_canonical_provider(provider)].default_model
+        except KeyError:
+            return ""
+    
+    @classmethod
+    def validate(cls, provider: str = None) -> bool:
+        """
+        Validate required configuration
+        
+        Args:
+            provider: Provider to validate (defaults to LLM_PROVIDER)
+        
+        Returns:
+            True if configuration is valid
+        """
+        provider = provider or cls.LLM_PROVIDER
+        # resolve_backend already accounts for providers that need no key
+        # (ollama) and for the OpenRouter fallback, and its error names the
+        # exact variables to set -- so a missing key is not the only signal.
+        try:
+            resolve_backend(provider)
+        except ValueError as exc:
+            print(f"ERROR: {exc}")
+            print("Please set it in .env file or as environment variable")
+            return False
+        
+        return True
+    
+    @classmethod
+    def create_directories(cls):
+        """Create necessary directories if they don't exist"""
+        os.makedirs(cls.RESULTS_DIR, exist_ok=True)
+        os.makedirs(cls.TEST_PDFS_DIR, exist_ok=True)
+    
+    @classmethod
+    def get_model_config(cls) -> dict:
+        """
+        Get model configuration as dictionary
+        
+        Returns:
+            Model configuration dict
+        """
+        return {
+            "model": cls.MODEL_NAME,
+            "temperature": _reasoning_safe_temperature(cls.MODEL_NAME, cls.MODEL_TEMPERATURE),
+            "max_tokens": cls.MODEL_MAX_TOKENS
+        }
+    
+    @classmethod
+    def print_config(cls):
+        """Print current configuration (hiding sensitive data)"""
+        provider = canonical_provider(cls.LLM_PROVIDER)
+        api_key = cls.get_api_key(provider)
+        print("\n" + "="*50)
+        print("CONFIGURATION")
+        print("="*50)
+        print(f"Provider: {provider}")
+        print(f"Model: {cls.MODEL_NAME}")
+        print(f"Temperature: {cls.MODEL_TEMPERATURE}")
+        print(f"Max Tokens: {cls.MODEL_MAX_TOKENS}")
+        print(f"Max Iterations: {cls.MAX_ITERATIONS}")
+        print(f"API Key Set: {'Yes' if api_key else 'No'}")
+        print(f"Log Level: {cls.LOG_LEVEL}")
+        print("="*50 + "\n")