| from .base import BaseAPI | |
| from typing import List | |
| class AzureAPI(BaseAPI): | |
| def get_headers(self): | |
| return { | |
| 'Content-Type': 'application/json', | |
| 'api-key': self.key | |
| } | |
| def get_caption(self, sys_prompt: str, user_prompt: str, images_base64: List[str]) -> str: | |
| url_suffix = f'/openai/deployments/{self.model}/chat/completions?api-version=2024-02-15-preview' | |
| return super().get_caption(sys_prompt, user_prompt, images_base64, url_suffix) | |
| def parse_response(self, response: dict) -> str: | |
| return response['choices'][0]['message']['content'] | |