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,85 @@
|
||||
// Example from @Codeshark-NET https://github.com/promptfoo/promptfoo/issues/922
|
||||
// @ts-check
|
||||
import { anthropic } from '@ai-sdk/anthropic';
|
||||
import { generateObject } from 'ai';
|
||||
import promptfoo from 'promptfoo';
|
||||
import { promptSchema } from './schemaValidation.mjs';
|
||||
|
||||
class CustomProvider {
|
||||
constructor(options) {
|
||||
// Provider ID can be overridden by the config file (e.g. when using multiple of the same provider)
|
||||
this.providerId = options.id || 'custom provider';
|
||||
|
||||
// options.config contains any custom options passed to the provider
|
||||
this.config = options.config;
|
||||
}
|
||||
|
||||
id() {
|
||||
return this.providerId;
|
||||
}
|
||||
|
||||
async callApi(prompt, context) {
|
||||
const cache = await promptfoo.default.cache.getCache();
|
||||
|
||||
// Create a unique cache key based on the prompt and context
|
||||
const cacheKey = `api:${this.providerId}:${prompt}}`; // :${JSON.stringify(context)
|
||||
|
||||
// Check if the response is already cached
|
||||
const cachedResponse = await cache.get(cacheKey);
|
||||
if (cachedResponse) {
|
||||
return {
|
||||
// Required
|
||||
output: JSON.parse(cachedResponse),
|
||||
|
||||
// Optional
|
||||
tokenUsage: {
|
||||
total: 0, // No tokens used because it's from the cache
|
||||
prompt: 0,
|
||||
completion: 0,
|
||||
},
|
||||
|
||||
cost: 0, // No cost because it's from the cache
|
||||
};
|
||||
}
|
||||
|
||||
// If not cached, make the function call
|
||||
const model = anthropic('claude-haiku-4-5-20251001');
|
||||
const { object, usage } = await generateObject({
|
||||
model,
|
||||
messages: JSON.parse(prompt),
|
||||
maxTokens: 4096,
|
||||
temperature: 0.4,
|
||||
maxRetries: 0,
|
||||
schema: promptSchema,
|
||||
mode: 'tool',
|
||||
});
|
||||
|
||||
const inputCost = 0.00025 / 1000; //config.cost ?? model.cost.input;
|
||||
const outputCost = 0.00125 / 1000; // config.cost ?? model.cost.output;
|
||||
const totalCost =
|
||||
inputCost * usage.promptTokens + outputCost * usage.completionTokens || undefined;
|
||||
|
||||
// Store the response in the cache
|
||||
try {
|
||||
await cache.set(cacheKey, JSON.stringify(object));
|
||||
} catch (error) {
|
||||
console.error('Failed to store response in cache:', error);
|
||||
}
|
||||
|
||||
return {
|
||||
// Required
|
||||
output: object,
|
||||
|
||||
// Optional
|
||||
tokenUsage: {
|
||||
total: usage.totalTokens,
|
||||
prompt: usage.promptTokens,
|
||||
completion: usage.completionTokens,
|
||||
},
|
||||
|
||||
cost: totalCost,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
export default CustomProvider;
|
||||
Reference in New Issue
Block a user