vsrinivas commited on
Commit
27ea219
1 Parent(s): f0888c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -12,7 +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
  data_analyst_agent = Agent(
18
  role="Data Analyst",
@@ -175,7 +184,8 @@ financial_trading_crew = Crew(
175
  execution_planning_task,
176
  risk_assessment_task],
177
 
178
- manager_llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7),
 
179
  full_output =True,
180
  process=Process.hierarchical,
181
  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 HuggingFaceHub
18
+
19
+ llm = HuggingFaceHub(
20
+ repo_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
21
+ huggingfacehub_api_token="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",
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=llm, temperature=0.7),
189
  full_output =True,
190
  process=Process.hierarchical,
191
  verbose=True