| import os | |
| import getpass | |
| from dotenv import load_dotenv | |
| from langchain_groq import ChatGroq | |
| from langchain_community.tools import TavilySearchResults | |
| load_dotenv() | |
| def _set_env(var: str): | |
| if not os.environ.get(var): | |
| os.environ[var] = getpass.getpass(f"Enter {var}: ") | |
| # Setting up environmentl_variables | |
| _set_env("GROQ_API_KEY") | |
| _set_env("TAVILY_API_KEY") | |
| _set_env("JINA_API_KEY") | |
| # llm defined | |
| llm = ChatGroq(temperature=0.5, model_name="gemma2-9b-it") | |
| # search tool | |
| tavily_search = TavilySearchResults( | |
| max_results=2, | |
| search_depth="advanced", | |
| include_answer=True, | |
| include_raw_content=True, | |
| include_images=True, | |
| ) | |