Spaces:
Sleeping
Sleeping
File size: 798 Bytes
99c3854 67c90cc 99c3854 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
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"
),
),
)
) |