Repository navigation
Expand file tree
/
Copy pathclaude.js
More file actions
72 lines (67 loc) · 2.1 KB
/
Copy pathclaude.js
File metadata and controls
72 lines (67 loc) · 2.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
const https = require('https');
// Helper to call Anthropic Claude API (claude-3-5-sonnet, claude-3-opus, claude-3-5-haiku)
async function analyzeImageWithClaude(apiKey, imageBuffer, mimeType, prompt, model = 'claude-3-5-sonnet-20241022') {
const base64Data = imageBuffer.toString('base64');
let cleanMime = mimeType || 'image/jpeg';
if (cleanMime.includes('jpg')) cleanMime = 'image/jpeg';
if (!['image/jpeg', 'image/png', 'image/gif', 'image/webp'].includes(cleanMime)) {
cleanMime = 'image/jpeg';
}
const cleanModel = (model || 'claude-3-5-sonnet-20241022').trim();
const requestBody = JSON.stringify({
model: cleanModel,
max_tokens: 1500,
messages: [
{
role: 'user',
content: [
{
type: 'image',
source: {
type: 'base64',
media_type: cleanMime,
data: base64Data
}
},
{
type: 'text',
text: prompt
}
]
}
]
});
return new Promise((resolve, reject) => {
const req = https.request('https://api.anthropic.com/v1/messages', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
'x-api-key': apiKey.trim(),
'anthropic-version': '2023-06-01',
'Content-Length': Buffer.byteLength(requestBody)
}
}, res => {
let data = '';
res.on('data', chunk => data += chunk);
res.on('end', () => {
try {
const parsed = JSON.parse(data);
if (parsed.error) {
return reject(new Error(parsed.error.message || 'Claude API Error'));
}
if (parsed.content && parsed.content[0] && parsed.content[0].text) {
resolve(parsed.content[0].text.trim());
} else {
reject(new Error('No response text generated by Claude'));
}
} catch (e) {
reject(new Error('Invalid response from Claude API: ' + e.message));
}
});
});
req.on('error', reject);
req.write(requestBody);
req.end();
});
}
module.exports = { analyzeImageWithClaude };