File size: 979 Bytes
7763c11 |
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 |
import os
import getpass
from dotenv import load_dotenv
from langchain_groq import ChatGroq
# from langchain_cohere import ChatCohere
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")
# llm defined
llm1 = ChatGroq(temperature=0.5, model_name="llama-3.1-8b-instant", api_key='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')
llm2 = ChatGroq(temperature=0.7, model_name="qwen-qwq-32b", api_key='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')
# llm3 = ChatCohere(model='command-a-03-2025', cohere_api_key='AjWeplJ8UcvyKktAT5lSVbSnTGxZu0l2scBExWVN')
# search tool
tavily_search = TavilySearchResults(
max_results=2,
search_depth="advanced",
include_answer=True,
include_raw_content=True,
include_images=True,
)
|