dynamic model definition

agents/dummy_agent.py

@@ -14,11 +14,11 @@ class DummyState(TypedDict):
 
 
 class DummyAgent:
-    def __init__(self, system_prompt_path: str = "prompts/system_prompt.txt"):
+    def __init__(self, system_prompt_path: str = "prompts/system_prompt.txt", model: str = "gpt-4o"):
         print("LangGraph Chatbot Agent initialized.")
         self.system_prompt = self.read_system_prompt(system_prompt_path)
         self.llm = ChatOpenAI(
-            model="gpt-3.5-turbo",
+            model=model,
             temperature=0
         )
         graph = StateGraph(DummyState)

app.py

@@ -10,7 +10,10 @@ from agents.dummy_agent import DummyAgent
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # >>>>>>>>>>>>>>>>>> Keep Constants as is <<<<<<<<<<<<<<<<<<<< #
 
-agent = DummyAgent(system_prompt_path="prompts/default_system_prompt.txt")
+agent = DummyAgent(
+    system_prompt_path="prompts/default_system_prompt.txt",
+    model="gpt-4o",
+    )
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """