Update app.py
Browse files
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 |
-
|
18 |
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
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 |
-
|
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
|