Spaces:
Sleeping
Sleeping
File size: 1,100 Bytes
6c7823c 72243d9 6c7823c 72243d9 |
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 |
from fastapi import APIRouter
from pydantic import BaseModel
from .market_data import price_change, earning_summary, portfolio_data, get_update
from .vectroDB import get_relevant_chunks
app = APIRouter()
class HistoricalData(BaseModel):
symbol: str
period: int
class EarningReq(BaseModel):
symbol: str
class PortfolioReq(BaseModel):
region: str
class KnowledgeReq(BaseModel):
query: str
@app.post("/get_historical_data")
def get_historical_data(req: HistoricalData):
symbol = req.symbol
period = req.period
return {"response": price_change(symbol, period)}
@app.post("/get_earning_metrics")
def get_eraning_metrics(req: EarningReq):
return {"response": earning_summary(req.symbol)}
@app.post("/get_portfolio_data")
def get_portfolio_data(req: PortfolioReq):
return {"response": portfolio_data(req.region)}
@app.post("/get_knowledge")
def get_knowledge(req: KnowledgeReq):
return {"response": get_relevant_chunks(req.query)}
@app.post("/get_update")
def get_ticker_update(req: EarningReq):
return {"response": get_update(req.symbol)}
|