Add ability to list modelsHEADmain

1 files changed, 12 insertions, 2 deletions

diff --git a/src/claude/chat.py b/src/claude/chat.py
index 6c009d7..64be99b 100644
--- a/src/claude/chat.py
+++ b/src/claude/chat.py
@@ -13,6 +13,12 @@ import httpx
 from .streaming import StreamParser, parse_sse_stream
 from .types import AssistantMessage, StreamChunk, ToolHandler, ToolResult, ToolUse
 
+MODELS: dict[str, dict[str, str]] = {
+    "claude-haiku-4-5-20251001": {"family": "haiku", "display": "Haiku 4.5"},
+    "claude-sonnet-4-6": {"family": "sonnet", "display": "Sonnet 4.6"},
+    "claude-opus-4-6": {"family": "opus", "display": "Opus 4.6"},
+}
+
 BETA_BASE = "oauth-2025-04-20,interleaved-thinking-2025-05-14,prompt-caching-scope-2026-01-05,claude-code-20250219"
 BETA_ADAPTIVE = "claude-code-20250219,oauth-2025-04-20,interleaved-thinking-2025-05-14,prompt-caching-scope-2026-01-05,effort-2025-11-24,adaptive-thinking-2026-01-28"
 
@@ -26,6 +32,10 @@ class ChatClient:
     BASE_URL = "https://api.anthropic.com"
     API_VERSION = "2023-06-01"
 
+    @staticmethod
+    def list_models() -> dict[str, dict[str, str]]:
+        return dict(MODELS)
+
     def __init__(
         self,
         api_key: str | None = None,
@@ -131,7 +141,7 @@ class ChatClient:
         )
         url = f"{self.BASE_URL}/v1/messages"
 
-        last_error = None
+        last_error: Exception | None = None
         backoff = self.backoff_factor
         for attempt in range(self.max_retries):
             try:
@@ -143,7 +153,7 @@ class ChatClient:
                 if attempt < self.max_retries - 1:
                     await asyncio.sleep(backoff)
                     backoff *= 2
-        raise last_error
+        raise last_error  # type: ignore[misc]
 
     async def stream(
         self,