leo-bourrel commited on
Commit
33c52e5
·
1 Parent(s): efc0118

feat: get openAI keys from env var

Browse files
Files changed (1) hide show
  1. 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="org-37xz2Z39bbKBClpp77UEz2fk",
40
- project="proj_TckSEWLOGDT0TM7qaMnD3zCA",
41
- api_key="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",
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