utils/anthropic.py

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']