chore: import upstream snapshot with attribution
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,99 @@
|
||||
# provider-python (Python Provider)
|
||||
|
||||
This example demonstrates how to create a custom Python provider for promptfoo that integrates with the OpenAI API.
|
||||
|
||||
You can run this example with:
|
||||
|
||||
```bash
|
||||
npx promptfoo@latest init --example provider-python
|
||||
cd provider-python
|
||||
```
|
||||
|
||||
## Overview
|
||||
|
||||
The Python provider allows you to use Python code as a provider in promptfoo evaluations. This is useful when you need to:
|
||||
|
||||
1. Call APIs from Python libraries
|
||||
2. Implement custom logic before or after calling LLMs
|
||||
3. Process responses in specific ways
|
||||
4. Track token usage and other metrics
|
||||
|
||||
## Environment Variables
|
||||
|
||||
This example requires the following environment variable:
|
||||
|
||||
- `OPENAI_API_KEY` - Your OpenAI API key
|
||||
|
||||
You can set this in a `.env` file or directly in your environment.
|
||||
|
||||
## Requirements
|
||||
|
||||
- Python with the OpenAI package installed (`pip install openai`)
|
||||
|
||||
## Files
|
||||
|
||||
- `provider.py` - The Python provider implementation that calls OpenAI's API
|
||||
- `promptfooconfig.yaml` - Configuration for promptfoo evaluation with proper YAML schema reference
|
||||
- `configs/` directory:
|
||||
- `fileConfig.yaml` - YAML configuration for model settings
|
||||
- `fileConfig.js` - JavaScript configuration for formatting options
|
||||
- `fileConfig.py` - Python configuration for additional parameters
|
||||
|
||||
## Implementation Details
|
||||
|
||||
The Python provider is defined in `provider.py` and includes:
|
||||
|
||||
1. A `call_api` function that makes API calls to OpenAI
|
||||
2. Token usage extraction from the API response
|
||||
3. Multiple sample functions showing different ways to call the API
|
||||
|
||||
By default, the example is configured to use `gpt-4.1-mini` model, but you can modify it to use other models as needed.
|
||||
|
||||
## Expected Output
|
||||
|
||||
When you run this example, you'll see:
|
||||
|
||||
1. The prompts being submitted to your Python provider
|
||||
2. Responses from the OpenAI API
|
||||
3. Token usage statistics for each completion
|
||||
4. Evaluation results in a table format
|
||||
|
||||
## File Reference Configuration
|
||||
|
||||
The example demonstrates how to load configuration values from external files using the `file://` protocol directly in the `promptfooconfig.yaml` file. It shows three main file types:
|
||||
|
||||
1. **YAML file** (`configs/fileConfig.yaml`): Contains model settings like temperature and max tokens
|
||||
2. **JavaScript file** (`configs/fileConfig.js`): Provides formatting options through a function export
|
||||
3. **Python file** (`configs/fileConfig.py`): Supplies additional parameters through a Python function
|
||||
|
||||
The provider supports loading from:
|
||||
|
||||
- JSON files (`.json`)
|
||||
- YAML files (`.yaml`, `.yml`)
|
||||
- JavaScript files (`.js`, `.mjs`, `.ts`, `.cjs`)
|
||||
- Python files (`.py`)
|
||||
- Text files (`.txt`, `.md`)
|
||||
|
||||
You can see how this works in the `promptfooconfig.yaml` file:
|
||||
|
||||
```yaml
|
||||
providers:
|
||||
- id: 'file://provider.py:call_api'
|
||||
config:
|
||||
# YAML
|
||||
settings: 'file://configs/fileConfig.yaml'
|
||||
# JavaScript file
|
||||
formatting: 'file://configs/fileConfig.js:getFormatConfig'
|
||||
nested: # Python file
|
||||
parameters: 'file://configs/fileConfig.py:get_params'
|
||||
```
|
||||
|
||||
Run the example with:
|
||||
|
||||
```bash
|
||||
npx promptfoo@latest evaluate -c examples/provider-python/promptfooconfig.yaml
|
||||
```
|
||||
|
||||
## Learn More
|
||||
|
||||
For more information on creating custom providers, see the [promptfoo documentation](https://promptfoo.dev/docs/providers/python/).
|
||||
Reference in New Issue
Block a user