initial commit

gpt_trainer.py

@@ -0,0 +1,38 @@
+import json
+from openai import OpenAI
+
+# we can use gpt-4o-mini , gpt-4o or gpt-3.5-turbo . right now we use mini for price reason
+gpt_model = "gpt-4o-mini"
+openai_api_key = "apikey"
+
+fine_tune_modelname = "mensa-hsai"
+
+
+def train_gpt(data):
+    client = OpenAI(api_key=openai_api_key)
+    training_temp_filename = "training.jsonl"
+
+    # create files that contain training data in jsonl format
+    with open(training_temp_filename, 'w') as file:
+        for entry in data:
+            json.dump(entry, file)
+            file.write('\n')
+
+    # upload file training to gpt
+    training_file = client.files.create(
+        file=open(training_temp_filename, 'rb'), purpose='fine-tune')
+    print("Training file id:", training_file.id)
+
+    # create job to start training
+    response = client.fine_tuning.jobs.create(
+        training_file=training_file.id,
+        model=gpt_model,
+        suffix=fine_tune_modelname
+    )
+
+    print("GPT Fine Tune job id:", response.id)
+
+
+
+
+