martinjosifoski
commited on
Commit
•
c7096b5
1
Parent(s):
2106e4a
Remove api_keys from input and add a ToDo.
Browse files- 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
|