Spaces:
Runtime error
Runtime error
File size: 612 Bytes
dc465b0 b7c7aa0 dc465b0 b7c7aa0 dc465b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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']
|