From 2c58d58b337920994df0a95be2e5694eb477f254 Mon Sep 17 00:00:00 2001
From: Lucas Brito <lucas.brito@agi.com.br>
Date: Tue, 19 Nov 2024 22:15:16 -0300
Subject: [PATCH] refactor: update model name to gpt-4o-mini in LLM analysis

---
 src/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index 5a2e65c..707ba7a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -194,6 +194,7 @@ def ask_training_plan() -> str:
         "Was there anything planned for this training?"
     ).ask()
 
+
 def ask_desired_language() -> str:
     return questionary.text(
         "In which language do you want the analysis to be provided? (Default is Portuguese)",
@@ -232,7 +233,7 @@ def perform_llm_analysis(data: TCXReader, sport: str, plan: str, language: str)
 
     openai_llm = ChatOpenAI(
         openai_api_key=os.getenv("OPENAI_API_KEY"),
-        model_name="gpt-4o",
+        model_name="gpt-4o-mini",
         max_tokens=2000,
         temperature=0.6,
         max_retries=5