# openaivec
**Transform your data analysis with AI-powered text processing at scale.**
**openaivec** enables data analysts to seamlessly integrate OpenAI's language models into their pandas and Spark workflows. Process thousands of text records with natural language instructions, turning unstructured data into actionable insights with just a few lines of code.
## 🚀 Quick Start: From Text to Insights in Seconds
Imagine analyzing 10,000 customer reviews. Instead of manual work, just write:
```python
import pandas as pd
from openaivec import pandas_ext
# Your data
reviews = pd.DataFrame({
"review": ["Great product, fast delivery!", "Terrible quality, very disappointed", ...]
})
# AI-powered analysis in one line
results = reviews.assign(
sentiment=lambda df: df.review.ai.responses("Classify sentiment: positive/negative/neutral"),
issues=lambda df: df.review.ai.responses("Extract main issues or compliments"),
priority=lambda df: df.review.ai.responses("Priority for follow-up: low/medium/high")
)
```
**Result**: Thousands of reviews classified and analyzed in minutes, not days.
📓 **[Try it yourself →](https://openaivec.anareg.design/examples/pandas/)**
## 💡 Real-World Impact
### Customer Feedback Analysis
```python
# Process 50,000 support tickets automatically
tickets.assign(
category=lambda df: df.description.ai.responses("Categorize: billing/technical/feature_request"),
urgency=lambda df: df.description.ai.responses("Urgency level: low/medium/high/critical"),
solution_type=lambda df: df.description.ai.responses("Best resolution approach")
)
```
### Market Research at Scale
```python
# Analyze multilingual social media data
social_data.assign(
english_text=lambda df: df.post.ai.responses("Translate to English"),
brand_mention=lambda df: df.english_text.ai.responses("Extract brand mentions and sentiment"),
market_trend=lambda df: df.english_text.ai.responses("Identify emerging trends or concerns")
)
```
### Survey Data Transformation
```python
# Convert free-text responses to structured data
from pydantic import BaseModel
class Demographics(BaseModel):
age_group: str
location: str
interests: list[str]
survey_responses.assign(
structured=lambda df: df.response.ai.responses(
"Extract demographics as structured data",
response_format=Demographics
)
).ai.extract("structured") # Auto-expands to columns
```
📓 **[See more examples →](https://openaivec.anareg.design/examples/)**
# Overview
This package provides a vectorized interface for the OpenAI API, enabling you to process multiple inputs with a single
API call instead of sending requests one by one.
This approach helps reduce latency and simplifies your code.
Additionally, it integrates effortlessly with Pandas DataFrames and Apache Spark UDFs, making it easy to incorporate
into your data processing pipelines.
## Features
- Vectorized API requests for processing multiple inputs at once.
- Seamless integration with Pandas DataFrames.
- A UDF builder for Apache Spark.
- Compatibility with multiple OpenAI clients, including Azure OpenAI.
## Key Benefits
- **🚀 Performance**: Vectorized processing handles thousands of records in minutes, not hours
- **💰 Cost Efficiency**: Automatic deduplication reduces API costs by 50-90% on typical datasets
- **🔗 Integration**: Works within existing pandas/Spark workflows without architectural changes
- **📈 Scalability**: Same API scales from exploratory analysis (100s of records) to production systems (millions of records)
- **🏢 Enterprise Ready**: Microsoft Fabric integration, Apache Spark UDFs, Azure OpenAI compatibility
## Requirements
- Python 3.10 or higher
## Installation
Install the package with:
```bash
pip install openaivec
```
If you want to uninstall the package, you can do so with:
```bash
pip uninstall openaivec
```
## Basic Usage
### Direct API Usage
For maximum control over batch processing:
```python
import os
from openai import OpenAI
from openaivec import BatchResponses
# Initialize the batch client
client = BatchResponses(
client=OpenAI(),
model_name="gpt-4o-mini",
system_message="Please answer only with 'xx family' and do not output anything else."
)
result = client.parse(["panda", "rabbit", "koala"], batch_size=32)
print(result) # Expected output: ['bear family', 'rabbit family', 'koala family']
```
📓 **[Complete tutorial →](https://openaivec.anareg.design/examples/pandas/)**
### Pandas Integration (Recommended)
The easiest way to get started with your DataFrames:
```python
import pandas as pd
from openaivec import pandas_ext
# Setup (optional - uses OPENAI_API_KEY environment variable by default)
pandas_ext.responses_model("gpt-4o-mini")
# Create your data
df = pd.DataFrame({"name": ["panda", "rabbit", "koala"]})
# Add AI-powered columns
result = df.assign(
family=lambda df: df.name.ai.responses("What animal family? Answer with 'X family'"),
habitat=lambda df: df.name.ai.responses("Primary habitat in one word"),
fun_fact=lambda df: df.name.ai.responses("One interesting fact in 10 words or less")
)
```
| name | family | habitat | fun_fact |
|--------|---------------|---------|-----------------------------|
| panda | bear family | forest | Eats bamboo 14 hours daily |
| rabbit | rabbit family | meadow | Can see nearly 360 degrees |
| koala | marsupial family | tree | Sleeps 22 hours per day |
📓 **[Interactive pandas examples →](https://openaivec.anareg.design/examples/pandas/)**
## Using with Apache Spark UDFs
Scale to enterprise datasets with distributed processing:
📓 **[Complete Spark tutorial →](https://openaivec.anareg.design/examples/spark/)**
First, obtain a Spark session:
```python
from pyspark.sql import SparkSession
spark = SparkSession.builder.getOrCreate()
```
Next, instantiate UDF builders using either OpenAI or Azure OpenAI credentials and register the UDFs.
```python
import os
from openaivec.spark import ResponsesUDFBuilder, EmbeddingsUDFBuilder, count_tokens_udf
from pydantic import BaseModel
# --- Option 1: Using OpenAI ---
resp_builder_openai = ResponsesUDFBuilder.of_openai(
api_key=os.getenv("OPENAI_API_KEY"),
model_name="gpt-4o-mini", # Model for responses
)
emb_builder_openai = EmbeddingsUDFBuilder.of_openai(
api_key=os.getenv("OPENAI_API_KEY"),
model_name="text-embedding-3-small", # Model for embeddings
)
# --- Option 2: Using Azure OpenAI ---
# resp_builder_azure = ResponsesUDFBuilder.of_azure_openai(
# api_key=os.getenv("AZURE_OPENAI_KEY"),
# endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
# api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
# model_name="<your-resp-deployment-name>", # Deployment for responses
# )
# emb_builder_azure = EmbeddingsUDFBuilder.of_azure_openai(
# api_key=os.getenv("AZURE_OPENAI_KEY"),
# endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
# api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
# model_name="<your-emb-deployment-name>", # Deployment for embeddings
# )
# --- Register Responses UDF (String Output) ---
# Use the builder corresponding to your setup (OpenAI or Azure)
spark.udf.register(
"parse_flavor",
resp_builder_openai.build( # or resp_builder_azure.build(...)
instructions="Extract flavor-related information. Return only the concise flavor name.",
response_format=str, # Specify string output
)
)
# --- Register Responses UDF (Structured Output with Pydantic) ---
class Translation(BaseModel):
en: str
fr: str
ja: str
spark.udf.register(
"translate_struct",
resp_builder_openai.build( # or resp_builder_azure.build(...)
instructions="Translate the text to English, French, and Japanese.",
response_format=Translation, # Specify Pydantic model for structured output
)
)
# --- Register Embeddings UDF ---
spark.udf.register(
"embed_text",
emb_builder_openai.build() # or emb_builder_azure.build()
)
# --- Register Token Counting UDF ---
spark.udf.register("count_tokens", count_tokens_udf("gpt-4o"))
```
You can now use these UDFs in Spark SQL:
```sql
-- Create a sample table (replace with your actual table)
CREATE OR REPLACE TEMP VIEW product_names AS SELECT * FROM VALUES
('4414732714624', 'Cafe Mocha Smoothie (Trial Size)'),
('4200162318339', 'Dark Chocolate Tea (New Product)'),
('4920122084098', 'Uji Matcha Tea (New Product)')
AS product_names(id, product_name);
-- Use the registered UDFs
SELECT
id,
product_name,
parse_flavor(product_name) AS flavor,
translate_struct(product_name) AS translation,
embed_text(product_name) AS embedding,
count_tokens(product_name) AS token_count
FROM product_names;
```
Example Output (structure might vary slightly):
| id | product_name | flavor | translation | embedding | token_count |
|---------------|-----------------------------------|-----------|----------------------------------|--------------------------------|-------------|
| 4414732714624 | Cafe Mocha Smoothie (Trial Size) | Mocha | {en: ..., fr: ..., ja: ...} | [0.1, -0.2, ..., 0.5] | 8 |
| 4200162318339 | Dark Chocolate Tea (New Product) | Chocolate | {en: ..., fr: ..., ja: ...} | [-0.3, 0.1, ..., -0.1] | 7 |
| 4920122084098 | Uji Matcha Tea (New Product) | Matcha | {en: ..., fr: ..., ja: ...} | [0.0, 0.4, ..., 0.2] | 8 |
## Building Prompts
Building prompt is a crucial step in using LLMs.
In particular, providing a few examples in a prompt can significantly improve an LLM’s performance,
a technique known as "few-shot learning." Typically, a few-shot prompt consists of a purpose, cautions,
and examples.
📓 **[Advanced prompting techniques →](https://openaivec.anareg.design/examples/prompt/)**
The `FewShotPromptBuilder` helps you create structured, high-quality prompts with examples, cautions, and automatic improvement.
### Basic Usage
`FewShotPromptBuilder` requires simply a purpose, cautions, and examples, and `build` method will
return rendered prompt with XML format.
Here is an example:
```python
from openaivec.prompt import FewShotPromptBuilder
prompt: str = (
FewShotPromptBuilder()
.purpose("Return the smallest category that includes the given word")
.caution("Never use proper nouns as categories")
.example("Apple", "Fruit")
.example("Car", "Vehicle")
.example("Tokyo", "City")
.example("Keiichi Sogabe", "Musician")
.example("America", "Country")
.build()
)
print(prompt)
```
The output will be:
```xml
<Prompt>
<Purpose>Return the smallest category that includes the given word</Purpose>
<Cautions>
<Caution>Never use proper nouns as categories</Caution>
</Cautions>
<Examples>
<Example>
<Input>Apple</Input>
<Output>Fruit</Output>
</Example>
<Example>
<Input>Car</Input>
<Output>Vehicle</Output>
</Example>
<Example>
<Input>Tokyo</Input>
<Output>City</Output>
</Example>
<Example>
<Input>Keiichi Sogabe</Input>
<Output>Musician</Output>
</Example>
<Example>
<Input>America</Input>
<Output>Country</Output>
</Example>
</Examples>
</Prompt>
```
### Improve with OpenAI
For most users, it can be challenging to write a prompt entirely free of contradictions, ambiguities, or
redundancies.
`FewShotPromptBuilder` provides an `improve` method to refine your prompt using OpenAI's API.
`improve` method will try to eliminate contradictions, ambiguities, and redundancies in the prompt with OpenAI's API,
and iterate the process up to `max_iter` times.
Here is an example:
```python
from openai import OpenAI
from openaivec.prompt import FewShotPromptBuilder
client = OpenAI(...)
model_name = "<your-model-name>"
improved_prompt: str = (
FewShotPromptBuilder()
.purpose("Return the smallest category that includes the given word")
.caution("Never use proper nouns as categories")
# Examples which has contradictions, ambiguities, or redundancies
.example("Apple", "Fruit")
.example("Apple", "Technology")
.example("Apple", "Company")
.example("Apple", "Color")
.example("Apple", "Animal")
# improve the prompt with OpenAI's API
.improve(client, model_name)
.build()
)
print(improved_prompt)
```
Then we will get the improved prompt with extra examples, improved purpose, and cautions:
```xml
<Prompt>
<Purpose>Classify a given word into its most relevant category by considering its context and potential meanings.
The input is a word accompanied by context, and the output is the appropriate category based on that context.
This is useful for disambiguating words with multiple meanings, ensuring accurate understanding and
categorization.
</Purpose>
<Cautions>
<Caution>Ensure the context of the word is clear to avoid incorrect categorization.</Caution>
<Caution>Be aware of words with multiple meanings and provide the most relevant category.</Caution>
<Caution>Consider the possibility of new or uncommon contexts that may not fit traditional categories.</Caution>
</Cautions>
<Examples>
<Example>
<Input>Apple (as a fruit)</Input>
<Output>Fruit</Output>
</Example>
<Example>
<Input>Apple (as a tech company)</Input>
<Output>Technology</Output>
</Example>
<Example>
<Input>Java (as a programming language)</Input>
<Output>Technology</Output>
</Example>
<Example>
<Input>Java (as an island)</Input>
<Output>Geography</Output>
</Example>
<Example>
<Input>Mercury (as a planet)</Input>
<Output>Astronomy</Output>
</Example>
<Example>
<Input>Mercury (as an element)</Input>
<Output>Chemistry</Output>
</Example>
<Example>
<Input>Bark (as a sound made by a dog)</Input>
<Output>Animal Behavior</Output>
</Example>
<Example>
<Input>Bark (as the outer covering of a tree)</Input>
<Output>Botany</Output>
</Example>
<Example>
<Input>Bass (as a type of fish)</Input>
<Output>Aquatic Life</Output>
</Example>
<Example>
<Input>Bass (as a low-frequency sound)</Input>
<Output>Music</Output>
</Example>
</Examples>
</Prompt>
```
## Using with Microsoft Fabric
[Microsoft Fabric](https://www.microsoft.com/en-us/microsoft-fabric/) is a unified, cloud-based analytics platform that
seamlessly integrates data engineering, warehousing, and business intelligence to simplify the journey from raw data to
actionable insights.
This section provides instructions on how to integrate and use `openaivec` within Microsoft Fabric. Follow these
steps:
1. **Create an Environment in Microsoft Fabric:**
- In Microsoft Fabric, click on **New item** in your workspace.
- Select **Environment** to create a new environment for Apache Spark.
- Determine the environment name, eg. `openai-environment`.
- 
_Figure: Creating a new Environment in Microsoft Fabric._
2. **Add `openaivec` to the Environment from Public Library**
- Once your environment is set up, go to the **Custom Library** section within that environment.
- Click on **Add from PyPI** and search for latest version of `openaivec`.
- Save and publish to reflect the changes.
- 
_Figure: Add `openaivec` from PyPI to Public Library_
3. **Use the Environment from a Notebook:**
- Open a notebook within Microsoft Fabric.
- Select the environment you created in the previous steps.
- 
_Figure: Using custom environment from a notebook._
- In the notebook, import and use `openaivec.spark.ResponsesUDFBuilder` as you normally would. For example:
```python
from openaivec.spark import ResponsesUDFBuilder
resp_builder = ResponsesUDFBuilder.of_azure_openai(
api_key="<your-api-key>",
endpoint="https://<your-resource-name>.openai.azure.com",
api_version="2024-10-21",
model_name="<your-deployment-name>"
)
```
Following these steps allows you to successfully integrate and use `openaivec` within Microsoft Fabric.
## Contributing
We welcome contributions to this project! If you would like to contribute, please follow these guidelines:
1. Fork the repository and create your branch from `main`.
2. If you've added code that should be tested, add tests.
3. Ensure the test suite passes.
4. Make sure your code lints.
### Installing Dependencies
To install the necessary dependencies for development, run:
```bash
uv sync --all-extras --dev
```
### Code Formatting
To reformat the code, use the following command:
```bash
uv run ruff check . --fix
```
## Additional Resources
📓 **[Customer feedback analysis →](https://openaivec.anareg.design/examples/customer_analysis/)** - Sentiment analysis & prioritization
📓 **[Survey data transformation →](https://openaivec.anareg.design/examples/survey_transformation/)** - Unstructured to structured data
📓 **[Asynchronous processing examples →](https://openaivec.anareg.design/examples/aio/)** - High-performance async workflows
📓 **[Auto-generate FAQs from documents →](https://openaivec.anareg.design/examples/generate_faq/)** - Create FAQs using AI
📓 **[All examples →](https://openaivec.anareg.design/examples/)** - Complete collection of tutorials and use cases
## Community
Join our Discord community for developers: https://discord.gg/vbb83Pgn
Raw data
{
"_id": null,
"home_page": null,
"name": "openaivec",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.10",
"maintainer_email": null,
"keywords": "llm, openai, openai-api, openai-python, pandas, pyspark",
"author": null,
"author_email": "Hiroki Mizukami <hmizukami@microsoft.com>",
"download_url": "https://files.pythonhosted.org/packages/f8/39/261c9f68af3f274209cd40447c0ab45f0a57669f4023b4d13a06ff3ceb76/openaivec-0.9.1.tar.gz",
"platform": null,
"description": "# openaivec\n\n**Transform your data analysis with AI-powered text processing at scale.**\n\n**openaivec** enables data analysts to seamlessly integrate OpenAI's language models into their pandas and Spark workflows. Process thousands of text records with natural language instructions, turning unstructured data into actionable insights with just a few lines of code.\n\n## \ud83d\ude80 Quick Start: From Text to Insights in Seconds\n\nImagine analyzing 10,000 customer reviews. Instead of manual work, just write:\n\n```python\nimport pandas as pd\nfrom openaivec import pandas_ext\n\n# Your data\nreviews = pd.DataFrame({\n \"review\": [\"Great product, fast delivery!\", \"Terrible quality, very disappointed\", ...]\n})\n\n# AI-powered analysis in one line\nresults = reviews.assign(\n sentiment=lambda df: df.review.ai.responses(\"Classify sentiment: positive/negative/neutral\"),\n issues=lambda df: df.review.ai.responses(\"Extract main issues or compliments\"),\n priority=lambda df: df.review.ai.responses(\"Priority for follow-up: low/medium/high\")\n)\n```\n\n**Result**: Thousands of reviews classified and analyzed in minutes, not days.\n\n\ud83d\udcd3 **[Try it yourself \u2192](https://openaivec.anareg.design/examples/pandas/)**\n\n## \ud83d\udca1 Real-World Impact\n\n### Customer Feedback Analysis\n```python\n# Process 50,000 support tickets automatically\ntickets.assign(\n category=lambda df: df.description.ai.responses(\"Categorize: billing/technical/feature_request\"),\n urgency=lambda df: df.description.ai.responses(\"Urgency level: low/medium/high/critical\"),\n solution_type=lambda df: df.description.ai.responses(\"Best resolution approach\")\n)\n```\n\n### Market Research at Scale\n```python\n# Analyze multilingual social media data\nsocial_data.assign(\n english_text=lambda df: df.post.ai.responses(\"Translate to English\"),\n brand_mention=lambda df: df.english_text.ai.responses(\"Extract brand mentions and sentiment\"),\n market_trend=lambda df: df.english_text.ai.responses(\"Identify emerging trends or concerns\")\n)\n```\n\n### Survey Data Transformation\n```python\n# Convert free-text responses to structured data\nfrom pydantic import BaseModel\n\nclass Demographics(BaseModel):\n age_group: str\n location: str\n interests: list[str]\n\nsurvey_responses.assign(\n structured=lambda df: df.response.ai.responses(\n \"Extract demographics as structured data\", \n response_format=Demographics\n )\n).ai.extract(\"structured\") # Auto-expands to columns\n```\n\n\ud83d\udcd3 **[See more examples \u2192](https://openaivec.anareg.design/examples/)**\n\n# Overview\n\nThis package provides a vectorized interface for the OpenAI API, enabling you to process multiple inputs with a single\nAPI call instead of sending requests one by one.\nThis approach helps reduce latency and simplifies your code.\n\nAdditionally, it integrates effortlessly with Pandas DataFrames and Apache Spark UDFs, making it easy to incorporate\ninto your data processing pipelines.\n\n## Features\n\n- Vectorized API requests for processing multiple inputs at once.\n- Seamless integration with Pandas DataFrames.\n- A UDF builder for Apache Spark.\n- Compatibility with multiple OpenAI clients, including Azure OpenAI.\n\n## Key Benefits\n\n- **\ud83d\ude80 Performance**: Vectorized processing handles thousands of records in minutes, not hours\n- **\ud83d\udcb0 Cost Efficiency**: Automatic deduplication reduces API costs by 50-90% on typical datasets \n- **\ud83d\udd17 Integration**: Works within existing pandas/Spark workflows without architectural changes\n- **\ud83d\udcc8 Scalability**: Same API scales from exploratory analysis (100s of records) to production systems (millions of records)\n- **\ud83c\udfe2 Enterprise Ready**: Microsoft Fabric integration, Apache Spark UDFs, Azure OpenAI compatibility\n\n## Requirements\n\n- Python 3.10 or higher\n\n## Installation\n\nInstall the package with:\n\n```bash\npip install openaivec\n```\n\nIf you want to uninstall the package, you can do so with:\n\n```bash\npip uninstall openaivec\n```\n\n## Basic Usage\n\n### Direct API Usage\n\nFor maximum control over batch processing:\n\n```python\nimport os\nfrom openai import OpenAI\nfrom openaivec import BatchResponses\n\n# Initialize the batch client\nclient = BatchResponses(\n client=OpenAI(),\n model_name=\"gpt-4o-mini\",\n system_message=\"Please answer only with 'xx family' and do not output anything else.\"\n)\n\nresult = client.parse([\"panda\", \"rabbit\", \"koala\"], batch_size=32)\nprint(result) # Expected output: ['bear family', 'rabbit family', 'koala family']\n```\n\n\ud83d\udcd3 **[Complete tutorial \u2192](https://openaivec.anareg.design/examples/pandas/)**\n\n### Pandas Integration (Recommended)\n\nThe easiest way to get started with your DataFrames:\n\n```python\nimport pandas as pd\nfrom openaivec import pandas_ext\n\n# Setup (optional - uses OPENAI_API_KEY environment variable by default)\npandas_ext.responses_model(\"gpt-4o-mini\")\n\n# Create your data\ndf = pd.DataFrame({\"name\": [\"panda\", \"rabbit\", \"koala\"]})\n\n# Add AI-powered columns\nresult = df.assign(\n family=lambda df: df.name.ai.responses(\"What animal family? Answer with 'X family'\"),\n habitat=lambda df: df.name.ai.responses(\"Primary habitat in one word\"),\n fun_fact=lambda df: df.name.ai.responses(\"One interesting fact in 10 words or less\")\n)\n```\n\n| name | family | habitat | fun_fact |\n|--------|---------------|---------|-----------------------------|\n| panda | bear family | forest | Eats bamboo 14 hours daily |\n| rabbit | rabbit family | meadow | Can see nearly 360 degrees |\n| koala | marsupial family | tree | Sleeps 22 hours per day |\n\n\ud83d\udcd3 **[Interactive pandas examples \u2192](https://openaivec.anareg.design/examples/pandas/)**\n\n## Using with Apache Spark UDFs\n\nScale to enterprise datasets with distributed processing:\n\n\ud83d\udcd3 **[Complete Spark tutorial \u2192](https://openaivec.anareg.design/examples/spark/)**\n\nFirst, obtain a Spark session:\n\n```python\nfrom pyspark.sql import SparkSession\n\nspark = SparkSession.builder.getOrCreate()\n```\n\nNext, instantiate UDF builders using either OpenAI or Azure OpenAI credentials and register the UDFs.\n\n```python\nimport os\nfrom openaivec.spark import ResponsesUDFBuilder, EmbeddingsUDFBuilder, count_tokens_udf\nfrom pydantic import BaseModel\n\n# --- Option 1: Using OpenAI ---\nresp_builder_openai = ResponsesUDFBuilder.of_openai(\n api_key=os.getenv(\"OPENAI_API_KEY\"),\n model_name=\"gpt-4o-mini\", # Model for responses\n)\nemb_builder_openai = EmbeddingsUDFBuilder.of_openai(\n api_key=os.getenv(\"OPENAI_API_KEY\"),\n model_name=\"text-embedding-3-small\", # Model for embeddings\n)\n\n# --- Option 2: Using Azure OpenAI ---\n# resp_builder_azure = ResponsesUDFBuilder.of_azure_openai(\n# api_key=os.getenv(\"AZURE_OPENAI_KEY\"),\n# endpoint=os.getenv(\"AZURE_OPENAI_ENDPOINT\"),\n# api_version=os.getenv(\"AZURE_OPENAI_API_VERSION\"),\n# model_name=\"<your-resp-deployment-name>\", # Deployment for responses\n# )\n# emb_builder_azure = EmbeddingsUDFBuilder.of_azure_openai(\n# api_key=os.getenv(\"AZURE_OPENAI_KEY\"),\n# endpoint=os.getenv(\"AZURE_OPENAI_ENDPOINT\"),\n# api_version=os.getenv(\"AZURE_OPENAI_API_VERSION\"),\n# model_name=\"<your-emb-deployment-name>\", # Deployment for embeddings\n# )\n\n# --- Register Responses UDF (String Output) ---\n# Use the builder corresponding to your setup (OpenAI or Azure)\nspark.udf.register(\n \"parse_flavor\",\n resp_builder_openai.build( # or resp_builder_azure.build(...)\n instructions=\"Extract flavor-related information. Return only the concise flavor name.\",\n response_format=str, # Specify string output\n )\n)\n\n# --- Register Responses UDF (Structured Output with Pydantic) ---\nclass Translation(BaseModel):\n en: str\n fr: str\n ja: str\n\nspark.udf.register(\n \"translate_struct\",\n resp_builder_openai.build( # or resp_builder_azure.build(...)\n instructions=\"Translate the text to English, French, and Japanese.\",\n response_format=Translation, # Specify Pydantic model for structured output\n )\n)\n\n# --- Register Embeddings UDF ---\nspark.udf.register(\n \"embed_text\",\n emb_builder_openai.build() # or emb_builder_azure.build()\n)\n\n# --- Register Token Counting UDF ---\nspark.udf.register(\"count_tokens\", count_tokens_udf(\"gpt-4o\"))\n\n```\n\nYou can now use these UDFs in Spark SQL:\n\n```sql\n-- Create a sample table (replace with your actual table)\nCREATE OR REPLACE TEMP VIEW product_names AS SELECT * FROM VALUES\n ('4414732714624', 'Cafe Mocha Smoothie (Trial Size)'),\n ('4200162318339', 'Dark Chocolate Tea (New Product)'),\n ('4920122084098', 'Uji Matcha Tea (New Product)')\nAS product_names(id, product_name);\n\n-- Use the registered UDFs\nSELECT\n id,\n product_name,\n parse_flavor(product_name) AS flavor,\n translate_struct(product_name) AS translation,\n embed_text(product_name) AS embedding,\n count_tokens(product_name) AS token_count\nFROM product_names;\n```\n\nExample Output (structure might vary slightly):\n\n| id | product_name | flavor | translation | embedding | token_count |\n|---------------|-----------------------------------|-----------|----------------------------------|--------------------------------|-------------|\n| 4414732714624 | Cafe Mocha Smoothie (Trial Size) | Mocha | {en: ..., fr: ..., ja: ...} | [0.1, -0.2, ..., 0.5] | 8 |\n| 4200162318339 | Dark Chocolate Tea (New Product) | Chocolate | {en: ..., fr: ..., ja: ...} | [-0.3, 0.1, ..., -0.1] | 7 |\n| 4920122084098 | Uji Matcha Tea (New Product) | Matcha | {en: ..., fr: ..., ja: ...} | [0.0, 0.4, ..., 0.2] | 8 |\n\n## Building Prompts\n\nBuilding prompt is a crucial step in using LLMs.\nIn particular, providing a few examples in a prompt can significantly improve an LLM\u2019s performance,\na technique known as \"few-shot learning.\" Typically, a few-shot prompt consists of a purpose, cautions,\nand examples.\n\n\ud83d\udcd3 **[Advanced prompting techniques \u2192](https://openaivec.anareg.design/examples/prompt/)**\n\nThe `FewShotPromptBuilder` helps you create structured, high-quality prompts with examples, cautions, and automatic improvement.\n\n### Basic Usage\n\n`FewShotPromptBuilder` requires simply a purpose, cautions, and examples, and `build` method will\nreturn rendered prompt with XML format.\n\nHere is an example:\n\n```python\nfrom openaivec.prompt import FewShotPromptBuilder\n\nprompt: str = (\n FewShotPromptBuilder()\n .purpose(\"Return the smallest category that includes the given word\")\n .caution(\"Never use proper nouns as categories\")\n .example(\"Apple\", \"Fruit\")\n .example(\"Car\", \"Vehicle\")\n .example(\"Tokyo\", \"City\")\n .example(\"Keiichi Sogabe\", \"Musician\")\n .example(\"America\", \"Country\")\n .build()\n)\nprint(prompt)\n```\n\nThe output will be:\n\n```xml\n\n<Prompt>\n <Purpose>Return the smallest category that includes the given word</Purpose>\n <Cautions>\n <Caution>Never use proper nouns as categories</Caution>\n </Cautions>\n <Examples>\n <Example>\n <Input>Apple</Input>\n <Output>Fruit</Output>\n </Example>\n <Example>\n <Input>Car</Input>\n <Output>Vehicle</Output>\n </Example>\n <Example>\n <Input>Tokyo</Input>\n <Output>City</Output>\n </Example>\n <Example>\n <Input>Keiichi Sogabe</Input>\n <Output>Musician</Output>\n </Example>\n <Example>\n <Input>America</Input>\n <Output>Country</Output>\n </Example>\n </Examples>\n</Prompt>\n```\n\n### Improve with OpenAI\n\nFor most users, it can be challenging to write a prompt entirely free of contradictions, ambiguities, or\nredundancies.\n`FewShotPromptBuilder` provides an `improve` method to refine your prompt using OpenAI's API.\n\n`improve` method will try to eliminate contradictions, ambiguities, and redundancies in the prompt with OpenAI's API,\nand iterate the process up to `max_iter` times.\n\nHere is an example:\n\n```python\nfrom openai import OpenAI\nfrom openaivec.prompt import FewShotPromptBuilder\n\nclient = OpenAI(...)\nmodel_name = \"<your-model-name>\"\nimproved_prompt: str = (\n FewShotPromptBuilder()\n .purpose(\"Return the smallest category that includes the given word\")\n .caution(\"Never use proper nouns as categories\")\n # Examples which has contradictions, ambiguities, or redundancies\n .example(\"Apple\", \"Fruit\")\n .example(\"Apple\", \"Technology\")\n .example(\"Apple\", \"Company\")\n .example(\"Apple\", \"Color\")\n .example(\"Apple\", \"Animal\")\n # improve the prompt with OpenAI's API\n .improve(client, model_name)\n .build()\n)\nprint(improved_prompt)\n```\n\nThen we will get the improved prompt with extra examples, improved purpose, and cautions:\n\n```xml\n<Prompt>\n <Purpose>Classify a given word into its most relevant category by considering its context and potential meanings.\n The input is a word accompanied by context, and the output is the appropriate category based on that context.\n This is useful for disambiguating words with multiple meanings, ensuring accurate understanding and\n categorization.\n </Purpose>\n <Cautions>\n <Caution>Ensure the context of the word is clear to avoid incorrect categorization.</Caution>\n <Caution>Be aware of words with multiple meanings and provide the most relevant category.</Caution>\n <Caution>Consider the possibility of new or uncommon contexts that may not fit traditional categories.</Caution>\n </Cautions>\n <Examples>\n <Example>\n <Input>Apple (as a fruit)</Input>\n <Output>Fruit</Output>\n </Example>\n <Example>\n <Input>Apple (as a tech company)</Input>\n <Output>Technology</Output>\n </Example>\n <Example>\n <Input>Java (as a programming language)</Input>\n <Output>Technology</Output>\n </Example>\n <Example>\n <Input>Java (as an island)</Input>\n <Output>Geography</Output>\n </Example>\n <Example>\n <Input>Mercury (as a planet)</Input>\n <Output>Astronomy</Output>\n </Example>\n <Example>\n <Input>Mercury (as an element)</Input>\n <Output>Chemistry</Output>\n </Example>\n <Example>\n <Input>Bark (as a sound made by a dog)</Input>\n <Output>Animal Behavior</Output>\n </Example>\n <Example>\n <Input>Bark (as the outer covering of a tree)</Input>\n <Output>Botany</Output>\n </Example>\n <Example>\n <Input>Bass (as a type of fish)</Input>\n <Output>Aquatic Life</Output>\n </Example>\n <Example>\n <Input>Bass (as a low-frequency sound)</Input>\n <Output>Music</Output>\n </Example>\n </Examples>\n</Prompt>\n```\n\n## Using with Microsoft Fabric\n\n[Microsoft Fabric](https://www.microsoft.com/en-us/microsoft-fabric/) is a unified, cloud-based analytics platform that\nseamlessly integrates data engineering, warehousing, and business intelligence to simplify the journey from raw data to\nactionable insights.\n\nThis section provides instructions on how to integrate and use `openaivec` within Microsoft Fabric. Follow these\nsteps:\n\n1. **Create an Environment in Microsoft Fabric:**\n\n - In Microsoft Fabric, click on **New item** in your workspace.\n - Select **Environment** to create a new environment for Apache Spark.\n - Determine the environment name, eg. `openai-environment`.\n - \n _Figure: Creating a new Environment in Microsoft Fabric._\n\n2. **Add `openaivec` to the Environment from Public Library**\n\n - Once your environment is set up, go to the **Custom Library** section within that environment.\n - Click on **Add from PyPI** and search for latest version of `openaivec`.\n - Save and publish to reflect the changes.\n - \n _Figure: Add `openaivec` from PyPI to Public Library_\n\n3. **Use the Environment from a Notebook:**\n\n - Open a notebook within Microsoft Fabric.\n - Select the environment you created in the previous steps.\n - \n _Figure: Using custom environment from a notebook._\n - In the notebook, import and use `openaivec.spark.ResponsesUDFBuilder` as you normally would. For example:\n\n ```python\n from openaivec.spark import ResponsesUDFBuilder\n\n resp_builder = ResponsesUDFBuilder.of_azure_openai(\n api_key=\"<your-api-key>\",\n endpoint=\"https://<your-resource-name>.openai.azure.com\",\n api_version=\"2024-10-21\",\n model_name=\"<your-deployment-name>\"\n )\n ```\n\nFollowing these steps allows you to successfully integrate and use `openaivec` within Microsoft Fabric.\n\n## Contributing\n\nWe welcome contributions to this project! If you would like to contribute, please follow these guidelines:\n\n1. Fork the repository and create your branch from `main`.\n2. If you've added code that should be tested, add tests.\n3. Ensure the test suite passes.\n4. Make sure your code lints.\n\n### Installing Dependencies\n\nTo install the necessary dependencies for development, run:\n\n```bash\nuv sync --all-extras --dev\n```\n\n### Code Formatting\n\nTo reformat the code, use the following command:\n\n```bash\nuv run ruff check . --fix\n```\n\n## Additional Resources\n\n\ud83d\udcd3 **[Customer feedback analysis \u2192](https://openaivec.anareg.design/examples/customer_analysis/)** - Sentiment analysis & prioritization \n\ud83d\udcd3 **[Survey data transformation \u2192](https://openaivec.anareg.design/examples/survey_transformation/)** - Unstructured to structured data \n\ud83d\udcd3 **[Asynchronous processing examples \u2192](https://openaivec.anareg.design/examples/aio/)** - High-performance async workflows \n\ud83d\udcd3 **[Auto-generate FAQs from documents \u2192](https://openaivec.anareg.design/examples/generate_faq/)** - Create FAQs using AI \n\ud83d\udcd3 **[All examples \u2192](https://openaivec.anareg.design/examples/)** - Complete collection of tutorials and use cases\n\n## Community\n\nJoin our Discord community for developers: https://discord.gg/vbb83Pgn\n",
"bugtrack_url": null,
"license": "MIT",
"summary": "Generative mutation for tabular calculation",
"version": "0.9.1",
"project_urls": {
"Homepage": "https://openaivec.anareg.design/",
"Repository": "https://github.com/anaregdesign/openaivec"
},
"split_keywords": [
"llm",
" openai",
" openai-api",
" openai-python",
" pandas",
" pyspark"
],
"urls": [
{
"comment_text": null,
"digests": {
"blake2b_256": "01979919cd352b555291764e0b0b315ffa3ec8b8a0b23672f025a6119772cf4f",
"md5": "ef24924a884f43eee1de47c17f1e9c41",
"sha256": "9ae3e020fac6eb7dadddb1106f73faf2e6df9fb3f313e498814cda7349cf0cf5"
},
"downloads": -1,
"filename": "openaivec-0.9.1-py3-none-any.whl",
"has_sig": false,
"md5_digest": "ef24924a884f43eee1de47c17f1e9c41",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 45514,
"upload_time": "2025-07-11T12:28:09",
"upload_time_iso_8601": "2025-07-11T12:28:09.280442Z",
"url": "https://files.pythonhosted.org/packages/01/97/9919cd352b555291764e0b0b315ffa3ec8b8a0b23672f025a6119772cf4f/openaivec-0.9.1-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": null,
"digests": {
"blake2b_256": "f839261c9f68af3f274209cd40447c0ab45f0a57669f4023b4d13a06ff3ceb76",
"md5": "26bb76560815a6130b3a2370e2be5529",
"sha256": "4ff0761430c7193104e8b34b9fc8a1c8b85884dd7802a706ef5d2f128b7815e7"
},
"downloads": -1,
"filename": "openaivec-0.9.1.tar.gz",
"has_sig": false,
"md5_digest": "26bb76560815a6130b3a2370e2be5529",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 171755,
"upload_time": "2025-07-11T12:28:10",
"upload_time_iso_8601": "2025-07-11T12:28:10.572470Z",
"url": "https://files.pythonhosted.org/packages/f8/39/261c9f68af3f274209cd40447c0ab45f0a57669f4023b4d13a06ff3ceb76/openaivec-0.9.1.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2025-07-11 12:28:10",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "anaregdesign",
"github_project": "openaivec",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "openaivec"
}
JSON
