| import json |
| from datetime import datetime |
| import logging |
| from groq import Groq |
|
|
| |
| logger = logging.getLogger(__name__) |
|
|
| class WeatherAgent: |
| def __init__(self, groq_api_key): |
| logger.info("Initializing WeatherAgent") |
| self.groq_client = Groq(api_key=groq_api_key) |
| |
| def analyze_weather_data(self, forecast_data): |
| """Use Groq to analyze weather data and provide insights""" |
| if not forecast_data: |
| logger.warning("No forecast data provided for analysis") |
| return "I couldn't access the weather data at the moment." |
| |
| logger.info("Creating weather analysis prompt") |
| |
| prompt = f""" |
| As a Burning Man weather expert, analyze this 7-day forecast for Black Rock City and provide insights: |
| |
| Raw Weather Data: |
| {json.dumps(forecast_data, indent=2)} |
| |
| Please provide: |
| 1. A summary of the overall weather pattern |
| 2. Key highlights and potential concerns |
| 3. Specific recommendations for Burners |
| 4. Any notable weather changes or patterns |
| |
| Format your response in a friendly, Burning Man style. |
| """ |
| |
| try: |
| logger.info("Sending weather data to Groq for analysis") |
| response = self.groq_client.chat.completions.create( |
| messages=[ |
| { |
| "role": "system", |
| "content": "You are a weather expert for Burning Man, experienced in desert climate patterns and their implications for festival participants." |
| }, |
| { |
| "role": "user", |
| "content": prompt |
| } |
| ], |
| model="llama-3.1-8b-instant", |
| temperature=0.7, |
| ) |
| |
| logger.info("Successfully received AI analysis") |
| return response.choices[0].message.content |
| |
| except Exception as e: |
| logger.error(f"Error in AI analysis: {str(e)}") |
| return f"I'm having trouble analyzing the weather patterns right now. Basic forecast data is still available: {self.format_basic_forecast(forecast_data)}" |
| |
| def format_basic_forecast(self, forecast): |
| """Fallback formatter for basic weather data""" |
| logger.info("Using basic forecast formatter") |
| if not forecast: |
| logger.warning("No forecast data available for basic formatting") |
| return "Weather data unavailable at the moment." |
| |
| report = "Here's the basic weather forecast for Black Rock City:\n\n" |
| for day in forecast: |
| report += f"📅 {day['date']}: High {day['high']}°F, Low {day['low']}°F\n" |
| report += f" Conditions: {day['conditions']}, Wind: {day['wind_speed']} mph\n\n" |
| return report |
