make_assistant_thread.py

from dataproc import Dart, OPENAICommunicator
import os
import time

d = Dart()
communicator = OPENAICommunicator()

# # 기업 정보 다운로드
# df = d.download_corp_list()

# 원하는 회사명, 연도, 보고서 타입 입력
corp_name = 'NAVER'
bsns_year = '2023'
reprt_type = '사업보고서' 
reprt_code = '11011' # 사업보고서
uid = '1'

# # 리포트 다운로드
d.download_report(corp_name, bsns_year, reprt_type)
d.parse_xml(corp_name, bsns_year, reprt_type)

'''

assistant_id_env_name = f"{corp_name}_{bsns_year}_{reprt_code}_assistant_id".upper()
thread_id_env_name = f"{corp_name}_{bsns_year}_{reprt_code}_thread_id_{uid}".upper()
vector_store_id_env_name = f"{corp_name}_{bsns_year}_{reprt_code}_vector_store_id".upper()

# assistant_id 확인 및 생성
if not os.getenv(assistant_id_env_name):
    vector_store_id = communicator.create_vector_store()
    file_ids = communicator.upload_files(corp_name, bsns_year, reprt_type, vector_store_id)  # 파일 업로드
    assistant_id = communicator.create_assistant(corp_name, bsns_year, reprt_code, vector_store_id)  # 어시스턴트 생성
    # communicator.update_assistant(assistant_id, vector_store_id)

    # 환경 변수에 assistant_id 저장
    with open(".env", "a") as env_file:
        env_file.write(f"{assistant_id_env_name}={assistant_id}\n")
        env_file.write(f"{vector_store_id_env_name}={vector_store_id}\n")
else:
    assistant_id = os.getenv(assistant_id_env_name)
    vector_store_id = os.getenv(vector_store_id_env_name)

# print(assistant_id)

# thread_id 확인 및 생성
if not os.getenv(thread_id_env_name):
    thread_id = communicator.create_thread()
    
    # 환경 변수에 thread_id 저장
    with open(".env", "a") as env_file:
        env_file.write(f"{thread_id_env_name}={thread_id}\n")
else:
    thread_id = os.getenv(thread_id_env_name)

# # message_id = communicator.create_message(thread_id)  # 최초 메시지 생성

instructions = input("지시사항을 입력해주세요 ('종료'를 입력하면 프로그램이 종료됩니다): ")

if instructions == '종료':
    print("프로그램을 종료합니다.")
else:
    communicator.create_run(thread_id, assistant_id, instructions)
    
'''