|
|
from dotenv import load_dotenv |
|
|
from langchain.chat_models import init_chat_model |
|
|
from langgraph.prebuilt import create_react_agent |
|
|
from langchain_core.prompts import PromptTemplate |
|
|
from langchain_core.messages import SystemMessage |
|
|
from langchain_community.tools.tavily_search import TavilySearchResults |
|
|
from langchain.tools import tool |
|
|
|
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
model = init_chat_model("gemini-2.0-flash", model_provider="google_genai") |
|
|
|
|
|
@tool |
|
|
def search_craft_tutorials(query: str) -> str: |
|
|
"""Search the Internet for written craft tutorials based on the given query and return a few relevant results.""" |
|
|
search_tool = TavilySearchResults(k=5) |
|
|
return search_tool.run(query) |
|
|
|
|
|
|
|
|
mentor_prompt = PromptTemplate.from_template( |
|
|
""" |
|
|
You are Craft Mentor and your job is to help the user understand specific craft terminology and offer guidance on a given craft project. In addition, you can search for written tutorials on the |
|
|
Internet to help the user or give guidance and ideas. To search the Internet for written tutorials, use the search_craft_tutorials tool. |
|
|
Do not try to make up YouTube links, only use the search tool to look for written tutorials. |
|
|
""" |
|
|
) |
|
|
|
|
|
mentor_agent = create_react_agent( |
|
|
model=model, |
|
|
tools=[search_craft_tutorials], |
|
|
prompt=SystemMessage(content=mentor_prompt.format()), |
|
|
name="mentor_agent" |
|
|
) |
|
|
|
