openaivec 0.13.6__py3-none-any.whl → 0.13.7__py3-none-any.whl

openaivec/__init__.py

@@ -1,9 +1,14 @@
 from .embeddings import AsyncBatchEmbeddings, BatchEmbeddings
+from .model import PreparedTask
+from .prompt import FewShotPrompt, FewShotPromptBuilder
 from .responses import AsyncBatchResponses, BatchResponses
 
 __all__ = [
-    "BatchResponses",
+    "AsyncBatchEmbeddings",
     "AsyncBatchResponses",
     "BatchEmbeddings",
-    "AsyncBatchEmbeddings",
+    "BatchResponses",
+    "FewShotPrompt",
+    "FewShotPromptBuilder",
+    "PreparedTask",
 ]

openaivec/di.py

@@ -2,6 +2,8 @@ from dataclasses import dataclass, field
 from threading import RLock
 from typing import Any, Callable, Dict, Set, Type, TypeVar
 
+__all__ = []
+
 """Simple dependency injection container with singleton lifecycle management.
 
 This module provides a lightweight dependency injection container that manages

openaivec/log.py

@@ -5,7 +5,7 @@ import uuid
 from logging import Logger
 from typing import Callable
 
-__all__ = ["observe"]
+__all__ = []
 
 
 def observe(logger: Logger):

openaivec/model.py

@@ -1,6 +1,10 @@
 from dataclasses import dataclass
 from typing import Generic, Type, TypeVar
 
+__all__ = [
+    "PreparedTask",
+]
+
 ResponseFormat = TypeVar("ResponseFormat")
 
 

openaivec/optimize.py

@@ -5,6 +5,8 @@ from dataclasses import dataclass, field
 from datetime import datetime, timezone
 from typing import List
 
+__all__ = []
+
 
 @dataclass(frozen=True)
 class PerformanceMetric:
@@ -20,7 +22,7 @@ class BatchSizeSuggester:
     min_batch_size: int = 10
     min_duration: float = 30.0
     max_duration: float = 60.0
-    step_ratio: float = 0.1
+    step_ratio: float = 0.2
     sample_size: int = 4
     _history: List[PerformanceMetric] = field(default_factory=list)
     _lock: threading.RLock = field(default_factory=threading.RLock, repr=False)

openaivec/pandas_ext.py

@@ -48,6 +48,13 @@ import numpy as np
 import pandas as pd
 import tiktoken
 from openai import AsyncOpenAI, OpenAI
+
+__all__ = [
+    "embeddings_model",
+    "responses_model",
+    "use",
+    "use_async",
+]
 from pydantic import BaseModel
 
 from openaivec.embeddings import AsyncBatchEmbeddings, BatchEmbeddings

openaivec/provider.py

@@ -15,6 +15,8 @@ from openaivec.model import (
 )
 from openaivec.util import TextChunker
 
+__all__ = []
+
 CONTAINER = di.Container()
 
 

openaivec/proxy.py

@@ -6,6 +6,8 @@ from typing import Any, Awaitable, Callable, Dict, Generic, List, TypeVar
 
 from openaivec.optimize import BatchSizeSuggester
 
+__all__ = []
+
 S = TypeVar("S", bound=Hashable)
 T = TypeVar("T")
 

openaivec/serialize.py

@@ -29,7 +29,7 @@ from typing import Any, Dict, List, Literal, Type
 
 from pydantic import BaseModel, Field, create_model
 
-__all__ = ["deserialize_base_model", "serialize_base_model"]
+__all__ = []
 
 
 def serialize_base_model(obj: Type[BaseModel]) -> Dict[str, Any]:

openaivec/spark.py

@@ -12,6 +12,15 @@ improved performance in I/O-bound operations.
 automatically cache duplicate inputs within each partition, significantly reducing
 API calls and costs when processing datasets with overlapping content.
 
+__all__ = [
+    "count_tokens_udf",
+    "embeddings_udf",
+    "responses_udf",
+    "similarity_udf",
+    "split_to_chunks_udf",
+    "task_udf",
+]
+
 ## Setup
 
 First, obtain a Spark session and configure authentication:

openaivec/util.py

@@ -8,6 +8,8 @@ from typing import Awaitable, Callable, List, Type, TypeVar
 import numpy as np
 import tiktoken
 
+__all__ = []
+
 T = TypeVar("T")
 U = TypeVar("U")
 V = TypeVar("V")

{openaivec-0.13.6.dist-info → openaivec-0.13.7.dist-info}/METADATA

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

{openaivec-0.13.6.dist-info → openaivec-0.13.7.dist-info}/RECORD

@@ -1,17 +1,17 @@
-openaivec/__init__.py,sha256=DQcfTw7y4CgPtyyMg-wO4chwVYT0jA_F-EjxT_bXxTg,236
-openaivec/di.py,sha256=eNewaSRx7f_O0IQcyjzGpIMak6O-bc6OeMqjytCfr88,10634
+openaivec/__init__.py,sha256=2nPIsylGgJcBXgrKrGzuThgw0Ycajlvf-bSzSCtg_ng,393
+openaivec/di.py,sha256=DsVeD-IJuAF3HS2OtHtQJ0BtWXyqzZeeNi7LDMWO3W0,10648
 openaivec/embeddings.py,sha256=ntnsdV2L6WAT4nKqgL3MZfhqf_Xnfu_J8oFavjIVcIU,7470
-openaivec/log.py,sha256=GofgzUpv_xDVuGC-gYmit5Oyu06it1SBXRck6COR5go,1439
-openaivec/model.py,sha256=u1zQLF__Qw9mp92K1LYaAaQEYisMZoMWuX9-dx2JYoE,3316
-openaivec/optimize.py,sha256=jjE-_ZOs_BPuSHHYVoykGHqTF04nlYkz_rFZ0Kcwtdc,3837
-openaivec/pandas_ext.py,sha256=QlTSZkVzVwO1MFAGv_MuFaINpXlxYjkbfO0zCsmbmSM,58717
+openaivec/log.py,sha256=1qhc9CF4D4bwiF_VWHilcYBPcTqIKyI0zuNEfn0MLNA,1430
+openaivec/model.py,sha256=xg3s9Ljqb2xK1t_a5bwWxGJfFSIuaNrFGMgQq4nQKrM,3351
+openaivec/optimize.py,sha256=-mKjD5YV_d1Z2nqfGfAcmx6mTKn6AODjFTrIKJPbAXQ,3851
+openaivec/pandas_ext.py,sha256=bvAG_2Uf39iNOAn4lSRXO63LoaMEI1J6t8kRSy_LK40,58807
 openaivec/prompt.py,sha256=ZzHOS2CRLRtPeP4Z734xkaAA5LFcfmhkDQE5URQUidQ,20842
-openaivec/provider.py,sha256=BYsJxioaemDpjOYdNuLMtasi54Je6LNrtyYupHTxWLA,6688
-openaivec/proxy.py,sha256=jsWUqEJVlRHj0854GlvAzG09mm-QmFFNREXeMOc1LsY,28901
+openaivec/provider.py,sha256=sEPf3g9Lhq66PaoEkpJdOVd9yG_PTCW3BU_qYuvZFjQ,6702
+openaivec/proxy.py,sha256=1evFOBifsn4Ecf-3GcFeQjt_dHlSV0iC6gxs-QjMCHk,28915
 openaivec/responses.py,sha256=v4MtyIh_g6bahmkq1Ra4y63Tlqg6-Ace1SErmoosvMg,21424
-openaivec/serialize.py,sha256=HXi5l_4b_4eUMNL7f7o_suyHHF_hz3RYsUsri5CQ7_4,7325
-openaivec/spark.py,sha256=uxXXgOhH1g_HEnBIuNGLlKPInJwZBCLaqrTnJT2UHk4,24252
-openaivec/util.py,sha256=A7-QSoGkdllH6e6CVHxuGk-fmAmsSCSZVSJ5VSDzmx4,6409
+openaivec/serialize.py,sha256=mSuSVfSpXMnsFZ4JjlOEe5b22ttBUHviPs9mF99lf0A,7277
+openaivec/spark.py,sha256=sBD4fiX-4ejf_m0SCWgv_e-aGptCwyuQ4FfvnxlT09g,24399
+openaivec/util.py,sha256=dFWwjouJyvF-tqNPs2933OAt5Fw9I2Q2BvmGIfGH5k4,6423
 openaivec/task/__init__.py,sha256=26Kd2eRRfUqRJnYH83sRrbuEuRwshypRRSkTWtT50Tw,6177
 openaivec/task/customer_support/__init__.py,sha256=KWfGyXPdZyfGdRH17x7hPpJJ1N2EP9PPhZx0fvBAwSI,884
 openaivec/task/customer_support/customer_sentiment.py,sha256=vf2GzUk_06JD1FeRTetg5bCcBFUSqH349J1TI5lsgkg,7616
@@ -29,7 +29,7 @@ openaivec/task/nlp/sentiment_analysis.py,sha256=9HdMpi7FkjHNXAswaN98k8jeKsatBBXT
 openaivec/task/nlp/translation.py,sha256=4fjKtbVvOvivWMrpZfreIsdg8d0DplDujO8kAdLbAKI,6625
 openaivec/task/table/__init__.py,sha256=kJz15WDJXjyC7UIHKBvlTRhCf347PCDMH5T5fONV2sU,83
 openaivec/task/table/fillna.py,sha256=vi8t5QEIU-W3e05wwpATb3MEUDyf8luVnE8U-5VebZo,6582
-openaivec-0.13.6.dist-info/METADATA,sha256=JDjFmGAcoqSIeoK5JSEQ0Kd98HVRHW_DPwiUvsheDIU,27566
-openaivec-0.13.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openaivec-0.13.6.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
-openaivec-0.13.6.dist-info/RECORD,,
+openaivec-0.13.7.dist-info/METADATA,sha256=4_OtOhijWBx952u7owZwpsf3S5MtPsieGVo3s5v_v7w,27566
+openaivec-0.13.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+openaivec-0.13.7.dist-info/licenses/LICENSE,sha256=ws_MuBL-SCEBqPBFl9_FqZkaaydIJmxHrJG2parhU4M,1141
+openaivec-0.13.7.dist-info/RECORD,,