21 lines
791 B
Python
21 lines
791 B
Python
class OpenAICredencial:
|
|
def __init__(self, titulo: str, api_key: str, organizacion: str = None):
|
|
"""
|
|
:param titulo: Nombre descriptivo para esta credencial.
|
|
:param api_key: Clave secreta de la API de OpenAI.
|
|
:param organizacion: (Opcional) ID de la organización asociada a la cuenta de OpenAI.
|
|
"""
|
|
self.titulo = titulo
|
|
self.api_key = api_key
|
|
self.organizacion = organizacion
|
|
|
|
def get_headers(self) -> dict:
|
|
"""
|
|
Retorna los encabezados necesarios para autenticar una petición HTTP a OpenAI.
|
|
"""
|
|
headers = {
|
|
"Authorization": f"Bearer {self.api_key}"
|
|
}
|
|
if self.organizacion:
|
|
headers["OpenAI-Organization"] = self.organizacion
|
|
return headers |