Spaces:
Sleeping
Sleeping
Commit
·
33c52e5
1
Parent(s):
efc0118
feat: get openAI keys from env var
Browse files- src/openai_agent.py +11 -3
src/openai_agent.py
CHANGED
@@ -35,10 +35,18 @@ class AgentResponse:
|
|
35 |
|
36 |
class Agent:
|
37 |
def __init__(self, prompt_folder: str = "prompts"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
self.client = OpenAI(
|
39 |
-
organization=
|
40 |
-
project=
|
41 |
-
api_key=
|
42 |
)
|
43 |
self.prompt_folder = prompt_folder
|
44 |
|
|
|
35 |
|
36 |
class Agent:
|
37 |
def __init__(self, prompt_folder: str = "prompts"):
|
38 |
+
organization = os.getenv("OPENAI_ORGANIZATION")
|
39 |
+
project = os.getenv("OPENAI_PROJECT")
|
40 |
+
api_key = os.getenv("OPENAI_API_KEY")
|
41 |
+
|
42 |
+
if not organization or not project or not api_key:
|
43 |
+
raise ValueError(
|
44 |
+
"Please set the OPENAI_ORGANIZATION, OPENAI_PROJECT, and OPENAI_API_KEY environment variables"
|
45 |
+
)
|
46 |
self.client = OpenAI(
|
47 |
+
organization=organization,
|
48 |
+
project=project,
|
49 |
+
api_key=api_key,
|
50 |
)
|
51 |
self.prompt_folder = prompt_folder
|
52 |
|