import os
import time
import httpx  # Import httpx for error handling

# Ensure your API keys and App ID are correctly set:
os.environ["NCP_CLOVASTUDIO_API_KEY"] = "NTA0MXU4vs+M"  # Replace with your actual API key
os.environ["NCP_APIGW_API_KEY"] = "OOJfdEhIsinKpG"  # Replace with your actual API key
os.environ["NCP_CLOVASTUDIO_APP_ID"] = "7a1b5a3a7"  # Replace with your actual App ID

from langchain_community.embeddings import ClovaXEmbeddings

# Verify that the model name is correct:
embeddings = ClovaXEmbeddings(
    model="clir-emb-dolphin"
)
query = "CLOVA Studio는 HyperCLOVA X 언어 모델을 활용하여 AI 서비스를 손쉽게 만들 수 있는 개발 도구입니다."
single_vector = embeddings.embed_query(query)

text1 = "CLOVA Studio는 HyperCLOVA X 언어 모델을 활용하여 AI 서비스를 손쉽게 만들 수 있는 개발 도구입니다."
text2 = "LangChain은 언어 모델 기반 애플리케이션 개발을 지원하는 오픈소스 프레임워크입니다."
document = [text1, text2]
multiple_vector = embeddings.embed_documents(document)


# Wrap the embed_documents call in a try-except block:
try:
    multiple_vector = embeddings.embed_documents(document)
    print("Embedding successful:", multiple_vector)
except httpx.HTTPStatusError as e:
    print(f"Error embedding documents: {e}")
    print("Check your API credentials, App ID, and input data.")
except Exception as e:
    print(f"An unexpected error occurred: {e}")