Api for writing task 2 v1.

helper/process_response.py

@@ -0,0 +1,58 @@
+import json
+import openai
+import os
+
+from dotenv import load_dotenv
+
+load_dotenv()
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+MAX_TOKENS = 4097
+TEMPERATURE = 0.1
+TOP_P = 0.9
+FREQUENCY_PENALTY = 0.5
+
+TRY_LIMIT = 1
+
+try_count = 0
+def process_response(input_string):
+    json_obj = {}
+    parsed_string = input_string.replace("'", "\"")
+    parsed_string = parsed_string.replace("\n\n", " ")
+    try:
+        json_obj = json.loads(parsed_string)
+    except json.JSONDecodeError:
+        print("Invalid JSON string!")
+
+    return json_obj
+
+
+def check_fields(obj):
+    if "overall" in obj and "task_response" in obj and "comment" in obj:
+        return True
+    else:
+        return False
+
+
+def make_openai_call(messages, token_count):
+    global try_count
+    result = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        max_tokens=int(MAX_TOKENS - token_count - 500),
+        temperature=float(TEMPERATURE),
+        top_p=float(TOP_P),
+        frequency_penalty=float(FREQUENCY_PENALTY),
+        messages=messages
+    )
+    processed_response = process_response(result["choices"][0]["message"]["content"])
+    if check_fields(processed_response) is False and try_count < TRY_LIMIT:
+        try_count = try_count + 1
+        return make_openai_call(messages, token_count)
+    elif try_count >= TRY_LIMIT:
+        try_count = 0
+        return result["choices"][0]["message"]["content"]
+    else:
+        try_count = 0
+        return processed_response
+
+