Fix imports

Author: andrewheard

FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift

@@ -17,7 +17,7 @@ import FirebaseAuthInterop
 import FirebaseCore
 import XCTest
 
-@testable import FirebaseVertexAI
+@testable public import FirebaseVertexAI
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
 final class GenerativeModelTests: XCTestCase {
@@ -1690,8 +1690,7 @@ struct AppCheckErrorFake: Error {}
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
 extension SafetyRating: Swift.Comparable {
-  public static func < (lhs: FirebaseVertexAI.SafetyRating,
-                        rhs: FirebaseVertexAI.SafetyRating) -> Bool {
+  public static func < (lhs: SafetyRating, rhs: SafetyRating) -> Bool {
     return lhs.category.rawValue < rhs.category.rawValue
   }
 }

FirebaseVertexAI/Tests/Unit/VertexComponentTests.swift

@@ -13,11 +13,10 @@
 // limitations under the License.
 
 @preconcurrency import FirebaseCore
+internal import FirebaseCoreExtension
 import Foundation
 import XCTest
 
-@_implementationOnly import FirebaseCoreExtension
-
 @testable import FirebaseVertexAI
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)