vsrinivas commited on
Commit
70d8a6a
1 Parent(s): 2bcb2a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -12,16 +12,16 @@ os.environ["SERPER_API_KEY"] = os.getenv('SERPER_API_KEY')
12
 
13
  search_tool = SerperDevTool()
14
  scrape_tool = ScrapeWebsiteTool()
15
- llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7)
16
 
17
- # from langchain_community.llms import HuggingFaceEndpoint
18
 
19
- # llm = HuggingFaceEndpoint(
20
- # repo_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
21
- # huggingfacehub_api_token=os.getenv('HF_KEY'),
22
- # task="text-generation",
23
- # temperature=0.7,
24
- # )
25
 
26
  data_analyst_agent = Agent(
27
  role="Data Analyst",
@@ -184,8 +184,8 @@ financial_trading_crew = Crew(
184
  execution_planning_task,
185
  risk_assessment_task],
186
 
187
- manager_llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7),
188
- # manager_llm=ChatOpenAI(model="meta-llama/Meta-Llama-3.1-8B-Instruct", temperature=0.7),
189
  full_output =True,
190
  process=Process.hierarchical,
191
  verbose=True
 
12
 
13
  search_tool = SerperDevTool()
14
  scrape_tool = ScrapeWebsiteTool()
15
+ # llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7)
16
 
17
+ from langchain_community.llms import HuggingFaceEndpoint
18
 
19
+ llm = HuggingFaceEndpoint(
20
+ repo_id="mistralai/Mistral-7B-Instruct-v0.3",
21
+ huggingfacehub_api_token=os.getenv('HF_KEY'),
22
+ task="text-generation",
23
+ temperature=0.7,
24
+ )
25
 
26
  data_analyst_agent = Agent(
27
  role="Data Analyst",
 
184
  execution_planning_task,
185
  risk_assessment_task],
186
 
187
+ # manager_llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7),
188
+ manager_llm=ChatOpenAI(model="mistralai/Mistral-7B-Instruct-v0.3", temperature=0.7),
189
  full_output =True,
190
  process=Process.hierarchical,
191
  verbose=True