Market-Analysis-Agent/main.py at dev · TaskerJang/Market-Analysis-Agent · GitHub

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
from dotenv import load_dotenv

from dependency_injector.wiring import Provide, inject
import uvicorn

# from apscheduler.schedulers.background import BackgroundScheduler

from api.server import APIBuilder

# from src.graph.nodes.us_financial_fmg import StockInfoNode
from src.graph.nodes import (
    NaverNewsSearcherNode,
    ReportAssistantNode,
    ChosunRSSFeederNode,
    RetrieveESGNode,
    # WeeklyReporterNode,
    WSJEconomyRSSFeederNode,
    WSJMarketRSSFeederNode,
    USFinancialAnalyzerNode,
    # GoogleSearchAPINode,  # 삭제됨 - google_searcher로 대체
    GoogleSearcherNode,
    CompanyFactsAnalyzerNode,
    # StockInfoNode,
)
from src.utils.logger import setup_logger
from src.graph.builder import SupervisorGraphBuilder

# from src.tasks.weekly_recap_scraper import scrape_jp_weekly_recap
from startup import Container
from rich.console import Console

# import os


console = Console()
load_dotenv(override=True)
logger = setup_logger("market_agent")
logo = """
[cyan]
==============================================================================================

███████ ██ ███    ██  █████   ██████  ███████ ███    ██ ████████       ██       █████  ██████
██      ██ ████   ██ ██   ██ ██       ██      ████   ██    ██          ██      ██   ██ ██   ██
█████   ██ ██ ██  ██ ███████ ██   ███ █████   ██ ██  ██    ██    █████ ██      ███████ ██████
██      ██ ██  ██ ██ ██   ██ ██    ██ ██      ██  ██ ██    ██          ██      ██   ██ ██   ██
██      ██ ██   ████ ██   ██  ██████  ███████ ██   ████    ██          ███████ ██   ██ ██████

----------------------------------------------------------------------------------------------
                 __  __          _       _                    _         _
                |  \/  |__ _ _ _| |_____| |_   __ _ _ _  __ _| |_  _ __(_)___
                | |\/| / _` | '_| / / -_)  _| / _` | ' \/ _` | | || (_-< (_-<
                |_|  |_\__,_|_| |_\_\___|\__| \__,_|_||_\__,_|_|\_, /__/_/__/
                                                                |__/
----------------------------------------------------------------------------------------------
# MEMBER(가나다 순)
- 전병훈 (팀장)
- 강동석
- 백인걸
- 엄창용        https://github.com/e7217
- 왕수연
- 장현상
----------------------------------------------------------------------------------------------
                                                    Since 2025.03.04, Let's study together!
==============================================================================================
"""


@inject
def main(
    graph_builder: SupervisorGraphBuilder = Provide[Container.supervisor_graph],
):
    console.print(logo)
    logger.info("Starting Market Analysis Agent service...")

    ## 그래프 빌더
    """
    에이전트 노드를 이곳에 추가해주세요.
    노드 추가 시, 다음의 기능을 동적으로 적용하게됩니다.
    - supervisor 노드에 멤버로 등록
    - 노드 이름을 기반으로 API 엔드포인트 생성(예: SampleNode -> /api/sample)

    Example:
    graph_builder.add_node(NewNode())
    """

    graph_builder.add_node(NaverNewsSearcherNode())
    graph_builder.add_node(GoogleSearcherNode())    # 검색 기반 (웹 스크래핑)
    # graph_builder.add_node(GoogleSearchAPINode())  # 삭제됨 - API 기반, google_searcher로 대체
    graph_builder.add_node(RetrieveESGNode())
    graph_builder.add_node(ReportAssistantNode())
    # graph_builder.add_node(WeeklyReporterNode())
    graph_builder.add_node(ChosunRSSFeederNode())
    graph_builder.add_node(WSJEconomyRSSFeederNode())
    graph_builder.add_node(WSJMarketRSSFeederNode())
    # graph_builder.add_node(StockInfoNode())  # TODO: 종합 처리 기능 적용 시 주석 해제
    graph_builder.add_node(CompanyFactsAnalyzerNode())

    # 한투 API 분석 에이전트 노드 주석 처리 (미국 주식 노드로 대체)
    # graph_builder.add_node(HantooFinancialAnalyzerNode())

    # 미국 주식 분석 에이전트 노드 추가 (Alpha Vantage API 사용)
    graph_builder.add_node(USFinancialAnalyzerNode())  # TODO: 종합 처리 기능 적용 시 주석 해제

    # # 주간 리포트 스크래핑 스케쥴러
    # if os.getenv("PRODUCTION", "false").lower() == "true":
    #     # 주간 리캡 스크래핑 노드 추가
    #     vector_store = Container.vector_store_recap()
    #     graph_builder.add_node(WeeklyReporterNode(vector_store))

    #     # 주간 리캡 스크래핑 작업 예약
    #     scheduler = BackgroundScheduler(daemon=True)
    #     scheduler.add_job(
    #         scrape_jp_weekly_recap,
    #         "cron",
    #         hour="6,9,12,15,18",
    #         minute=0,
    #         args=[vector_store],
    #     )
    #     scheduler.start()

    graph_builder.build()

    ## API 서버 빌더
    api_builder = APIBuilder()
    app = api_builder.create_app()
    for node in graph_builder.get_nodes():
        app.add_api_route(
            f"/api/{node.__class__.__name__.lower().replace('node', '')}",
            methods=["POST"],
            endpoint=node.invoke,
        )

    uvicorn.run(app, host="0.0.0.0", port=8000)


if __name__ == "__main__":
    container = Container()
    container.wire(
        modules=[
            __name__,
            "api.route",
            # "src.tasks.weekly_recap_scraper"
        ]
    )
    main()