| from .base import BaseAPI | |
| from typing import List | |
| class AnthropicAPI(BaseAPI): | |
| def get_headers(self): | |
| return { | |
| 'Content-Type': 'application/json', | |
| 'x-api-key': self.key | |
| } | |
| def get_caption(self, prompt: str, user_prompt: str, images_base64: List[str]) -> str: | |
| url_suffix = f'/v1/messages' | |
| return super().get_caption(prompt, user_prompt, images_base64, url_suffix) | |
| def parse_response(self, response: dict) -> str: | |
| return response['completion'] | |