from crewai_tools import SerperDevTool, WebsiteSearchTool, PDFSearchTool
import os 
from dotenv import load_dotenv
load_dotenv()

search_tool = SerperDevTool(
    serper_api_key=os.environ['SERPER_API_KEY']
)

website_search_tool = WebsiteSearchTool(
	config=dict(
		llm=dict(
			provider='google',
			config=dict(
				model="gemini/gemini-1.5-flash-002",
			),
		),
		embedder=dict(
			provider="google",
			config=dict(
				model="models/embedding-001/",
				task_type="retrieval_document"
			),
		),
	)
)

pdf_search_tool = PDFSearchTool(
	config=dict(
		llm=dict(
			provider='google',
			config=dict(
				model="gemini/gemini-1.5-flash-002",
			),
		),
		embedder=dict(
			provider="google",
			config=dict(
				model="models/embedding-001/",
				task_type="retrieval_document"
			),
		),
	)
)