File size: 893 Bytes
a47e415
 
 
07097fe
a47e415
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from workflow.agents_workflow import build_agents_workflow
from workflow.graph_state import GraphState

# Run locally without gradio

app = build_agents_workflow(llm_model_name="gpt-4o-mini")

def run_user_query(ticker):
    QUERY = f"Produce investor-ready insights for {ticker}."
    init_state: GraphState = {
        "ticker": ticker,
        "query": QUERY,
        "news_summary": None,
        "earnings_summary": None,
        "market_summary": None,
        "completed": [],
        "final_recommendation": None,
    }
    final_state = app.invoke(init_state)

    return final_state

state = run_user_query("AAPL")

print("\n" + "=" * 80)
print(f"### NEWS SUMMARY\n{state['news_summary']}\n")
print(f"### EARNINGS SUMMARY\n{state['earnings_summary']}\n")
print(f"### MARKET SUMMARY\n{state['market_summary']}\n")
print(f"### FINAL RECOMMENDATION\n{state['final_recommendation']}\n")