Initial commit for TTS project

scripts/tools/perplexity_config.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""
+Perplexity API配置管理
+
+Author: AI Assistant
+Date: 2026-01-12
+Version: 1.0
+
+这个文件提供了Perplexity API的配置管理功能，
+允许通过环境变量或配置文件设置API密钥和其他参数。
+"""
+
+import os
+from typing import Optional
+
+
+class PerplexityConfig:
+    """
+    Perplexity API配置类
+    """
+    
+    def __init__(self):
+        # 从环境变量加载配置
+        self.api_key: Optional[str] = os.getenv("PERPLEXITY_API_KEY")
+        self.api_base: str = os.getenv("PERPLEXITY_API_BASE", "https://api.perplexity.ai")
+        self.model: str = os.getenv("PERPLEXITY_MODEL", "pplx-70b-online")
+        
+    def validate(self) -> bool:
+        """
+        验证配置是否有效
+        """
+        if not self.api_key:
+            print("警告: PERPLEXITY_API_KEY 环境变量未设置")
+            return False
+        return True
+    
+    def get_api_key(self) -> Optional[str]:
+        """
+        获取API密钥
+        """
+        return self.api_key
+    
+    def get_api_base(self) -> str:
+        """
+        获取API基础URL
+        """
+        return self.api_base
+    
+    def get_model(self) -> str:
+        """
+        获取默认模型
+        """
+        return self.model
+    
+    def set_api_key(self, api_key: str):
+        """
+        设置API密钥
+        """
+        self.api_key = api_key
+        # 同时设置环境变量，以便其他使用环境变量的代码也能访问
+        os.environ["PERPLEXITY_API_KEY"] = api_key
+
+
+# 创建全局配置实例
+perplexity_config = PerplexityConfig()
+
+
+def get_perplexity_config() -> PerplexityConfig:
+    """
+    获取全局Perplexity配置实例
+    """
+    return perplexity_config