convert inputs to bf objects

Author: tswast

bigframes/bigquery/_operations/obj.py

@@ -27,7 +27,6 @@
     information, see the `launch stage descriptions
     <https://cloud.google.com/products?hl=en#product-launch-stages>`_.
 
-
 .. note::
 
     To provide feedback or request support for this feature, send an email to
@@ -38,11 +37,12 @@
 from __future__ import annotations
 
 import datetime
-from typing import Optional, Union
+from typing import Optional, Sequence, Union
 
 import numpy as np
 import pandas as pd
 
+from bigframes.core import convert
 from bigframes.core.logging import log_adapter
 import bigframes.core.utils as utils
 import bigframes.operations as ops
@@ -56,12 +56,13 @@ def fetch_metadata(
     """The OBJ.FETCH_METADATA function returns Cloud Storage metadata for a partially populated ObjectRef value.
 
     Args:
-        objectref (bigframes.series.Series):
+        objectref (bigframes.pandas.Series):
             A partially populated ObjectRef value, in which the uri and authorizer fields are populated and the details field isn't.
 
     Returns:
-        bigframes.series.Series: A fully populated ObjectRef value. The metadata is provided in the details field of the returned ObjectRef value.
+        bigframes.pandas.Series: A fully populated ObjectRef value. The metadata is provided in the details field of the returned ObjectRef value.
     """
+    objectref = convert.to_bf_series(objectref, default_index=None)
     return objectref._apply_unary_op(ops.obj_fetch_metadata_op)
 
 
@@ -74,7 +75,7 @@ def get_access_url(
     """The OBJ.GET_ACCESS_URL function returns JSON that contains reference information for the input ObjectRef value, and also access URLs that you can use to read or modify the Cloud Storage object.
 
     Args:
-        objectref (bigframes.series.Series):
+        objectref (bigframes.pandas.Series):
             An ObjectRef value that represents a Cloud Storage object.
         mode (str):
             A STRING value that identifies the type of URL that you want to be returned. The following values are supported:
@@ -84,8 +85,10 @@ def get_access_url(
             An optional INTERVAL value that specifies how long the generated access URLs remain valid. You can specify a value between 30 minutes and 6 hours. For example, you could specify INTERVAL 2 HOUR to generate URLs that expire after 2 hours. The default value is 6 hours.
 
     Returns:
-        bigframes.series.Series: A JSON value that contains the Cloud Storage object reference information from the input ObjectRef value, and also one or more URLs that you can use to access the Cloud Storage object.
+        bigframes.pandas.Series: A JSON value that contains the Cloud Storage object reference information from the input ObjectRef value, and also one or more URLs that you can use to access the Cloud Storage object.
     """
+    objectref = convert.to_bf_series(objectref, default_index=None)
+
     duration_micros = None
     if duration is not None:
         duration_micros = utils.timedelta_to_micros(duration)
@@ -97,23 +100,25 @@ def get_access_url(
 
 @log_adapter.method_logger(custom_base_name="bigquery_obj")
 def make_ref(
-    uri_or_json: series.Series,
-    authorizer: Optional[series.Series] = None,
+    uri_or_json: Union[series.Series, Sequence[str]],
+    authorizer: Union[series.Series, str, None] = None,
 ) -> series.Series:
     """Use the OBJ.MAKE_REF function to create an ObjectRef value that contains reference information for a Cloud Storage object.
 
     Args:
-        uri_or_json (bigframes.series.Series):
-            A STRING value that contains the URI for the Cloud Storage object, for example, gs://mybucket/flowers/12345.jpg.
+        uri_or_json (bigframes.pandas.Series or str):
+            A series of STRING values that contains the URI for the Cloud Storage object, for example, gs://mybucket/flowers/12345.jpg.
             OR
-            A JSON value that represents a Cloud Storage object.
-        authorizer (bigframes.series.Series, optional):
+            A series of JSON value that represents a Cloud Storage object.
+        authorizer (bigframes.pandas.Series or str, optional):
             A STRING value that contains the Cloud Resource connection used to access the Cloud Storage object.
-            Required if uri_or_json is a URI string.
+            Required if ``uri_or_json`` is a URI string.
 
     Returns:
-        bigframes.series.Series: An ObjectRef value.
+        bigframes.pandas.Series: An ObjectRef value.
     """
+    uri_or_json = convert.to_bf_series(uri_or_json, default_index=None)
+
     if authorizer is not None:
         return uri_or_json._apply_binary_op(authorizer, ops.obj_make_ref_op)