89 lines
2.1 KiB
Python
89 lines
2.1 KiB
Python
from fastapi import FastAPI, HTTPException, status
|
|
from pydantic import BaseModel
|
|
import uvicorn
|
|
from datetime import datetime
|
|
|
|
# Import your trading agents
|
|
from service import enqueue_analysis
|
|
from tradingagents.dataflows.config import get_config
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
config = get_config()
|
|
|
|
# Create FastAPI app instance
|
|
app = FastAPI(
|
|
title="TradingAgents API",
|
|
description="API for TradingAgents financial trading framework",
|
|
version="0.1.0"
|
|
)
|
|
|
|
class TradingAnalyzeRequest(BaseModel):
|
|
symbol: str
|
|
date: str
|
|
|
|
class TradingAnalyzeResponse(BaseModel):
|
|
symbol: str
|
|
date: str
|
|
job_id: str
|
|
timestamp: str
|
|
status: str
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
"""Root endpoint"""
|
|
return {"message": "Welcome to TradingAgents API"}
|
|
|
|
@app.get("/ping")
|
|
async def ping():
|
|
"""Simple ping endpoint that returns pong"""
|
|
return {"message": "pong"}
|
|
|
|
@app.get("/health")
|
|
async def health_check():
|
|
"""Health check endpoint"""
|
|
return {
|
|
"status": "healthy",
|
|
"timestamp": datetime.now().isoformat(),
|
|
"service": "tradingagents-api"
|
|
}
|
|
|
|
@app.post("/v1/trading/analyze", response_model=TradingAnalyzeResponse, status_code=status.HTTP_202_ACCEPTED,)
|
|
async def analyze_trading_decision(request: TradingAnalyzeRequest):
|
|
"""
|
|
Analyze trading decision for a given symbol and date
|
|
|
|
Example usage:
|
|
POST /trading/analyze
|
|
{
|
|
"symbol": "BTC/USDT",
|
|
"date": "2024-05-10"
|
|
}
|
|
"""
|
|
response = enqueue_analysis(request.symbol, request.date)
|
|
print(f"INFO: Enqueue response: {response}")
|
|
|
|
if response.status == "error":
|
|
raise HTTPException(status_code=500, detail=response.message)
|
|
|
|
return TradingAnalyzeResponse(
|
|
symbol=request.symbol,
|
|
date=request.date,
|
|
job_id=response.job_id,
|
|
timestamp=datetime.now().isoformat(),
|
|
status=response.status
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Run the server
|
|
uvicorn.run(
|
|
"webapp:app",
|
|
host=config.get("APP_HOST", "localhost"),
|
|
port=config.get("APP_PORT", 8000),
|
|
reload=True,
|
|
log_level="info"
|
|
)
|