본문 바로가기
배워야 산다/OPENAI

openai - 파인튜닝:모델명확인

by 인라인타지마 2023. 7. 13.

 

 

import openai
import datetime

# OpenAI API 키 설정
openai.api_key = "[openai_api_key]"

# 모든 모델 불러오기
response = openai.Model.list()

def get_owner(model):
    if "owned_by" in model:
        # 만약 "owned_by" 필드가 존재하고 현재 사용자의 것일 경우 소유자로 반환
        if model["owned_by"] == "user-5hpbi6qihf8tgrefddg9ok7f":
            return "Your Model"
    return f"owned_by"

def convert_unix_timestamp(timestamp):
  dt = datetime.datetime.fromtimestamp(timestamp)
  korea_tz = datetime.timezone(datetime.timedelta(hours=9))
  korea_dt = dt.astimezone(korea_tz)
  return korea_dt.strftime('%Y-%m-%d %H:%M:%S')

# 모든 모델 불러오기
response = openai.Model.list()

# 각 모델의 ID와 상태값, 소유자 출력
# for model in response["data"]:
#     model_id = model["id"]
#     status = model.get("status", "pending")
#     owner = get_owner(model)
#     print("Model ID: {}, Status: {}, Owner: {}".format(model_id, status, owner))

print(response)

# 각 모델의 ID와 소유자 출력
for model in response["data"]:
    model_id = model["id"]
    owned_by = model.get("owned_by", "Unknown")
    created_timestamp = model.get("created")
    created_korea_time = convert_unix_timestamp(created_timestamp)
    # print("Model ID: {}, Owned by: {}".format(model_id, owned_by))
    print("Model ID: {}, Owned by: {}, Created: {}".format(model_id, owned_by, created_korea_time))

# 각 모델의 ID, 소유자, 생성일을 리스트에 저장
models = []
for model in response["data"]:
    model_id = model["id"]
    owned_by = model.get("owned_by", "Unknown")
    created_timestamp = model.get("created")
    created_korea_time = convert_unix_timestamp(created_timestamp)
    models.append({"model_id": model_id, "owned_by": owned_by, "created_time": created_korea_time})

# 한국 시간으로 변환된 생성일을 기준으로 정렬하여 출력
sorted_models = sorted(models, key=lambda x: x["created_time"])

for model in sorted_models:
    print("Model ID: {}, Owned by: {}, Created: {}".format(model["model_id"], model["owned_by"], model["created_time"]))
반응형