fix(ai): readme (#357)

posthog-ai/README.md

@@ -5,56 +5,43 @@ Initial Typescript SDK for LLM Observability
 ## Installation
 
 ```bash
-yarn add @posthog/posthog-ai
+npm install @posthog/ai
 ```
 
 ## Usage
 
-### Before
-
-```typescript
-import { OpenAI } from 'openai'
-
-const client = new OpenAI({
-  apiKey: process.env.OPENAI_API_KEY || '',
-})
-
-await client.chat.completions.create({
-  model: 'gpt-4',
-  messages: [{ role: 'user', content: 'Hello, world!' }],
-})
-```
-
-### After
-
 ```typescript
 import { OpenAI } from '@posthog/ai'
 import { PostHog } from 'posthog-node'
 
 const phClient = new PostHog(
-  process.env.POSTHOG_API_KEY, {
-    host: process.env.POSTHOG_HOST || 'https://us.posthog.com',
-  }
-})
+  '<YOUR_PROJECT_API_KEY>',
+  { host: 'https://us.i.posthog.com' }
+);
 
 const client = new OpenAI({
-  apiKey: process.env.OPENAI_API_KEY || '',
+  apiKey: '<YOUR_OPENAI_API_KEY>',
   posthog: phClient,
-})
+});
 
-await client.chat.completions.create({
-  model: 'gpt-4',
-  messages: [{ role: 'user', content: 'Hello, world!' }],
-  posthogDistinctId: 'test-user-id',
-  posthogProperties: {
-    testProperty: 'testValue',
-  }
-})
+const completion = await client.chat.completions.create({
+    model: "gpt-3.5-turbo",
+    messages: [{ role: "user", content: "Tell me a fun fact about hedgehogs" }],
+    posthogDistinctId: "user_123", // optional
+    posthogTraceId: "trace_123", // optional
+    posthogProperties: { conversation_id: "abc123", paid: true }, //optional
+    posthogGroups: { company: "company_id_in_your_db" }, // optional 
+    posthogPrivacyMode: false // optional
+});
+
+console.log(completion.choices[0].message.content)
 
 // YOU HAVE TO HAVE THIS OR THE CLIENT MAY NOT SEND EVENTS
 await phClient.shutdown()
 ```
 
+LLM Observability [docs](https://posthog.com/docs/ai-engineering/observability)
+
 Please see the main [PostHog docs](https://www.posthog.com/docs).
 
 ## Questions?

posthog-ai/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/ai",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "PostHog Node.js AI integrations",
   "repository": {
     "type": "git",