| from pydantic import BaseModel, Field |
| from typing import List, Dict, Optional, Any |
| from datetime import datetime |
|
|
| |
| class AnalysisRequest(BaseModel): |
| subreddits: List[str] = Field(..., description="List of subreddit names") |
| posts_per_subreddit: int = Field(10, ge=1, le=50, description="Number of posts per subreddit") |
| time_filter: str = Field("day", description="Time filter (hour, day, week, month, year, all)") |
|
|
| class AnalysisResponse(BaseModel): |
| status: str |
| message: str |
| insights: List[Dict[str, Any]] = [] |
|
|
| |
| class InsightResponse(BaseModel): |
| id: str |
| subreddit: str |
| post_title: str |
| post_url: str |
| timestamp: datetime |
| model: str |
| insight: Dict[str, Any] |
|
|
| |
| class QueryRequest(BaseModel): |
| query: str = Field(..., description="The user's question") |
| n_results: int = Field(5, ge=1, le=20, description="Number of sources to retrieve") |
|
|
| class QueryResponse(BaseModel): |
| answer: str |
| sources: List[Dict[str, Any]] |
|
|
| |
| class MarketIntelligenceRequest(BaseModel): |
| idea: str = Field(..., description="Startup idea to analyze") |
|
|
| class MarketIntelligenceResponse(BaseModel): |
| interest_score: int |
| competitors: List[str] |
| funding_news: List[Dict[str, str]] |
|
|
| |
| class GeneratePlanRequest(BaseModel): |
| insight: Dict[str, Any] = Field(..., description="The insight object to generate a plan for") |
|
|
| class GeneratePlanResponse(BaseModel): |
| plan: str |
|
|
| |
| class TrendRequest(BaseModel): |
| subreddit: Optional[str] = Field(None, description="Filter by subreddit (optional)") |
|
|
| class TrendResponse(BaseModel): |
| dates: List[str] |
| counts: List[int] |
|
|
| |
| class UserSignup(BaseModel): |
| email: str |
| password: str |
| full_name: Optional[str] = None |
|
|
| class UserLogin(BaseModel): |
| email: str |
| password: str |
|
|
| class UserProfile(BaseModel): |
| id: str |
| email: str |
| full_name: Optional[str] |
| subscription_tier: str |
| subscription_status: str |
| created_at: datetime |
|
|
| |
| class SubscriptionPlan(BaseModel): |
| id: str |
| name: str |
| price: float |
| features: Dict[str, Any] |
