mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-04-18 01:52:20 +00:00
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""Anthropic (Claude) provider implementation.
|
|
|
|
Anthropic's Claude models are excellent for text generation and translation.
|
|
Claude Haiku is particularly fast and affordable for notification enhancement.
|
|
"""
|
|
from typing import Optional
|
|
from .base import AIProvider, AIProviderError
|
|
|
|
|
|
class AnthropicProvider(AIProvider):
|
|
"""Anthropic provider using their Messages API."""
|
|
|
|
NAME = "anthropic"
|
|
DEFAULT_MODEL = "claude-3-haiku-20240307"
|
|
REQUIRES_API_KEY = True
|
|
API_URL = "https://api.anthropic.com/v1/messages"
|
|
API_VERSION = "2023-06-01"
|
|
|
|
def generate(self, system_prompt: str, user_message: str,
|
|
max_tokens: int = 200) -> Optional[str]:
|
|
"""Generate a response using Anthropic's API.
|
|
|
|
Note: Anthropic uses a different API format than OpenAI.
|
|
The system prompt goes in a separate field, not in messages.
|
|
|
|
Args:
|
|
system_prompt: System instructions
|
|
user_message: User message to process
|
|
max_tokens: Maximum response length
|
|
|
|
Returns:
|
|
Generated text or None if failed
|
|
|
|
Raises:
|
|
AIProviderError: If API key is missing or request fails
|
|
"""
|
|
if not self.api_key:
|
|
raise AIProviderError("API key required for Anthropic")
|
|
|
|
# Anthropic uses a different format - system is a top-level field
|
|
payload = {
|
|
'model': self.model,
|
|
'system': system_prompt,
|
|
'messages': [
|
|
{'role': 'user', 'content': user_message},
|
|
],
|
|
'max_tokens': max_tokens,
|
|
}
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'x-api-key': self.api_key,
|
|
'anthropic-version': self.API_VERSION,
|
|
}
|
|
|
|
result = self._make_request(self.API_URL, payload, headers)
|
|
|
|
try:
|
|
# Anthropic returns content as array of content blocks
|
|
content = result['content']
|
|
if isinstance(content, list) and len(content) > 0:
|
|
return content[0].get('text', '').strip()
|
|
return str(content).strip()
|
|
except (KeyError, IndexError) as e:
|
|
raise AIProviderError(f"Unexpected response format: {e}")
|