AI Proxy

import { HiveForgeClient } from '@producthacker/hiveforge-sdk';
 
const hiveforge = new HiveForgeClient({
  deploymentId: process.env.HIVEFORGE_DEPLOYMENT_ID!,
  deploymentSecret: process.env.HIVEFORGE_DEPLOYMENT_SECRET!,
});
await hiveforge.initialize();
 
// Check if AI is available for this tier
if (hiveforge.ai.isEnabled()) {
  const response = await hiveforge.ai.complete({
    messages: [{ role: 'user', content: 'Hello!' }],
  });
  console.log(response.content);
}

const response = await hiveforge.ai.complete({
  messages: [
    { role: 'system', content: 'You are a helpful customer support agent.' },
    { role: 'user', content: 'How do I reset my password?' },
  ],
  model: 'gpt-4o-mini',
  temperature: 0.7,
  max_tokens: 500,
});
 
console.log(response.content);
console.log(`Tokens used: ${response.tokens_used}`);
console.log(`Model: ${response.model}`);

let fullResponse = '';
 
for await (const chunk of hiveforge.ai.stream({
  messages: [
    { role: 'user', content: 'Write a short poem about TypeScript.' },
  ],
  model: 'gpt-4o-mini',
})) {
  process.stdout.write(chunk.content);
  fullResponse += chunk.content;
 
  if (chunk.done) {
    console.log('\n--- Stream complete ---');
  }
}

for await (const chunk of hiveforge.ai.stream({
  messages: [{ role: 'user', content: 'Tell me a story' }],
  onChunk: (chunk) => {
    // Called for each chunk
    process.stdout.write(chunk.content);
  },
  onComplete: (fullContent) => {
    // Called when streaming finishes
    console.log(`\nTotal length: ${fullContent.length} chars`);
  },
  onError: (error) => {
    console.error('Stream error:', error.message);
  },
})) {
  // You can also process chunks here
}

const fullResponse = await hiveforge.ai.streamToString({
  messages: [{ role: 'user', content: 'Summarize this document...' }],
  onChunk: (chunk) => process.stdout.write(chunk.content),
});
 
console.log('Final response:', fullResponse);

const result = await hiveforge.ai.embed({
  text: 'How do I reset my password?',
  model: 'text-embedding-3-small',
});
 
console.log(`Dimensions: ${result.dimensions}`);
console.log(`Tokens used: ${result.tokens_used}`);
console.log(`Embedding length: ${result.embeddings[0].length}`);

const result = await hiveforge.ai.embed({
  text: [
    'How do I reset my password?',
    'Where can I update my billing info?',
    'How to enable two-factor authentication',
  ],
});
 
// result.embeddings is an array of embedding arrays
for (const embedding of result.embeddings) {
  console.log(`Vector with ${embedding.length} dimensions`);
}

const quota = await hiveforge.ai.getQuota();
 
console.log(`Used: ${quota.used} tokens`);
console.log(`Limit: ${quota.limit ?? 'unlimited'}`);
console.log(`Remaining: ${quota.remaining ?? 'unlimited'}`);
console.log(`Resets at: ${quota.resets_at}`);
console.log(`Tier: ${quota.tier}`);

const models = await hiveforge.ai.getModels();
 
console.log(`Tier: ${models.tier}`);
console.log('Available:', models.available_models);
console.log('All models:', models.all_models);

if (hiveforge.ai.isEnabled()) {
  // Safe to make AI calls
}

const remaining = hiveforge.ai.getRemainingQuota();
if (remaining !== null && remaining < 1000) {
  console.warn('Running low on AI tokens');
}

import { HiveForgeClient, AIProxyException } from '@producthacker/hiveforge-sdk';
 
const hiveforge = new HiveForgeClient({
  deploymentId: process.env.HIVEFORGE_DEPLOYMENT_ID!,
  deploymentSecret: process.env.HIVEFORGE_DEPLOYMENT_SECRET!,
});
await hiveforge.initialize();
 
async function chat(userMessage: string, history: Array<{ role: string; content: string }>) {
  if (!hiveforge.ai.isEnabled()) {
    throw new Error('AI features are not available on your current plan.');
  }
 
  const remaining = hiveforge.ai.getRemainingQuota();
  if (remaining !== null && remaining < 100) {
    throw new Error('AI token quota nearly exhausted. Please upgrade your plan.');
  }
 
  try {
    const messages = [
      { role: 'system' as const, content: 'You are a helpful assistant.' },
      ...history.map(m => ({ role: m.role as 'user' | 'assistant', content: m.content })),
      { role: 'user' as const, content: userMessage },
    ];
 
    let response = '';
    for await (const chunk of hiveforge.ai.stream({ messages })) {
      response += chunk.content;
      // Update UI with chunk.content
    }
    return response;
  } catch (error) {
    if (error instanceof AIProxyException) {
      if (error.isQuotaExceeded) {
        // Redirect to upgrade page
        window.location.href = error.upgradeUrl ?? '/pricing';
      }
    }
    throw error;
  }
}