Re-scope image validation method for Core ML adapter (#42)

Author: mattt

Sources/AnyLanguageModel/Models/CoreMLLanguageModel.swift

@@ -210,6 +210,31 @@
 
             return LanguageModelSession.ResponseStream(stream: stream)
         }
+
+        // MARK: - Image Validation
+
+        private func validateNoImageSegments(in session: LanguageModelSession) throws {
+            // Check for image segments in instructions
+            if let instructions = session.instructions {
+                for segment in instructions.segments {
+                    if case .image = segment {
+                        throw CoreMLLanguageModelError.unsupportedFeature
+                    }
+                }
+            }
+
+            // Check for image segments in the most recent prompt
+            for entry in session.transcript.reversed() {
+                if case .prompt(let p) = entry {
+                    for segment in p.segments {
+                        if case .image = segment {
+                            throw CoreMLLanguageModelError.unsupportedFeature
+                        }
+                    }
+                    break
+                }
+            }
+        }
     }
 
     /// Errors that can occur when working with Core ML language models.
@@ -242,31 +267,6 @@
         }
     }
 
-    // MARK: - Image Validation
-
-    private func validateNoImageSegments(in session: LanguageModelSession) throws {
-        // Check for image segments in instructions
-        if let instructions = session.instructions {
-            for segment in instructions.segments {
-                if case .image = segment {
-                    throw CoreMLLanguageModelError.unsupportedFeature
-                }
-            }
-        }
-
-        // Check for image segments in the most recent prompt
-        for entry in session.transcript.reversed() {
-            if case .prompt(let p) = entry {
-                for segment in p.segments {
-                    if case .image = segment {
-                        throw CoreMLLanguageModelError.unsupportedFeature
-                    }
-                }
-                break
-            }
-        }
-    }
-
     // MARK: -
 
     private func toGenerationConfig(_ options: GenerationOptions) -> GenerationConfig {