Update dependencies to llama-stack version 0.5.2 and add llama-stack-api; adjust constants and tests accordingly

Updated `test.containerfile` to rhoai-3.4

Update base image in test.containerfile to use upstream Red Hat UBI

fixed type error

addressed comments
- updated from 0.5.0 -> 0.5.2

fixed mypy

Author: jrobertboos

pyproject.toml

@@ -28,9 +28,9 @@ dependencies = [
     # Used by authentication/k8s integration
     "kubernetes>=30.1.0",
     # Used to call Llama Stack APIs
-    "llama-stack==0.4.3",
-    "llama-stack-client==0.4.3",
-    "llama-stack-api==0.4.4",
+    "llama-stack==0.5.2",
+    "llama-stack-client==0.5.2",
+    "llama-stack-api==0.5.2",
     # Used by Logger
     "rich>=14.0.0",
     # Used by JWK token auth handler

src/constants.py

@@ -2,7 +2,7 @@
 
 # Minimal and maximal supported Llama Stack version
 MINIMAL_SUPPORTED_LLAMA_STACK_VERSION = "0.2.17"
-MAXIMAL_SUPPORTED_LLAMA_STACK_VERSION = "0.4.3"
+MAXIMAL_SUPPORTED_LLAMA_STACK_VERSION = "0.5.2"
 
 UNABLE_TO_PROCESS_RESPONSE = "Unable to process this request"
 

src/utils/responses.py

@@ -1139,15 +1139,16 @@ def _extract_text_from_content(
 
     text_fragments: list[str] = []
     for part in content:
-        if part.type == "input_text":
+        part_type = getattr(part, "type", None)
+        if part_type == "input_text":
             input_text_part = cast(InputTextPart, part)
             if input_text_part.text:
                 text_fragments.append(input_text_part.text.strip())
-        elif part.type == "output_text":
+        elif part_type == "output_text":
             output_text_part = cast(OutputTextPart, part)
             if output_text_part.text:
                 text_fragments.append(output_text_part.text.strip())
-        elif part.type == "refusal":
+        elif part_type == "refusal":
             refusal_part = cast(ContentPartRefusal, part)
             if refusal_part.refusal:
                 text_fragments.append(refusal_part.refusal.strip())

test.containerfile

@@ -1,4 +1,3 @@
-# Upstream llama-stack built from Red Hat UBI
 FROM registry.access.redhat.com/ubi9/ubi-minimal
 
 USER root

tests/e2e/features/info.feature

@@ -16,7 +16,7 @@ Feature: Info tests
      When I access REST API endpoint "info" using HTTP GET method
      Then The status code of the response is 200
       And The body of the response has proper name Lightspeed Core Service (LCS) and version 0.4.2
-      And The body of the response has llama-stack version 0.4.3
+      And The body of the response has llama-stack version 0.5.2
 
   @skip-in-library-mode
   Scenario: Check if info endpoint reports error when llama-stack connection is not working