openaivec 0.14.2__py3-none-any.whl → 0.14.4__py3-none-any.whl

openaivec/spark.py

@@ -237,6 +237,7 @@ def responses_udf(
     temperature: float | None = 0.0,
     top_p: float = 1.0,
     max_concurrency: int = 8,
+    **api_kwargs,
 ) -> UserDefinedFunction:
     """Create an asynchronous Spark pandas UDF for generating responses.
 
@@ -276,6 +277,11 @@ def responses_udf(
             Higher values increase throughput but may hit OpenAI rate limits.
             Recommended: 4-12 per executor. Defaults to 8.
 
+    Additional Keyword Args:
+        Arbitrary OpenAI Responses API parameters (e.g. ``frequency_penalty``, ``presence_penalty``,
+        ``seed``, ``max_output_tokens``, etc.) are forwarded verbatim to the underlying API calls.
+        These parameters are applied to all API requests made by the UDF.
+
     Returns:
         UserDefinedFunction: A Spark pandas UDF configured to generate responses asynchronously.
             Output schema is `StringType` or a struct derived from `response_format`.
@@ -313,6 +319,7 @@ def responses_udf(
                             temperature=temperature,
                             top_p=top_p,
                             cache=cache,
+                            **api_kwargs,
                         )
                     )
                     yield pd.DataFrame(predictions.map(_safe_dump).tolist())
@@ -340,6 +347,7 @@ def responses_udf(
                             temperature=temperature,
                             top_p=top_p,
                             cache=cache,
+                            **api_kwargs,
                         )
                     )
                     yield predictions.map(_safe_cast_str)
@@ -357,6 +365,7 @@ def task_udf(
     model_name: str = "gpt-4.1-mini",
     batch_size: int | None = None,
     max_concurrency: int = 8,
+    **api_kwargs,
 ) -> UserDefinedFunction:
     """Create an asynchronous Spark pandas UDF from a predefined task.
 
@@ -381,6 +390,12 @@ def task_udf(
             Higher values increase throughput but may hit OpenAI rate limits.
             Recommended: 4-12 per executor. Defaults to 8.
 
+    Additional Keyword Args:
+        Arbitrary OpenAI Responses API parameters (e.g. ``frequency_penalty``, ``presence_penalty``,
+        ``seed``, ``max_output_tokens``, etc.) are forwarded verbatim to the underlying API calls.
+        These parameters are applied to all API requests made by the UDF and override any
+        parameters set in the task configuration.
+
     Returns:
         UserDefinedFunction: A Spark pandas UDF configured to execute the specified task
             asynchronously with automatic caching for duplicate inputs within each partition.
@@ -429,6 +444,7 @@ def task_udf(
                             temperature=task_temperature,
                             top_p=task_top_p,
                             cache=cache,
+                            **api_kwargs,
                         )
                     )
                     yield pd.DataFrame(predictions.map(_safe_dump).tolist())
@@ -456,6 +472,7 @@ def task_udf(
                             temperature=task_temperature,
                             top_p=task_top_p,
                             cache=cache,
+                            **api_kwargs,
                         )
                     )
                     yield predictions.map(_safe_cast_str)
