initial commit

GPT4Flow.py

@@ -0,0 +1,12 @@
+dependencies = [
+    {"url": "martinjosifoski/OpenAIChatAtomicFlow", "revision": "main"},
+]
+from flows import flow_verse
+flow_verse.sync_dependencies(dependencies)
+
+from flow_modules.martinjosifoski.OpenAIChatAtomicFlow import OpenAIChatAtomicFlow
+
+class GPT4Flow(OpenAIChatAtomicFlow):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+

GPT4Flow.yaml

@@ -0,0 +1,57 @@
+name: "GPT4Flow"
+description: "A General OpenAI GPT4 Chatbot implemented as an AtomicFlow"
+
+model_name: "gpt-4"
+generation_parameters:
+  n: 1
+  max_tokens: 3000
+  temperature: 0.3
+
+  model_kwargs:
+    top_p: 0.2
+    frequency_penalty: 0
+    presence_penalty: 0
+
+system_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: |2-
+    You are a helpful assistant.
+
+  input_variables: []
+  template_format: jinja2
+
+human_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: "{{query}}"
+  input_variables:
+    - "query"
+  template_format: jinja2
+
+query_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: |2-
+    {{query}}
+
+  input_variables:
+    - "query" # # "human_input"
+  template_format: jinja2
+
+keep_raw_response: False
+
+output_data_transformations:
+  - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
+    regex: '(.*)'
+    input_key: "raw_response"
+    output_key: "response"
+    strip: True
+    assert_unique: True
+    verbose: False
+
+
+input_keys:
+  - "query" # "human_input"
+
+output_keys:
+  - "response"
+
+

__init__.py

@@ -0,0 +1 @@
+from .GPT4Flow import GPT4Flow