Akshayram1 commited on
Commit
a9fba1e
·
verified ·
1 Parent(s): a7eaa97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -26,10 +26,11 @@ class QueryAnalysisOutput(BaseModel):
26
  if sambanova_key:
27
  llm = LLM(model="sambanova/DeepSeek-R1-Distill-Llama-70B", temperature=0.7, api_key=sambanova_key)
28
 
29
- # Define CrewAI Agents
30
  query_parser_agent = Agent(
31
  role="Stock Data Analyst",
32
  goal="Extract stock details from user query.",
 
33
  llm=llm,
34
  verbose=True,
35
  memory=True,
@@ -44,6 +45,7 @@ if sambanova_key:
44
  code_writer_agent = Agent(
45
  role="Senior Python Developer",
46
  goal="Write Python code to visualize stock data.",
 
47
  llm=llm,
48
  verbose=True,
49
  )
@@ -57,6 +59,7 @@ if sambanova_key:
57
  code_execution_agent = Agent(
58
  role="Code Execution Expert",
59
  goal="Execute the generated code to visualize stock data.",
 
60
  tools=[code_interpreter_tool],
61
  allow_code_execution=True,
62
  llm=llm,
 
26
  if sambanova_key:
27
  llm = LLM(model="sambanova/DeepSeek-R1-Distill-Llama-70B", temperature=0.7, api_key=sambanova_key)
28
 
29
+ # Define CrewAI Agents with backstory
30
  query_parser_agent = Agent(
31
  role="Stock Data Analyst",
32
  goal="Extract stock details from user query.",
33
+ backstory="An expert in analyzing stock data and trends.",
34
  llm=llm,
35
  verbose=True,
36
  memory=True,
 
45
  code_writer_agent = Agent(
46
  role="Senior Python Developer",
47
  goal="Write Python code to visualize stock data.",
48
+ backstory="A seasoned developer with expertise in financial data visualization.",
49
  llm=llm,
50
  verbose=True,
51
  )
 
59
  code_execution_agent = Agent(
60
  role="Code Execution Expert",
61
  goal="Execute the generated code to visualize stock data.",
62
+ backstory="Specializes in running and debugging Python scripts for data analysis.",
63
  tools=[code_interpreter_tool],
64
  allow_code_execution=True,
65
  llm=llm,