name: "OpenAIChatGPT4"
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


input_keys:
  - "query" # "human_input"

output_keys:
  - "response"