Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|