커스텀 도구 배포하기

from crewai.tools import BaseTool
from pydantic import BaseModel, Field


class GeolocateInput(BaseModel):
    """GeolocateTool의 입력 스키마."""
    address: str = Field(..., description="지오코딩할 도로명 주소.")


class GeolocateTool(BaseTool):
    name: str = "Geolocate"
    description: str = "도로명 주소를 위도/경도 좌표로 변환합니다."
    args_schema: type[BaseModel] = GeolocateInput

    def _run(self, address: str) -> str:
        # 구현 로직
        return f"40.7128, -74.0060"

from crewai.tools import tool


@tool("Geolocate")
def geolocate(address: str) -> str:
    """도로명 주소를 위도/경도 좌표로 변환합니다."""
    return "40.7128, -74.0060"

class GeolocateTool(BaseTool):
    name: str = "Geolocate"
    description: str = "도로명 주소를 위도/경도 좌표로 변환합니다."

    def _run(self, address: str) -> str:
        # 동기 구현
        ...

    async def _arun(self, address: str) -> str:
        # 비동기 구현
        ...

from pydantic import BaseModel, Field


class TranslateInput(BaseModel):
    """TranslateTool의 입력 스키마."""
    text: str = Field(..., description="번역할 텍스트.")
    target_language: str = Field(
        default="en",
        description="대상 언어의 ISO 639-1 언어 코드.",
    )

from crewai.tools import BaseTool, EnvVar


class GeolocateTool(BaseTool):
    name: str = "Geolocate"
    description: str = "도로명 주소를 위도/경도 좌표로 변환합니다."
    env_vars: list[EnvVar] = [
        EnvVar(
            name="GEOCODING_API_KEY",
            description="지오코딩 서비스 API 키.",
            required=True,
        ),
    ]

    def _run(self, address: str) -> str:
        ...

crewai-geolocate/
├── pyproject.toml
├── LICENSE
├── README.md
└── src/
    └── crewai_geolocate/
        ├── __init__.py
        └── tools.py

[project]
name = "crewai-geolocate"
version = "0.1.0"
description = "도로명 주소를 지오코딩하는 CrewAI 도구."
requires-python = ">=3.10"
dependencies = [
    "crewai",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

from crewai_geolocate.tools import GeolocateTool

__all__ = ["GeolocateTool"]

from crewai import Agent, Crew, Task
from crewai_geolocate import GeolocateTool

agent = Agent(
    role="Location Analyst",
    goal="주어진 주소의 좌표를 찾습니다.",
    backstory="지리공간 데이터 전문가.",
    tools=[GeolocateTool()],
)

task = Task(
    description="1600 Pennsylvania Avenue, Washington, DC의 좌표를 찾으세요.",
    expected_output="해당 주소의 위도와 경도.",
    agent=agent,
)

crew = Crew(agents=[agent], tasks=[task])
result = crew.kickoff()
print(result)

# 패키지 빌드
uv build

# PyPI에 게시
uv publish

pip install crewai-geolocate

uv add crewai-geolocate

from crewai_geolocate import GeolocateTool

agent = Agent(
    role="Location Analyst",
    tools=[GeolocateTool()],
    # ...
)