Skip to content

Commit

Permalink
refactor: update model name to gpt-4o-mini in LLM analysis
Browse files Browse the repository at this point in the history
  • Loading branch information
Lucas Brito committed Nov 20, 2024
1 parent 5dafb9d commit 2c58d58
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)",
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 2c58d58

Please sign in to comment.