더북(TheBook)

예제를 살펴보겠습니다.

examples/9_tool_test.py

from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage
from langchain_core.tools import tool, ToolException

load_dotenv()

@tool
def get_word_count(text: str) -> int:
    """텍스트의 단어 수를 반환합니다.
    텍스트 길이나 단어 수를 요청할 때 사용합니다."""
    if not text.strip():
        raise ToolException("텍스트가 비어 있습니다.")
    return len(text.split())