@@ -594,7 +611,10 @@ def similarity_udf() -> UserDefinedFunction:
             Cosine similarity between the two vectors.
         """
         # Import pandas_ext to ensure .ai accessor is available in Spark workers
-        from openaivec import pandas_ext  # noqa: F401
+        from openaivec import pandas_ext
+
+        # Explicitly reference pandas_ext to satisfy linters
+        assert pandas_ext is not None
 
         return pd.DataFrame({"a": a, "b": b}).ai.similarity("a", "b")
 

{openaivec-0.14.2.dist-info → openaivec-0.14.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openaivec
-Version: 0.14.2
+Version: 0.14.4
 Summary: Generative mutation for tabular calculation
 Project-URL: Homepage, https://microsoft.github.io/openaivec/
 Project-URL: Repository, https://github.com/microsoft/openaivec

{openaivec-0.14.2.dist-info → openaivec-0.14.4.dist-info}/RECORD

@@ -6,12 +6,13 @@ openaivec/_model.py,sha256=xg3s9Ljqb2xK1t_a5bwWxGJfFSIuaNrFGMgQq4nQKrM,3351
 openaivec/_optimize.py,sha256=-mKjD5YV_d1Z2nqfGfAcmx6mTKn6AODjFTrIKJPbAXQ,3851
 openaivec/_prompt.py,sha256=KoJbFK4gTEDRtu9OMweJq_jQLkSPFy2Kcvao30qKhAQ,20844
 openaivec/_provider.py,sha256=dNr9Y2C97GK-pkY81odurKoDup59dLK31V3EGT2HOwE,6711
-openaivec/_proxy.py,sha256=giOxRlCCO11XQ0gNVf2IksjZZj9RwvTHkHbmbQXadEk,28916
-openaivec/_responses.py,sha256=SSa52mCYh2jF52pHaXRtj0tJQUEn52Uy8ypMT57vSks,21428
+openaivec/_proxy.py,sha256=J0qGDcZqSab26ScA8OXxzornfwuXtrVycqup-JPq464,29719
+openaivec/_responses.py,sha256=xtkiOn01RkauHq2FAKRAcjPglH8rmbaSz0-VE0ClTe8,24026
+openaivec/_schema.py,sha256=9enwqE2idLLUKbQxjiNn09uhdKz14kihEwUXglRqxx0,20543
 openaivec/_serialize.py,sha256=NLCKl4opc1WS24_duwpI2UGBepQ8SBh4YRxBlLwzDLw,8403
 openaivec/_util.py,sha256=dFWwjouJyvF-tqNPs2933OAt5Fw9I2Q2BvmGIfGH5k4,6423
-openaivec/pandas_ext.py,sha256=xOZvuJN6qPQZZmaHKr870kelG2XmuFHkqeAbFNayUZU,58825
-openaivec/spark.py,sha256=MqzS8nZe6R3wT_QJdopYSMZWDpagN6gyJNh-CAtvHA8,24403
+openaivec/pandas_ext.py,sha256=m4H6mrE__Jmr5R6hl6d8yc2JhVT0-wdf5GOKWIITeLU,63366
+openaivec/spark.py,sha256=lI-noacLvuxu6gBztKdcYd9vfK3eNI3aCGwJylkzv7E,25367
 openaivec/task/__init__.py,sha256=lrgoc9UIox7XnxZ96dQRl88a-8QfuZRFBHshxctpMB8,6178
 openaivec/task/customer_support/__init__.py,sha256=KWfGyXPdZyfGdRH17x7hPpJJ1N2EP9PPhZx0fvBAwSI,884
 openaivec/task/customer_support/customer_sentiment.py,sha256=r_NJEz11zdMCw6x8S2jqEhcFZSJDn0Plgf0ED8JlvxQ,7618
@@ -29,7 +30,7 @@ openaivec/task/nlp/sentiment_analysis.py,sha256=BNwWtNT-MNA76eIJbb31641upukmRwM9
 openaivec/task/nlp/translation.py,sha256=XTZM11JFjbgTK9wHnxFgVDabXZ5bqbabXK_bq2nEkyQ,6627
 openaivec/task/table/__init__.py,sha256=kJz15WDJXjyC7UIHKBvlTRhCf347PCDMH5T5fONV2sU,83
 openaivec/task/table/fillna.py,sha256=ZVcOpuh7ULVhrt1VsWy5fPhk53XNaiD7kXGCPhh83M8,6636
-openaivec-0.14.2.dist-info/METADATA,sha256=LfzZLjiWdjJASIdXqIb0Tniou0ZYPYuwjsGtka9sIYg,27566
-openaivec-0.14.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openaivec-0.14.2.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
-openaivec-0.14.2.dist-info/RECORD,,
+openaivec-0.14.4.dist-info/METADATA,sha256=RF6rZDL5B4qYCqXIbC0jexv-IzHv48WBDV-MZtNHcvY,27566
+openaivec-0.14.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+openaivec-0.14.4.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+openaivec-0.14.4.dist-info/RECORD,,