refactor: Dockerfile

Author: lpm0073

Dockerfile

@@ -1,12 +1,14 @@
 # Use the official Python image from the Docker Hub.
 # This runs on Debian Linux.
-FROM python:3.13-slim-trixie
+FROM python:3.13-slim-trixie AS base
+
+
+FROM base AS requirements
 
 # Set the working directory to /app
 WORKDIR /dist
 
 # Copy the current directory contents into the container at /app
-COPY app /dist/app
 COPY requirements/prod.txt requirements.txt
 
 # Set environment variables
@@ -16,6 +18,12 @@ ENV PYTHONPATH=/dist
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
+FROM requirements AS app
+
+WORKDIR /dist
+COPY app /dist/app
+
+FROM app AS runtime
 
 # Run the application when the container launches
 CMD ["python", "-m", "app.agent"]

app/agent.py

@@ -39,17 +39,27 @@ def main(prompts: Optional[Tuple[str, ...]] = None) -> None:
             followup_question = None
 
         if "get_courses" in functions_called:
-            user_prompt = (
-                prompts[i]
-                if prompts and len(prompts) >= i
-                else input(followup_question or "Would you like to register for a course? ")
-            )
+            default_prompt = "Would you like to register for a course? "
         elif "register_course" in functions_called:
-            user_prompt = (
-                prompts[i]
-                if prompts and len(prompts) >= i
-                else input(followup_question or "Can I help you with anything else? ")
-            )
+            default_prompt = "Can I help you with anything else? "
+        else:
+            default_prompt = "Please let me know: "
+
+        user_prompt = prompts[i] if prompts and len(prompts) > i else input(followup_question or default_prompt)
+
+        if user_prompt and user_prompt.lower().strip() in [
+            "no",
+            "no thanks",
+            "nothing",
+            "exit",
+            "quit",
+            "bye",
+            "goodbye",
+            "that's all",
+            "nothing else",
+        ]:
+            print("Thank you for using Stackademy! Goodbye!")
+            break
 
         response, functions_called = completion(prompt=user_prompt)
 

app/prompt.py

@@ -168,13 +168,16 @@ def handle_completion(tools, tool_choice) -> ChatCompletion:
     functions_called = []
 
     response = handle_completion(
-        tool_choice={"type": "function", "function": {"name": "get_courses"}},
+        # tool_choice={"type": "function", "function": {"name": "get_courses"}},
+        tool_choice="required",
         tools=[stackademy_app.tool_factory_get_courses()],
     )
     logger.debug("Initial response: %s", response.model_dump())
 
     message = response.choices[0].message
     while message.tool_calls:
+        if message.content and "Goodbye!" in message.content:
+            break
         functions_called = process_tool_calls(message)
 
         response = handle_completion(

app/stackademy.py

@@ -66,23 +66,27 @@ def _log_success(self, message: str) -> None:
 
     def tool_factory_get_courses(self) -> ChatCompletionFunctionToolParam:
         """LLM Factory function to create a tool for getting courses"""
+        schema = StackademyGetCoursesParams.model_json_schema()
+        schema["required"] = []  # Both parameters are optional
         return ChatCompletionFunctionToolParam(
             type="function",
             function={
                 "name": "get_courses",
                 "description": "returns up to 10 rows of course detail data, filtered by the maximum cost a student is willing to pay for a course and the area of specialization.",
-                "parameters": StackademyGetCoursesParams.model_json_schema(),
+                "parameters": schema,
             },
         )
 
     def tool_factory_register(self) -> ChatCompletionFunctionToolParam:
         """LLMFactory function to create a tool for registering a user"""
+        schema = StackademyRegisterCourseParams.model_json_schema()
+        schema["required"] = ["course_code", "email", "full_name"]  # All parameters are required
         return ChatCompletionFunctionToolParam(
             type="function",
             function={
                 "name": "register_course",
                 "description": "Register a student in a course with the provided details.",
-                "parameters": StackademyRegisterCourseParams.model_json_schema(),
+                "parameters": schema,
             },
         )