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,79 @@
|
||||
import express from 'express';
|
||||
import OpenAI from 'openai';
|
||||
import { v4 as uuidv4 } from 'uuid';
|
||||
|
||||
const app = express();
|
||||
const port = 8080;
|
||||
|
||||
app.use(express.json());
|
||||
|
||||
// In-memory storage for conversation histories
|
||||
const conversationHistories = new Map();
|
||||
|
||||
// Initialize OpenAI client
|
||||
const openai = new OpenAI({
|
||||
apiKey: process.env.OPENAI_API_KEY,
|
||||
});
|
||||
|
||||
app.get('/session', (req, res) => {
|
||||
const sessionId = uuidv4();
|
||||
console.log(`Session started: ${sessionId}`);
|
||||
conversationHistories.set(sessionId, []);
|
||||
res.status(201).send(sessionId);
|
||||
});
|
||||
|
||||
app.delete('/session/:sessionId', (req, res) => {
|
||||
const { sessionId } = req.params;
|
||||
console.log(`Session ended: ${sessionId}`);
|
||||
conversationHistories.delete(sessionId);
|
||||
res.status(204).send(sessionId);
|
||||
});
|
||||
|
||||
// Inference endpoint
|
||||
app.post('/inference', async (req, res) => {
|
||||
try {
|
||||
const { message, sessionId } = req.body;
|
||||
|
||||
// Get or initialize conversation history for this session
|
||||
if (!conversationHistories.has(sessionId)) {
|
||||
conversationHistories.set(sessionId, []);
|
||||
}
|
||||
|
||||
const history = conversationHistories.get(sessionId);
|
||||
|
||||
// Add new messages to history
|
||||
const newMessages = Array.isArray(message) ? message : [{ role: 'user', content: message }];
|
||||
history.push(...newMessages);
|
||||
|
||||
console.log(`New user message (${sessionId}: ${history.length} messages)`);
|
||||
|
||||
// Make inference request with full conversation history
|
||||
const completion = await openai.chat.completions.create({
|
||||
model: 'gpt-4o-2024-08-06',
|
||||
messages: history,
|
||||
});
|
||||
|
||||
const assistantMessage = {
|
||||
role: 'assistant',
|
||||
content: completion.choices[0].message.content,
|
||||
};
|
||||
|
||||
// Add assistant response to history
|
||||
history.push(assistantMessage);
|
||||
|
||||
// Update the stored history
|
||||
conversationHistories.set(sessionId, history);
|
||||
|
||||
res.json({ output: completion.choices[0].message.content });
|
||||
} catch (error) {
|
||||
console.error('Inference error:', error);
|
||||
res.status(500).json({
|
||||
error: 'Inference failed',
|
||||
message: error.message,
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
app.listen(port, () => {
|
||||
console.log(`Server running on http://localhost:${port}`);
|
||||
});
|
||||
Reference in New Issue
Block a user