Spaces:

Dharma20
/

UseCaseGenie-GenAI-Usecase-Generator

Sleeping

UseCaseGenie-GenAI-Usecase-Generator / setup.py

Update setup.py

7763c11 verified over 1 year ago

979 Bytes

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,
)

	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,
	)