Update src/llm_boilers.py

src/llm_boilers.py

@@ -56,7 +56,17 @@ def gpt_loader(model_id: str, openai_key: str):
     openai.api_key = openai_key  # os.getenv("OPENAI_API_KEY")
     logging.warning(f"model id: {model_id}")
 
-    return model_id
+    model = openai.ChatCompletion.create(
+        model=model_id,
+        messages=[],
+        temperature=0.0,
+        max_tokens=0,
+        n=1,
+        stop=None,
+        log_level="info",
+    )
+
+    return model
 
 
 LOAD_MODEL_FUNCTIONS.append(gpt_loader)