ChronoStellar commited on
Commit
4e121cc
·
verified ·
1 Parent(s): cb977d7

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +8 -0
setup.py CHANGED
@@ -1,3 +1,11 @@
 
 
 
 
 
 
 
 
1
  # Model configuration
2
  from smolagents import OpenAIModel
3
  model = OpenAIModel(
 
1
+ import os
2
+
3
+ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
4
+
5
+ if not OPENAI_API_KEY:
6
+ raise RuntimeError("Missing OPENAI_API_KEY environment variable")
7
+
8
+
9
  # Model configuration
10
  from smolagents import OpenAIModel
11
  model = OpenAIModel(