diff --git a/FirebaseAI/Sources/Types/Public/Tools/URLContext.swift b/FirebaseAI/Sources/Types/Internal/Tools/URLContext.swift
similarity index 93%
rename from FirebaseAI/Sources/Types/Public/Tools/URLContext.swift
rename to FirebaseAI/Sources/Types/Internal/Tools/URLContext.swift
index eba8a48fcba..2033bb940f1 100644
--- a/FirebaseAI/Sources/Types/Public/Tools/URLContext.swift
+++ b/FirebaseAI/Sources/Types/Internal/Tools/URLContext.swift
@@ -13,6 +13,6 @@
 // limitations under the License.
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
-public struct URLContext: Sendable, Encodable {
+struct URLContext: Sendable, Encodable {
   init() {}
 }
diff --git a/FirebaseAI/Sources/Types/Public/URLContextMetadata.swift b/FirebaseAI/Sources/Types/Public/URLContextMetadata.swift
index 3364def8963..d3c87887ab6 100644
--- a/FirebaseAI/Sources/Types/Public/URLContextMetadata.swift
+++ b/FirebaseAI/Sources/Types/Public/URLContextMetadata.swift
@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-/// Metadata related to the ``URLContext`` tool.
+/// Metadata related to the ``Tool/urlContext()`` tool.
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
 public struct URLContextMetadata: Sendable, Hashable {
   /// List of URL context.
diff --git a/FirebaseAI/Sources/Types/Public/URLMetadata.swift b/FirebaseAI/Sources/Types/Public/URLMetadata.swift
index 960541aebe1..1e42759b909 100644
--- a/FirebaseAI/Sources/Types/Public/URLMetadata.swift
+++ b/FirebaseAI/Sources/Types/Public/URLMetadata.swift
@@ -49,7 +49,7 @@ public struct URLMetadata: Sendable, Hashable {
       AILog.MessageCode.urlMetadataUnrecognizedURLRetrievalStatus
   }
 
-  /// The URL retrieved by the ``URLContext`` tool.
+  /// The URL retrieved by the ``Tool/urlContext()`` tool.
   public let retrievedURL: URL?
 
   /// The status of the URL retrieval.