vsrinivas commited on
Commit
2bcb2a9
1 Parent(s): 2101e5c

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="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
  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