cooney
315105cebb
- 투자 의견 조회
ㄴ 전반적인 주식 판단 기준(ex. 채무, 리스크 등등)을 기준으로 판단하여 도출
- 투자 추천
ex)
{
"tickers": [
"NVDA", "GOOGL", "AAPL"
],
"risk_type": "aggressive"
}
21 lines
427 B
Python
21 lines
427 B
Python
from pydantic import BaseModel
|
|
from enum import Enum
|
|
|
|
class StockAnalyzeReq(BaseModel):
|
|
query: str
|
|
|
|
class TickerInfo(BaseModel):
|
|
ticker: 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 |