@observyze/sdk 0.1.0 → 0.1.2

package/examples/auto-instrumentation.ts

@@ -1,210 +0,0 @@
-/**
- * Example: Auto-Instrumentation with OpenAI and Anthropic
- * 
- * This example demonstrates how to use Observyze's auto-instrumentation
- * to automatically capture traces from OpenAI and Anthropic API calls.
- */
-
-import { ObservyzeClient } from '../src'
-
-// Mock OpenAI and Anthropic clients for demonstration
-// In a real application, you would import these from their respective packages:
-// import OpenAI from 'openai'
-// import Anthropic from '@anthropic-ai/sdk'
-
-async function main() {
-  // Initialize Observyze SDK
-  const nw = new ObservyzeClient({
-    apiKey: process.env.Observyze_API_KEY || 'demo-key',
-    organizationId: 'demo-org',
-    projectId: 'demo-project',
-    endpoint: 'http://localhost:3001',
-    debug: true,
-    dryRun: true // Set to false in production
-  })
-
-  console.log('🚀 Observyze SDK initialized\n')
-
-  // Example 1: OpenAI Auto-Instrumentation
-  console.log('📝 Example 1: OpenAI Auto-Instrumentation')
-  console.log('─'.repeat(50))
-  
-  // Create a mock OpenAI client
-  const mockOpenAI = {
-    chat: {
-      completions: {
-        create: async (params: any) => {
-          console.log('  → Calling OpenAI API...')
-          // Simulate API call
-          await new Promise(resolve => setTimeout(resolve, 100))
-          return {
-            id: 'chatcmpl-123',
-            model: params.model,
-            choices: [
-              {
-                message: {
-                  role: 'assistant',
-                  content: 'Hello! I am an AI assistant. How can I help you today?'
-                },
-                finish_reason: 'stop'
-              }
-            ],
-            usage: {
-              prompt_tokens: 15,
-              completion_tokens: 25,
-              total_tokens: 40
-            }
-          }
-        }
-      }
-    }
-  }
-
-  // Wrap the OpenAI client
-  nw.wrap(mockOpenAI)
-  console.log('  ✓ OpenAI client wrapped')
-
-  // Make a call - it will be automatically traced!
-  const openaiResponse = await mockOpenAI.chat.completions.create({
-    model: 'gpt-4',
-    messages: [
-      { role: 'user', content: 'Hello, how are you?' }
-    ],
-    temperature: 0.7
-  })
-
-  console.log('  ✓ Response received:', openaiResponse.choices[0].message.content)
-  console.log('  ✓ Trace automatically captured!\n')
-
-  // Example 2: Anthropic Auto-Instrumentation
-  console.log('📝 Example 2: Anthropic Auto-Instrumentation')
-  console.log('─'.repeat(50))
-
-  // Create a mock Anthropic client
-  const mockAnthropic = {
-    messages: {
-      create: async (params: any) => {
-        console.log('  → Calling Anthropic API...')
-        // Simulate API call
-        await new Promise(resolve => setTimeout(resolve, 100))
-        return {
-          id: 'msg_123',
-          type: 'message',
-          role: 'assistant',
-          content: [
-            {
-              type: 'text',
-              text: 'Hello! I am Claude, an AI assistant created by Anthropic.'
-            }
-          ],
-          model: params.model,
-          stop_reason: 'end_turn',
-          usage: {
-            input_tokens: 20,
-            output_tokens: 30
-          }
-        }
-      }
-    }
-  }
-
-  // Wrap the Anthropic client
-  nw.wrap(mockAnthropic)
-  console.log('  ✓ Anthropic client wrapped')
-
-  // Make a call - it will be automatically traced!
-  const anthropicResponse = await mockAnthropic.messages.create({
-    model: 'claude-3-opus-20240229',
-    max_tokens: 1024,
-    messages: [
-      { role: 'user', content: 'Hello, Claude!' }
-    ]
-  })
-
-  console.log('  ✓ Response received:', anthropicResponse.content[0].text)
-  console.log('  ✓ Trace automatically captured!\n')
-
-  // Example 3: Streaming Response
-  console.log('📝 Example 3: Streaming Response (OpenAI)')
-  console.log('─'.repeat(50))
-
-  // Create a mock streaming OpenAI client
-  const mockStreamingOpenAI = {
-    chat: {
-      completions: {
-        create: async (params: any) => {
-          console.log('  → Starting streaming call...')
-          // Return an async generator that simulates streaming
-          return {
-            async *[Symbol.asyncIterator]() {
-              const chunks = ['Hello', ' there', '!', ' How', ' can', ' I', ' help', '?']
-              for (const chunk of chunks) {
-                await new Promise(resolve => setTimeout(resolve, 50))
-                yield {
-                  id: 'chatcmpl-stream-123',
-                  model: params.model,
-                  choices: [
-                    {
-                      delta: { content: chunk },
-                      finish_reason: null
-                    }
-                  ]
-                }
-              }
-              // Final chunk
-              yield {
-                id: 'chatcmpl-stream-123',
-                model: params.model,
-                choices: [
-                  {
-                    delta: {},
-                    finish_reason: 'stop'
-                  }
-                ]
-              }
-            }
-          }
-        }
-      }
-    }
-  }
-
-  // Wrap the streaming client
-  nw.wrap(mockStreamingOpenAI)
-  console.log('  ✓ Streaming client wrapped')
-
-  // Make a streaming call
-  const stream = await mockStreamingOpenAI.chat.completions.create({
-    model: 'gpt-4',
-    messages: [{ role: 'user', content: 'Hello!' }],
-    stream: true
-  })
-
-  console.log('  → Streaming response: ', { newline: false })
-  for await (const chunk of stream) {
-    if (chunk.choices[0]?.delta?.content) {
-      process.stdout.write(chunk.choices[0].delta.content)
-    }
-  }
-  console.log('\n  ✓ Stream completed and trace captured!\n')
-
-  // Show buffer status
-  console.log('📊 SDK Status')
-  console.log('─'.repeat(50))
-  console.log(`  Buffered traces: ${nw.bufferSize}`)
-  console.log('  ✓ All traces will be automatically flushed\n')
-
-  // Flush and shutdown
-  console.log('🔄 Flushing traces...')
-  await nw.flush()
-  console.log('  ✓ Traces flushed')
-
-  console.log('👋 Shutting down SDK...')
-  await nw.shutdown()
-  console.log('  ✓ SDK shutdown complete\n')
-
-  console.log('✨ Demo complete! In production, traces would be sent to Observyze.')
-}
-
-// Run the example
-main().catch(console.error)