Enterprise Setup Guide

async function getEnterpriseToken() {
  const response = await fetch('https://api.context.ai/v1/auth/enterprise/token', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
      'X-API-Key': process.env.CONTEXT_API_KEY
    },
    body: JSON.stringify({
      grant_type: 'client_credentials',
      client_id: process.env.CONTEXT_ENTERPRISE_ORG_ID,
      client_secret: process.env.CONTEXT_API_SECRET,
      scope: 'context-engine:read context-engine:write background-agents:read background-agents:write'
    })
  });

  const { access_token, expires_in } = await response.json();
  
  // Store token securely for reuse
  return {
    token: access_token,
    expiresAt: Date.now() + (expires_in * 1000)
  };
}

const ContextEngine = require('@context-ai/context-engine-sdk');

const contextEngine = new ContextEngine({
  apiKey: process.env.CONTEXT_API_KEY,
  organizationId: process.env.CONTEXT_ENTERPRISE_ORG_ID,
  environment: 'production',
  timeout: 30000,
  retryAttempts: 3
});

// Test connection
async function testContextEngine() {
  try {
    const health = await contextEngine.health.check();
    console.log('Context Engine connection successful:', health);
  } catch (error) {
    console.error('Context Engine connection failed:', error);
  }
}

testContextEngine();

async function setupDataSources() {
  // Configure financial data integration
  await contextEngine.dataSources.create({
    name: "enterprise_financial_data",
    type: "database",
    connection: {
      type: "postgresql",
      host: "your-db-host.com",
      port: 5432,
      database: "financial_data",
      username: "${DB_USERNAME}",
      password: "${DB_PASSWORD}",
      ssl: true
    },
    indexing_schedule: "0 2 * * *", // Daily at 2 AM
    security: {
      encryption_at_rest: true,
      access_control: "role_based"
    }
  });

  // Configure document repository
  await contextEngine.dataSources.create({
    name: "corporate_documents",
    type: "sharepoint",
    connection: {
      tenant_id: "${SHAREPOINT_TENANT_ID}",
      client_id: "${SHAREPOINT_CLIENT_ID}",
      client_secret: "${SHAREPOINT_CLIENT_SECRET}",
      site_url: "https://yourcompany.sharepoint.com/sites/corporate"
    },
    indexing_options: {
      file_types: ["pdf", "docx", "xlsx", "pptx"],
      extract_metadata: true,
      ocr_enabled: true
    }
  });

  console.log('Data sources configured successfully');
}

// test/integration.test.js
const test = require('ava');
const ContextEnterprise = require('@context-ai/enterprise-sdk');

const client = new ContextEnterprise({
  apiKey: process.env.CONTEXT_API_KEY,
  organizationId: process.env.CONTEXT_ENTERPRISE_ORG_ID,
  environment: 'production'
});

test('Context Engine query integration', async t => {
  const response = await client.contextEngine.query({
    query: "Test integration query",
    sources: ["test_data"],
    processing_mode: "fast"
  });

  t.is(response.status, 'completed');
  t.truthy(response.results);
});

test('Background Agent task submission', async t => {
  const task = await client.backgroundAgents.submitTask({
    task_type: "research_agent",
    priority: 5,
    input: {
      query: "Test task submission",
      sources: ["test_data"]
    }
  });

  t.truthy(task.task_id);
  t.is(task.status, 'accepted');
});

test('WebSocket connection', async t => {
  const ws = await client.backgroundAgents.connectWebSocket();
  
  return new Promise((resolve) => {
    ws.on('authenticated', () => {
      t.pass('WebSocket authentication successful');
      ws.close();
      resolve();
    });
    
    ws.on('error', (error) => {
      t.fail(`WebSocket connection failed: ${error}`);
      resolve();
    });
  });
});

{
  "dashboard": {
    "title": "Context Enterprise Monitoring",
    "panels": [
      {
        "title": "Background Agent Task Throughput",
        "type": "graph",
        "targets": [
          {
            "expr": "rate(background_agent_tasks_completed_total[5m])",
            "legendFormat": "Tasks/sec"
          }
        ]
      },
      {
        "title": "Context Engine Response Time",
        "type": "graph",
        "targets": [
          {
            "expr": "histogram_quantile(0.95, rate(context_engine_request_duration_seconds_bucket[5m]))",
            "legendFormat": "95th percentile"
          }
        ]
      }
    ]
  }
}