랭그래프 활용한 주식 정보 도출 프로젝트

- 투자 의견 조회
ㄴ 전반적인 주식 판단 기준(ex. 채무, 리스크 등등)을 기준으로 판단하여 도출
- 투자 추천
ex)
{
  "tickers": [
    "NVDA", "GOOGL", "AAPL"
  ],
  "risk_type": "aggressive"
}

project/STOCK_APP/backend/schemas/stock_schemas.py

@@ -1,9 +1,21 @@
 from pydantic import BaseModel
-
+from enum import Enum
 
 class StockAnalyzeReq(BaseModel):
     query: str
 
 class TickerInfo(BaseModel):
     ticker: str
-    company_name: str
+    company_name: str
+
+
+# enum
+# risk_type : aggressive, balanced, conservative
+class RiskType(str, Enum):
+    AGGRESSIVE = "aggressive"
+    BALANCED = "balanced"
+    CONSERVATIVE = "conservative"
+
+class RecommendStock(BaseModel):
+    tickers: list[str]
+    risk_type : RiskType