martinjosifoski commited on
Commit
c7096b5
1 Parent(s): 2106e4a

Remove api_keys from input and add a ToDo.

Browse files
Files changed (1) hide show
  1. OpenAIChatAtomicFlow.py +1 -4
OpenAIChatAtomicFlow.py CHANGED
@@ -232,9 +232,6 @@ class OpenAIChatAtomicFlow(AtomicFlow):
232
  input_data: Dict[str, Any],
233
  private_keys: Optional[List[str]] = [],
234
  keys_to_ignore_for_hash: Optional[List[str]] = []) -> Dict[str, Any]:
235
- self.api_keys = input_data["api_keys"]
236
- del input_data["api_keys"]
237
-
238
  # ~~~ Process input ~~~
239
  self._process_input(input_data)
240
 
@@ -245,4 +242,4 @@ class OpenAIChatAtomicFlow(AtomicFlow):
245
  content=response
246
  )
247
 
248
- return response
 
232
  input_data: Dict[str, Any],
233
  private_keys: Optional[List[str]] = [],
234
  keys_to_ignore_for_hash: Optional[List[str]] = []) -> Dict[str, Any]:
 
 
 
235
  # ~~~ Process input ~~~
236
  self._process_input(input_data)
237
 
 
242
  content=response
243
  )
244
 
245
+ return response # ToDo: Make this a dictionary {"api_output": response} and propagate the change to CC_Flows