vsrinivas commited on
Commit
c4ebc39
1 Parent(s): 66b46ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -18
app.py CHANGED
@@ -17,27 +17,20 @@ scrape_tool = ScrapeWebsiteTool()
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
- llm_1 = HuggingFaceEndpoint(
27
- repo_id="meta-llama/Meta-Llama-3.1-8B",
28
- huggingfacehub_api_token=os.getenv('HF_KEY'),
29
- task="text-generation",
30
- temperature=0.7,
31
- )
32
 
33
- manager = Agent(
34
- role="Project Manager",
35
- goal="Efficiently manage the crew and ensure high-quality task completion",
36
- backstory="You're an experienced project manager, specialized in financial investments & trading, skilled in overseeing complex projects and guiding teams to success. Your role is to coordinate the efforts of the crew members, ensuring that each task is completed on time and to the highest standard.",
37
- allow_delegation=True,
38
- tools = [scrape_tool, search_tool],
39
- llm=llm_1
40
- )
41
 
42
 
43
  data_analyst_agent = Agent(
@@ -202,10 +195,10 @@ financial_trading_crew = Crew(
202
  risk_assessment_task],
203
 
204
  # manager_llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7),
205
- # manager_llm=ChatOpenAI(model="mistralai/Mistral-7B-Instruct-v0.3", temperature=0.7),
206
- manager_agent=manager,
207
  full_output =True,
208
- # process=Process.hierarchical,
209
  verbose=True
210
  )
211
 
 
17
  from langchain_community.llms import HuggingFaceEndpoint
18
 
19
  llm = HuggingFaceEndpoint(
20
+ repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
21
  huggingfacehub_api_token=os.getenv('HF_KEY'),
22
  task="text-generation",
23
  temperature=0.7,
24
  )
25
 
 
 
 
 
 
 
26
 
27
+ # manager = Agent(
28
+ # role="Project Manager",
29
+ # goal="Efficiently manage the crew and ensure high-quality task completion",
30
+ # backstory="You're an experienced project manager, specialized in financial investments & trading, skilled in overseeing complex projects and guiding teams to success. Your role is to coordinate the efforts of the crew members, ensuring that each task is completed on time and to the highest standard.",
31
+ # allow_delegation=True,
32
+ # tools = [scrape_tool, search_tool],
33
+ # )
 
34
 
35
 
36
  data_analyst_agent = Agent(
 
195
  risk_assessment_task],
196
 
197
  # manager_llm=ChatOpenAI(model="gpt-4o-mini", openai_api_key =os.getenv('OPENAI_API_KEY'), temperature=0.7),
198
+ manager_llm=ChatOpenAI(model="mistralai/Mixtral-8x7B-Instruct-v0.1", temperature=0.7),
199
+ # manager_agent=manager,
200
  full_output =True,
201
+ process=Process.hierarchical,
202
  verbose=True
203
  )
204