import os
os.environ["OPENAI_API_KEY"] = "sk…"  # OpenAI API 키를 환경 변수로 설정

# 랭체인에서 함수를 도구로 등록할 수 있게 해주는 데코레이터
from langchain_core.tools import tool  
# OpenAI LLM을 랭체인에서 사용하기 위한 클래스
from langchain_openai import ChatOpenAI  
# 에이전트 초기화 및 도구 구성에 필요한 클래스들
from langchain.agents import initialize_agent, AgentType, Tool  

# 간단한 덧셈 함수를 정의하고 랭체인 도구로 등록
@tool
def add(a: int, b: int) -> int:
    """두 숫자를 더합니다."""
    return a + b

# 간단한 뺄셈 함수를 정의하고 랭체인 도구로 등록
@tool
def subtract(a: int, b: int) -> int:
    """두 숫자를 뺍니다."""
    return a - b

# 위에서 정의한 도구들을 리스트로 구성
tools = [add, subtract]