From adecb2a2d3bc5b5fb12280c54096706974e58201 Mon Sep 17 00:00:00 2001
From: Dalton Flanagan <6599399+dltn@users.noreply.github.com>
Date: Fri, 1 Nov 2024 14:36:50 -0400
Subject: [PATCH] update for message parsing on ios

---
 .../impls/ios/inference/LocalInferenceImpl/Parsing.swift      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/llama_stack/providers/impls/ios/inference/LocalInferenceImpl/Parsing.swift b/llama_stack/providers/impls/ios/inference/LocalInferenceImpl/Parsing.swift
index 89f24a561..84da42d1b 100644
--- a/llama_stack/providers/impls/ios/inference/LocalInferenceImpl/Parsing.swift
+++ b/llama_stack/providers/impls/ios/inference/LocalInferenceImpl/Parsing.swift
@@ -81,7 +81,9 @@ func encodeMessage(message: Components.Schemas.ChatCompletionRequest.messagesPay
     switch (m.content) {
     case .case1(let c):
       prompt += _processContent(c)
-    case .case2(let c):
+    case .ImageMedia(let c):
+      prompt += _processContent(c)
+    case .case3(let c):
       prompt += _processContent(c)
     }
   case .CompletionMessage(let m):