LangChain 도구 활용하기

import os
from dotenv import load_dotenv
from crewai import Agent, Task, Crew
from crewai.tools import BaseTool
from pydantic import Field
from langchain_community.utilities import GoogleSerperAPIWrapper

# .env 파일에 SERPER_API_KEY를 설정한다. 예시:
# SERPER_API_KEY=<your api key>
load_dotenv()

search = GoogleSerperAPIWrapper()

class SearchTool(BaseTool):
    name: str = "Search"
    description: str = "검색 기반 쿼리에 유용하다. 이 도구를 사용해 시장, 기업, 트렌드에 대한 최신 정보를 찾을 수 있다."
    search: GoogleSerperAPIWrapper = Field(default_factory=GoogleSerperAPIWrapper)

    def _run(self, query: str) -> str:
        """검색 쿼리를 실행하고 결과를 반환한다."""
        try:
            return self.search.run(query)
        except Exception as e:
            return f"검색 중 오류 발생: {str(e)}"

# 에이전트 생성
researcher = Agent(
    role='리서치 애널리스트',
    goal='최신 시장 데이터와 트렌드 수집',
    backstory="""시장 정보 수집 분야에서 수년간의 경험을 가진 전문 리서치 애널리스트다. 
    관련성 있고 최신의 시장 정보를 찾아내고 이를 명확하고 실행 가능한 형식으로 
    제공하는 능력으로 유명하다.""",
    tools=[SearchTool()],
    verbose=True
)

# 나머지 코드 ...