change the dependency revision to main

.gitignore

@@ -158,4 +158,8 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
-.*cache*/
+.*cache*/
+
+
+# auto-generated by flows, all synced modules will be ignored by default
+FLOW_MODULE_ID

OpenAIChatGPT4.py

@@ -1,8 +1,10 @@
-import os
+dependencies = [
+    {"url": "martinjosifoski/OpenAIChatAtomicFlow", "revision": "main"},
+]
+from flows import flow_verse
+flow_verse.sync_dependencies(dependencies) 
 
-from flows.flow_verse import load_class
-OpenAIChatAtomicFlow_repository_id = os.environ.get("OpenAIChatAtomicFlow")
-OpenAIChatAtomicFlow = load_class(OpenAIChatAtomicFlow_repository_id, "OpenAIChatAtomicFlow")
+from flow_modules.martinjosifoski.OpenAIChatAtomicFlow import OpenAIChatAtomicFlow
 
 class OpenAIChatGPT4(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):

OpenAIChatGPT4.yaml

@@ -37,12 +37,13 @@ query_message_prompt_template:
     - "query" # # "human_input"
   template_format: jinja2
 
+keep_raw_response: False
 
-response_annotators:
-  code_extractor:
-    _target_: flows.message_annotators.RegexFirstOccurrenceExtractor
+output_data_transformations:
+  - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
     regex: '(.*)'
-    key: "response"
+    input_key: "raw_response"
+    output_key: "response"
     strip: True
     assert_unique: True
     verbose: True