diff --git a/chat_history_stream.py b/chat_history_stream.py index 9bf96f0..6782514 100644 --- a/chat_history_stream.py +++ b/chat_history_stream.py @@ -41,6 +41,7 @@ model=MODEL_NAME, messages=messages, temperature=0.7, + stream=True, ) print("\nAnswer: ") diff --git a/spanish/structured_outputs_enum.py b/spanish/structured_outputs_enum.py index a60eac7..168073a 100644 --- a/spanish/structured_outputs_enum.py +++ b/spanish/structured_outputs_enum.py @@ -1,5 +1,5 @@ import os -from enum import Enum +from enum import StrEnum import azure.identity import openai @@ -34,7 +34,7 @@ MODEL_NAME = os.environ["OPENAI_MODEL"] -class DayOfWeek(str, Enum): +class DayOfWeek(StrEnum): DOMINGO = "Domingo" LUNES = "Lunes" MARTES = "Martes" diff --git a/structured_outputs_enum.py b/structured_outputs_enum.py index 39a99dc..2961235 100644 --- a/structured_outputs_enum.py +++ b/structured_outputs_enum.py @@ -1,5 +1,5 @@ import os -from enum import Enum +from enum import StrEnum import azure.identity import openai @@ -34,7 +34,7 @@ MODEL_NAME = os.environ["OPENAI_MODEL"] -class DayOfWeek(str, Enum): +class DayOfWeek(StrEnum): SUNDAY = "Sunday" MONDAY = "Monday" TUESDAY = "Tuesday"