|
11 | 11 | dotenv.load_dotenv() |
12 | 12 |
|
13 | 13 | OPENAI_API_KEY = os.getenv('OPENAI_API_KEY') |
14 | | -if OPENAI_API_KEY is None: |
| 14 | +if not OPENAI_API_KEY: |
15 | 15 | raise KeyError("OPENAI_API_KEY is not set in environment variables") |
16 | 16 |
|
17 | 17 | # Get current folder |
@@ -124,6 +124,10 @@ def _get_llm_summary(formatted_events): |
124 | 124 | with open(os.path.join(examples_dir, "current_prompt.md"), 'w') as txt: |
125 | 125 | txt.write(prompt) # save for reference |
126 | 126 |
|
| 127 | + OPENAI_API_KEY = os.getenv('OPENAI_API_KEY') |
| 128 | + if not OPENAI_API_KEY: |
| 129 | + raise KeyError("OPENAI_API_KEY is not set in environment variables (summary func)") |
| 130 | + |
127 | 131 | print("First 10 characters of key", OPENAI_API_KEY[:10] if OPENAI_API_KEY else "None") |
128 | 132 | openai.api_key = OPENAI_API_KEY |
129 | 133 |
|
@@ -153,7 +157,7 @@ def _post_to_slack(message, slack_webhook_url): |
153 | 157 | response.raise_for_status() |
154 | 158 | logger.info("Message posted to Slack successfully") |
155 | 159 |
|
156 | | -def summarise_events_with_llm(events_file=EVENTS_FILE): |
| 160 | +def load_events_and_summarise(events_file=EVENTS_FILE): |
157 | 161 | try: |
158 | 162 | events = _load_events(events_file) |
159 | 163 | future_events = _filter_future_events(events) |
@@ -202,7 +206,7 @@ def summarise_events_with_llm(events_file=EVENTS_FILE): |
202 | 206 | raise KeyError("SLACK_BOT_TEST_WEBHOOK or SLACK_BOT_WEBHOOK must be set in environment variables") |
203 | 207 |
|
204 | 208 | try: |
205 | | - summary = summarise_events_with_llm(args.events_file) |
| 209 | + summary = load_events_and_summarise(args.events_file) |
206 | 210 | _post_to_slack(summary, slack_webhook_url=SLACK_WEBHOOK) |
207 | 211 | except Exception as e: |
208 | 212 | logger.error(f"Failed to summarize and post events: {e}", exc_info=True) |
|
0 commit comments