330 lines
14 KiB
Python
330 lines
14 KiB
Python
"""
|
|
Prompt template for strategy analysis.
|
|
"""
|
|
from typing import List
|
|
from models.input_models import EnrichedTelemetryWebhook, RaceContext, Strategy
|
|
from utils.validators import TelemetryAnalyzer
|
|
from config import get_settings
|
|
|
|
|
|
def build_analyze_prompt_fast(
|
|
enriched_telemetry: List[EnrichedTelemetryWebhook],
|
|
race_context: RaceContext,
|
|
strategies: List[Strategy]
|
|
) -> str:
|
|
"""Build a faster, more concise analyze prompt."""
|
|
latest = max(enriched_telemetry, key=lambda x: x.lap)
|
|
tire_rate = TelemetryAnalyzer.calculate_tire_degradation_rate(enriched_telemetry)
|
|
tire_cliff = TelemetryAnalyzer.project_tire_cliff(enriched_telemetry, race_context.race_info.current_lap)
|
|
|
|
strategies_summary = [f"#{s.strategy_id}: {s.strategy_name} ({s.stop_count}-stop, laps {s.pit_laps}, {s.tire_sequence}, {s.risk_level})" for s in strategies[:20]]
|
|
|
|
return f"""Analyze {len(strategies)} strategies and select TOP 3 for {race_context.driver_state.driver_name} at {race_context.race_info.track_name}.
|
|
|
|
CURRENT: Lap {race_context.race_info.current_lap}/{race_context.race_info.total_laps}, P{race_context.driver_state.current_position}
|
|
TELEMETRY: Tire deg {latest.tire_degradation_index:.2f} (cliff lap {tire_cliff}), Aero {latest.aero_efficiency:.2f}, Fuel {latest.fuel_optimization_score:.2f}, Driver {latest.driver_consistency:.2f}
|
|
|
|
STRATEGIES:
|
|
{chr(10).join(strategies_summary)}
|
|
|
|
Select TOP 3:
|
|
1. RECOMMENDED (highest podium %)
|
|
2. ALTERNATIVE (viable backup)
|
|
3. CONSERVATIVE (safest)
|
|
|
|
Return JSON in this EXACT format:
|
|
{{
|
|
"top_strategies": [
|
|
{{
|
|
"rank": 1,
|
|
"strategy_id": 7,
|
|
"strategy_name": "Strategy Name",
|
|
"classification": "RECOMMENDED",
|
|
"predicted_outcome": {{
|
|
"finish_position_most_likely": 3,
|
|
"p1_probability": 10,
|
|
"p2_probability": 25,
|
|
"p3_probability": 40,
|
|
"p4_or_worse_probability": 25,
|
|
"confidence_score": 75
|
|
}},
|
|
"risk_assessment": {{
|
|
"risk_level": "medium",
|
|
"key_risks": ["Risk 1", "Risk 2"],
|
|
"success_factors": ["Factor 1", "Factor 2"]
|
|
}},
|
|
"telemetry_insights": {{
|
|
"tire_wear_projection": "Tire analysis based on {latest.tire_degradation_index:.2f}",
|
|
"aero_status": "Aero at {latest.aero_efficiency:.2f}",
|
|
"fuel_margin": "Fuel at {latest.fuel_optimization_score:.2f}",
|
|
"driver_form": "Driver at {latest.driver_consistency:.2f}"
|
|
}},
|
|
"engineer_brief": {{
|
|
"title": "Brief title",
|
|
"summary": "One sentence",
|
|
"key_points": ["Point 1", "Point 2"],
|
|
"execution_steps": ["Step 1", "Step 2"]
|
|
}},
|
|
"driver_audio_script": "Radio message to driver",
|
|
"ecu_commands": {{
|
|
"fuel_mode": "RICH",
|
|
"ers_strategy": "AGGRESSIVE_DEPLOY",
|
|
"engine_mode": "PUSH",
|
|
"brake_balance_adjustment": 0,
|
|
"differential_setting": "BALANCED"
|
|
}}
|
|
}},
|
|
{{
|
|
"rank": 2,
|
|
"strategy_id": 12,
|
|
"strategy_name": "Alternative",
|
|
"classification": "ALTERNATIVE",
|
|
"predicted_outcome": {{"finish_position_most_likely": 4, "p1_probability": 5, "p2_probability": 20, "p3_probability": 35, "p4_or_worse_probability": 40, "confidence_score": 70}},
|
|
"risk_assessment": {{"risk_level": "medium", "key_risks": ["Risk 1"], "success_factors": ["Factor 1"]}},
|
|
"telemetry_insights": {{"tire_wear_projection": "...", "aero_status": "...", "fuel_margin": "...", "driver_form": "..."}},
|
|
"engineer_brief": {{"title": "...", "summary": "...", "key_points": ["..."], "execution_steps": ["..."]}},
|
|
"driver_audio_script": "...",
|
|
"ecu_commands": {{"fuel_mode": "STANDARD", "ers_strategy": "BALANCED", "engine_mode": "STANDARD", "brake_balance_adjustment": 0, "differential_setting": "BALANCED"}}
|
|
}},
|
|
{{
|
|
"rank": 3,
|
|
"strategy_id": 3,
|
|
"strategy_name": "Conservative",
|
|
"classification": "CONSERVATIVE",
|
|
"predicted_outcome": {{"finish_position_most_likely": 5, "p1_probability": 2, "p2_probability": 15, "p3_probability": 28, "p4_or_worse_probability": 55, "confidence_score": 80}},
|
|
"risk_assessment": {{"risk_level": "low", "key_risks": ["Risk 1"], "success_factors": ["Factor 1", "Factor 2"]}},
|
|
"telemetry_insights": {{"tire_wear_projection": "...", "aero_status": "...", "fuel_margin": "...", "driver_form": "..."}},
|
|
"engineer_brief": {{"title": "...", "summary": "...", "key_points": ["..."], "execution_steps": ["..."]}},
|
|
"driver_audio_script": "...",
|
|
"ecu_commands": {{"fuel_mode": "LEAN", "ers_strategy": "CONSERVATIVE", "engine_mode": "SAVE", "brake_balance_adjustment": 0, "differential_setting": "CONSERVATIVE"}}
|
|
}}
|
|
],
|
|
"situational_context": {{
|
|
"critical_decision_point": "Key decision info",
|
|
"telemetry_alert": "Important telemetry status",
|
|
"key_assumption": "Main assumption",
|
|
"time_sensitivity": "Timing requirement"
|
|
}}
|
|
}}"""
|
|
|
|
|
|
def build_analyze_prompt(
|
|
enriched_telemetry: List[EnrichedTelemetryWebhook],
|
|
race_context: RaceContext,
|
|
strategies: List[Strategy]
|
|
) -> str:
|
|
"""
|
|
Build the analyze prompt for Gemini.
|
|
|
|
Args:
|
|
enriched_telemetry: Recent enriched telemetry data
|
|
race_context: Current race context
|
|
strategies: Strategies to analyze
|
|
|
|
Returns:
|
|
Formatted prompt string
|
|
"""
|
|
# Generate telemetry summary
|
|
telemetry_summary = TelemetryAnalyzer.generate_telemetry_summary(enriched_telemetry)
|
|
|
|
# Calculate key metrics
|
|
tire_rate = TelemetryAnalyzer.calculate_tire_degradation_rate(enriched_telemetry)
|
|
tire_cliff_lap = TelemetryAnalyzer.project_tire_cliff(
|
|
enriched_telemetry,
|
|
race_context.race_info.current_lap
|
|
)
|
|
aero_avg = TelemetryAnalyzer.calculate_aero_efficiency_avg(enriched_telemetry)
|
|
ers_pattern = TelemetryAnalyzer.analyze_ers_pattern(enriched_telemetry)
|
|
fuel_critical = TelemetryAnalyzer.is_fuel_critical(enriched_telemetry)
|
|
driver_form = TelemetryAnalyzer.assess_driver_form(enriched_telemetry)
|
|
|
|
# Get latest telemetry
|
|
latest = max(enriched_telemetry, key=lambda x: x.lap)
|
|
|
|
# Format strategies for prompt
|
|
strategies_data = []
|
|
for s in strategies:
|
|
strategies_data.append({
|
|
"strategy_id": s.strategy_id,
|
|
"strategy_name": s.strategy_name,
|
|
"stop_count": s.stop_count,
|
|
"pit_laps": s.pit_laps,
|
|
"tire_sequence": s.tire_sequence,
|
|
"brief_description": s.brief_description,
|
|
"risk_level": s.risk_level,
|
|
"key_assumption": s.key_assumption
|
|
})
|
|
|
|
# Format competitors
|
|
competitors_data = []
|
|
for c in race_context.competitors:
|
|
competitors_data.append({
|
|
"position": c.position,
|
|
"driver": c.driver,
|
|
"tire_compound": c.tire_compound,
|
|
"tire_age_laps": c.tire_age_laps,
|
|
"gap_seconds": round(c.gap_seconds, 1)
|
|
})
|
|
|
|
prompt = f"""You are Stratega, expert F1 Chief Strategist AI. Analyze the 20 proposed strategies and select the TOP 3.
|
|
|
|
CURRENT RACE STATE:
|
|
Track: {race_context.race_info.track_name}
|
|
Current Lap: {race_context.race_info.current_lap} / {race_context.race_info.total_laps}
|
|
Weather: {race_context.race_info.weather_condition}
|
|
|
|
DRIVER STATE:
|
|
Driver: {race_context.driver_state.driver_name}
|
|
Position: P{race_context.driver_state.current_position}
|
|
Current Tires: {race_context.driver_state.current_tire_compound} ({race_context.driver_state.tire_age_laps} laps old)
|
|
Fuel Remaining: {race_context.driver_state.fuel_remaining_percent}%
|
|
|
|
COMPETITORS:
|
|
{competitors_data}
|
|
|
|
TELEMETRY ANALYSIS:
|
|
{telemetry_summary}
|
|
|
|
KEY METRICS:
|
|
- Current tire degradation index: {latest.tire_degradation_index:.3f}
|
|
- Tire degradation rate: {tire_rate:.3f} per lap
|
|
- Projected tire cliff: Lap {tire_cliff_lap}
|
|
- Aero efficiency: {aero_avg:.3f} average
|
|
- ERS pattern: {ers_pattern}
|
|
- Fuel critical: {'YES' if fuel_critical else 'NO'}
|
|
- Driver form: {driver_form}
|
|
|
|
PROPOSED STRATEGIES ({len(strategies_data)} total):
|
|
{strategies_data}
|
|
|
|
ANALYSIS FRAMEWORK:
|
|
|
|
1. TIRE DEGRADATION PROJECTION:
|
|
- Current tire_degradation_index: {latest.tire_degradation_index:.3f}
|
|
- Rate of change: {tire_rate:.3f} per lap
|
|
- Performance cliff (0.85): Projected lap {tire_cliff_lap}
|
|
- Strategies pitting before cliff = higher probability
|
|
|
|
2. AERO EFFICIENCY IMPACT:
|
|
- Current aero_efficiency: {aero_avg:.3f}
|
|
- If <0.7: Lap times degrading, prioritize earlier stops
|
|
- If >0.8: Car performing well, can extend stints
|
|
|
|
3. FUEL MANAGEMENT:
|
|
- Fuel optimization score: {latest.fuel_optimization_score:.3f}
|
|
- Fuel critical: {'YES - Must save fuel' if fuel_critical else 'NO - Can push'}
|
|
- Remaining: {race_context.driver_state.fuel_remaining_percent}%
|
|
|
|
4. DRIVER CONSISTENCY:
|
|
- Driver consistency: {latest.driver_consistency:.3f}
|
|
- Form: {driver_form}
|
|
- If <0.75: Higher margin for error needed, prefer conservative
|
|
- If >0.9: Can execute aggressive/risky strategies
|
|
|
|
5. WEATHER & TRACK POSITION:
|
|
- Weather impact: {latest.weather_impact}
|
|
- Track: {race_context.race_info.track_name}
|
|
- Overtaking difficulty consideration
|
|
|
|
6. COMPETITOR ANALYSIS:
|
|
- Current position: P{race_context.driver_state.current_position}
|
|
- Our tire age: {race_context.driver_state.tire_age_laps} laps
|
|
- Compare with competitors for undercut/overcut opportunities
|
|
|
|
SELECTION CRITERIA:
|
|
- Rank 1 (RECOMMENDED): Highest probability of podium (P1-P3), balanced risk
|
|
- Rank 2 (ALTERNATIVE): Different approach, viable if conditions change
|
|
- Rank 3 (CONSERVATIVE): Safest option, minimize risk of finishing outside points
|
|
|
|
OUTPUT FORMAT (JSON only, no markdown):
|
|
{{
|
|
"top_strategies": [
|
|
{{
|
|
"rank": 1,
|
|
"strategy_id": 7,
|
|
"strategy_name": "Aggressive Undercut",
|
|
"classification": "RECOMMENDED",
|
|
"predicted_outcome": {{
|
|
"finish_position_most_likely": 3,
|
|
"p1_probability": 8,
|
|
"p2_probability": 22,
|
|
"p3_probability": 45,
|
|
"p4_or_worse_probability": 25,
|
|
"confidence_score": 78
|
|
}},
|
|
"risk_assessment": {{
|
|
"risk_level": "medium",
|
|
"key_risks": [
|
|
"Requires pit stop under 2.5s",
|
|
"Traffic on out-lap could cost 3-5s"
|
|
],
|
|
"success_factors": [
|
|
"Tire degradation index trending at {tire_rate:.3f} per lap",
|
|
"Window open for undercut"
|
|
]
|
|
}},
|
|
"telemetry_insights": {{
|
|
"tire_wear_projection": "Current tire_degradation_index {latest.tire_degradation_index:.3f}, will hit 0.85 cliff by lap {tire_cliff_lap}",
|
|
"aero_status": "aero_efficiency {aero_avg:.3f} - car performing {'well' if aero_avg > 0.8 else 'adequately' if aero_avg > 0.7 else 'poorly'}",
|
|
"fuel_margin": "fuel_optimization_score {latest.fuel_optimization_score:.3f} - {'excellent, no fuel saving needed' if latest.fuel_optimization_score > 0.85 else 'adequate' if latest.fuel_optimization_score > 0.7 else 'critical, fuel saving required'}",
|
|
"driver_form": "driver_consistency {latest.driver_consistency:.3f} - {driver_form} confidence in execution"
|
|
}},
|
|
"engineer_brief": {{
|
|
"title": "Recommended: Strategy Name",
|
|
"summary": "One sentence summary with win probability",
|
|
"key_points": [
|
|
"Tire degradation accelerating: {latest.tire_degradation_index:.3f} index now, cliff projected lap {tire_cliff_lap}",
|
|
"Key tactical consideration",
|
|
"Performance advantage analysis",
|
|
"Critical execution requirement"
|
|
],
|
|
"execution_steps": [
|
|
"Lap X: Action 1",
|
|
"Lap Y: Action 2",
|
|
"Lap Z: Expected outcome"
|
|
]
|
|
}},
|
|
"driver_audio_script": "Clear radio message to driver about the strategy execution",
|
|
"ecu_commands": {{
|
|
"fuel_mode": "RICH",
|
|
"ers_strategy": "AGGRESSIVE_DEPLOY",
|
|
"engine_mode": "PUSH",
|
|
"brake_balance_adjustment": 0,
|
|
"differential_setting": "BALANCED"
|
|
}}
|
|
}},
|
|
{{
|
|
"rank": 2,
|
|
"strategy_id": 12,
|
|
"strategy_name": "Alternative Strategy",
|
|
"classification": "ALTERNATIVE",
|
|
"predicted_outcome": {{ "finish_position_most_likely": 4, "p1_probability": 5, "p2_probability": 18, "p3_probability": 38, "p4_or_worse_probability": 39, "confidence_score": 72 }},
|
|
"risk_assessment": {{ "risk_level": "medium", "key_risks": ["Risk 1", "Risk 2"], "success_factors": ["Factor 1", "Factor 2"] }},
|
|
"telemetry_insights": {{ "tire_wear_projection": "...", "aero_status": "...", "fuel_margin": "...", "driver_form": "..." }},
|
|
"engineer_brief": {{ "title": "...", "summary": "...", "key_points": ["..."], "execution_steps": ["..."] }},
|
|
"driver_audio_script": "...",
|
|
"ecu_commands": {{ "fuel_mode": "STANDARD", "ers_strategy": "BALANCED", "engine_mode": "STANDARD", "brake_balance_adjustment": 0, "differential_setting": "BALANCED" }}
|
|
}},
|
|
{{
|
|
"rank": 3,
|
|
"strategy_id": 3,
|
|
"strategy_name": "Conservative Strategy",
|
|
"classification": "CONSERVATIVE",
|
|
"predicted_outcome": {{ "finish_position_most_likely": 5, "p1_probability": 2, "p2_probability": 10, "p3_probability": 25, "p4_or_worse_probability": 63, "confidence_score": 85 }},
|
|
"risk_assessment": {{ "risk_level": "low", "key_risks": ["Risk 1"], "success_factors": ["Factor 1", "Factor 2", "Factor 3"] }},
|
|
"telemetry_insights": {{ "tire_wear_projection": "...", "aero_status": "...", "fuel_margin": "...", "driver_form": "..." }},
|
|
"engineer_brief": {{ "title": "...", "summary": "...", "key_points": ["..."], "execution_steps": ["..."] }},
|
|
"driver_audio_script": "...",
|
|
"ecu_commands": {{ "fuel_mode": "STANDARD", "ers_strategy": "CONSERVATIVE", "engine_mode": "SAVE", "brake_balance_adjustment": 0, "differential_setting": "CONSERVATIVE" }}
|
|
}}
|
|
],
|
|
"situational_context": {{
|
|
"critical_decision_point": "Next 3 laps crucial. Tire degradation index rising faster than expected.",
|
|
"telemetry_alert": "aero_efficiency status and any concerns",
|
|
"key_assumption": "Analysis assumes no safety car. If SC deploys, recommend boxing immediately.",
|
|
"time_sensitivity": "Decision needed within 2 laps to execute strategy effectively."
|
|
}}
|
|
}}"""
|
|
|
|
return prompt
|