feat(llm_clients): 添加CodexOAuthClient支持OAuth认证的Codex模型

tradingagents/llm_clients/codex_oauth_client.py

@@ -0,0 +1,43 @@
+from typing import Any, Optional
+
+from .base_client import BaseLLMClient
+from .validators import validate_model
+
+
+class CodexOAuthClient(BaseLLMClient):
+    """Client for ChatGPT OAuth Codex models."""
+
+    def __init__(self, model: str, base_url: Optional[str] = None, **kwargs):
+        super().__init__(model, base_url, **kwargs)
+
+    def get_llm(self) -> Any:
+        """Return configured ChatCodexOAuth instance."""
+        try:
+            from langchain_codex_oauth import ChatCodexOAuth
+        except ModuleNotFoundError as exc:
+            raise ModuleNotFoundError(
+                "langchain-codex-oauth is required for llm_provider='codex_oauth'. "
+                "Install dependencies and run `tradingagents auth login`."
+            ) from exc
+
+        llm_kwargs = {"model": self.model}
+
+        if self.base_url:
+            llm_kwargs["base_url"] = self.base_url
+
+        for key in (
+            "timeout",
+            "max_retries",
+            "reasoning_effort",
+            "max_tokens",
+            "temperature",
+            "callbacks",
+        ):
+            if key in self.kwargs:
+                llm_kwargs[key] = self.kwargs[key]
+
+        return ChatCodexOAuth(**llm_kwargs)
+
+    def validate_model(self) -> bool:
+        """Validate model for codex_oauth."""
+        return validate_model("codex_oauth", self.model)