OpenAi nos facilita un método para listar sus modelos en producción o desarrollo, y alguna características más como si este se puede afinar o simplemente utilizar. Este sencillo código te permite obtenerlos en consola y en dos ficheros del tipo «CSV».
import os
from colorama import init
init(autoreset=True)
from colorama import Fore, Back, Style
import openai
import pandas as pd
openai.api_key = "Aqui ty KEY"
os.system("cls")
def ready(data):
if str(data) == ("True"):
salida = Fore.GREEN +str(data) + Style.RESET_ALL
else:
if str(data) == ("openai"):
salida = Fore.GREEN +str(data) + Style.RESET_ALL
else:
if str(data) == ("system"):
salida = Fore.YELLOW +str(data) + Style.RESET_ALL
else:
salida = Fore.RED +str(data) + Style.RESET_ALL
return salida
Modelos = openai.Model.list()
Engines = openai.Engine.list()
print("__________________________________\n")
print("MOTORES totales: ", len(Engines["data"]))
print("Información de MOTORES disponibles")
print("__________________________________")
for i, elemento in enumerate(Engines["data"]):
print(i,elemento["id"],(ready(elemento["ready"])))
print("__________________________________\n")
print("MODELOS totales: ", len(Modelos["data"]))
print("Información de MODELOS disponibles")
print("__________________________________")
for i, elemento in enumerate(Modelos["data"]):
print(i,elemento["id"],(ready(elemento["owned_by"])))
data = pd.DataFrame(Engines["data"])
data.to_csv("Motores.csv")
data = pd.DataFrame(Modelos["data"])
data.to_csv("Modelos.csv")El resultado será algo similar a esto:
