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,42 @@
|
||||
# config-pdf-files (PDF Document Analysis)
|
||||
|
||||
This example demonstrates how to analyze PDF documents using vision-capable LLMs from multiple providers.
|
||||
|
||||
You can run this example with:
|
||||
|
||||
```bash
|
||||
npx promptfoo@latest init --example config-pdf-files
|
||||
cd config-pdf-files
|
||||
```
|
||||
|
||||
## Setup
|
||||
|
||||
Set your API keys for the providers you want to use:
|
||||
|
||||
- **Google Gemini**: Set `GOOGLE_API_KEY` or `GEMINI_API_KEY`
|
||||
- **Anthropic Claude**: Set `ANTHROPIC_API_KEY`
|
||||
- **OpenAI GPT**: Set `OPENAI_API_KEY`
|
||||
|
||||
## Usage
|
||||
|
||||
This example is pre-configured in `promptfooconfig.yaml`. Run:
|
||||
|
||||
```bash
|
||||
promptfoo eval
|
||||
```
|
||||
|
||||
View results:
|
||||
|
||||
```bash
|
||||
promptfoo view
|
||||
```
|
||||
|
||||
## How It Works
|
||||
|
||||
The `prompt.py` file handles provider-specific PDF formatting:
|
||||
|
||||
- **Anthropic**: Uses the `document` content type with base64 data
|
||||
- **OpenAI**: Uses `input_file` with data URI format
|
||||
- **Google Gemini**: Uses `inline_data` within message parts
|
||||
|
||||
Each provider has different requirements for sending PDF documents, which is why a Python prompt function is used to dynamically format the request.
|
||||
Binary file not shown.
@@ -0,0 +1,113 @@
|
||||
import base64
|
||||
import os
|
||||
import typing
|
||||
|
||||
# Get the directory where this script is located
|
||||
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
|
||||
class Vars(typing.TypedDict):
|
||||
file_path: str
|
||||
|
||||
|
||||
class Provider(typing.TypedDict):
|
||||
id: str
|
||||
|
||||
|
||||
class PromptFunctionContext(typing.TypedDict):
|
||||
vars: Vars
|
||||
provider: Provider
|
||||
|
||||
|
||||
def get_file_base64(file_path: str) -> str:
|
||||
"""
|
||||
Read a local file and convert it to a base64-encoded string.
|
||||
|
||||
Args:
|
||||
file_path: Path to the file (relative to this script's directory).
|
||||
|
||||
Returns:
|
||||
The base64-encoded file data.
|
||||
"""
|
||||
resolved_path = os.path.join(SCRIPT_DIR, file_path)
|
||||
with open(resolved_path, "rb") as f:
|
||||
return base64.b64encode(f.read()).decode("utf-8")
|
||||
|
||||
|
||||
SYSTEM_PROMPT = "Count the number of diagrams on this page"
|
||||
PDF_MIME_TYPE = "application/pdf"
|
||||
|
||||
|
||||
def format_pdf_prompt(context: PromptFunctionContext) -> list[dict[str, typing.Any]]:
|
||||
"""
|
||||
Format the prompt for PDF analysis based on the AI provider.
|
||||
|
||||
Each provider has a different format for sending PDF documents.
|
||||
This function handles the provider-specific formatting.
|
||||
|
||||
Args:
|
||||
context: Dictionary containing provider info and variables.
|
||||
|
||||
Returns:
|
||||
Formatted prompt as a list of message dictionaries.
|
||||
|
||||
Raises:
|
||||
ValueError: If an unsupported provider is specified.
|
||||
"""
|
||||
provider_id = context["provider"]["id"]
|
||||
file_path = context["vars"]["file_path"]
|
||||
file_data = get_file_base64(file_path)
|
||||
|
||||
if provider_id.startswith("anthropic:"):
|
||||
return [
|
||||
{"role": "system", "content": SYSTEM_PROMPT},
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{
|
||||
"type": "document",
|
||||
"source": {
|
||||
"type": "base64",
|
||||
"media_type": PDF_MIME_TYPE,
|
||||
"data": file_data,
|
||||
},
|
||||
}
|
||||
],
|
||||
},
|
||||
]
|
||||
|
||||
if provider_id.startswith("openai:"):
|
||||
return [
|
||||
{
|
||||
"role": "system",
|
||||
"content": [{"type": "input_text", "text": SYSTEM_PROMPT}],
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{
|
||||
"type": "input_file",
|
||||
"filename": "document.pdf",
|
||||
"file_data": f"data:{PDF_MIME_TYPE};base64,{file_data}",
|
||||
}
|
||||
],
|
||||
},
|
||||
]
|
||||
|
||||
if provider_id.startswith("google:") or provider_id.startswith("vertex:"):
|
||||
return [
|
||||
{
|
||||
"role": "user",
|
||||
"parts": [
|
||||
{
|
||||
"inline_data": {
|
||||
"mime_type": PDF_MIME_TYPE,
|
||||
"data": file_data,
|
||||
}
|
||||
},
|
||||
{"text": SYSTEM_PROMPT},
|
||||
],
|
||||
}
|
||||
]
|
||||
|
||||
raise ValueError(f"Unsupported provider: {provider_id}")
|
||||
@@ -0,0 +1,20 @@
|
||||
# yaml-language-server: $schema=https://promptfoo.dev/config-schema.json
|
||||
description: 'PDF Document Analysis'
|
||||
|
||||
prompts:
|
||||
- file://prompt.py:format_pdf_prompt
|
||||
|
||||
providers:
|
||||
- id: google:gemini-2.5-flash
|
||||
- id: anthropic:messages:claude-sonnet-4-6
|
||||
- id: openai:responses:gpt-5.4
|
||||
|
||||
defaultTest:
|
||||
assert:
|
||||
- type: contains
|
||||
value: '2'
|
||||
|
||||
tests:
|
||||
- description: 'Count diagrams in a research paper PDF'
|
||||
vars:
|
||||
file_path: pdfs/page.pdf
|
||||
Reference in New Issue
Block a user