Adam-Ben-Khalifa commited on
Commit
48d7c91
·
verified ·
1 Parent(s): a8597aa

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -18,7 +18,7 @@ client = AsyncOpenAI(
18
 
19
  # Defining the PromptEnhancer class containing the necessary components for the Advanced Prompt Generation Pipeline
20
  class PromptEnhancer:
21
- def __init__(self, model="gpt-4o-mini", tools_dict={}):
22
  self.model = model
23
  self.prompt_tokens = 0
24
  self.completion_tokens = 0
 
18
 
19
  # Defining the PromptEnhancer class containing the necessary components for the Advanced Prompt Generation Pipeline
20
  class PromptEnhancer:
21
+ def __init__(self, model="gpt-4o-mini", temperature=0.0, tools_dict={}):
22
  self.model = model
23
  self.prompt_tokens = 0
24
  self.completion_tokens = 0