| from .base import BaseAPI | |
| from typing import List | |
| class OpenAIAPI(BaseAPI): | |
| def get_caption(self, prompt: str, user_prompt: str, images_base64: List[str]) -> str: | |
| url_suffix = f'/v1/completions' | |
| return super().get_caption(prompt, user_prompt, images_base64, url_suffix) | |
| def parse_response(self, response: dict) -> str: | |
| return response['choices'][0]['text'] | |