Dharma20 commited on
Commit
7763c11
·
verified ·
1 Parent(s): 0c14628

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +33 -33
setup.py CHANGED
@@ -1,33 +1,33 @@
1
- import os
2
- import getpass
3
- from dotenv import load_dotenv
4
- from langchain_groq import ChatGroq
5
- from langchain_cohere import ChatCohere
6
- from langchain_community.tools import TavilySearchResults
7
-
8
- load_dotenv()
9
-
10
- def _set_env(var: str):
11
- if not os.environ.get(var):
12
- os.environ[var] = getpass.getpass(f"Enter {var}: ")
13
-
14
-
15
- # Setting up environmentl_variables
16
- # _set_env("GROQ_API_KEY")
17
- _set_env("TAVILY_API_KEY")
18
-
19
-
20
-
21
- # llm defined
22
- llm1 = ChatGroq(temperature=0.5, model_name="llama-3.1-8b-instant", api_key='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')
23
- llm2 = ChatGroq(temperature=0.7, model_name="qwen-qwq-32b", api_key='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')
24
- # llm3 = ChatCohere(model='command-a-03-2025', cohere_api_key='AjWeplJ8UcvyKktAT5lSVbSnTGxZu0l2scBExWVN')
25
-
26
- # search tool
27
- tavily_search = TavilySearchResults(
28
- max_results=2,
29
- search_depth="advanced",
30
- include_answer=True,
31
- include_raw_content=True,
32
- include_images=True,
33
- )
 
1
+ import os
2
+ import getpass
3
+ from dotenv import load_dotenv
4
+ from langchain_groq import ChatGroq
5
+ # from langchain_cohere import ChatCohere
6
+ from langchain_community.tools import TavilySearchResults
7
+
8
+ load_dotenv()
9
+
10
+ def _set_env(var: str):
11
+ if not os.environ.get(var):
12
+ os.environ[var] = getpass.getpass(f"Enter {var}: ")
13
+
14
+
15
+ # Setting up environmentl_variables
16
+ # _set_env("GROQ_API_KEY")
17
+ _set_env("TAVILY_API_KEY")
18
+
19
+
20
+
21
+ # llm defined
22
+ llm1 = ChatGroq(temperature=0.5, model_name="llama-3.1-8b-instant", api_key='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')
23
+ llm2 = ChatGroq(temperature=0.7, model_name="qwen-qwq-32b", api_key='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')
24
+ # llm3 = ChatCohere(model='command-a-03-2025', cohere_api_key='AjWeplJ8UcvyKktAT5lSVbSnTGxZu0l2scBExWVN')
25
+
26
+ # search tool
27
+ tavily_search = TavilySearchResults(
28
+ max_results=2,
29
+ search_depth="advanced",
30
+ include_answer=True,
31
+ include_raw_content=True,
32
+ include_images=True,
33
+ )