Mount cerebras instrumentation in SDK and add sample

Author: airhorns

package-lock.json

@@ -40,6 +40,7 @@
     "@anthropic-ai/sdk": "^0.32.1",
     "@aws-sdk/client-bedrock-runtime": "^3.709.0",
     "@azure/openai": "^1.0.0-beta.13",
+    "@cerebras/cerebras_cloud_sdk": "^1.29.0",
     "@google-cloud/aiplatform": "^3.34.0",
     "@google-cloud/vertexai": "^1.9.2",
     "@langchain/community": "^0.3.18",

packages/sample-app/package.json

@@ -0,0 +1,26 @@
+import * as traceloop from "@traceloop/node-server-sdk";
+import { Cerebras } from "@cerebras/cerebras_cloud_sdk";
+
+traceloop.initialize({
+  appName: "sample_cerebras",
+  apiKey: process.env.TRACELOOP_API_KEY,
+  disableBatch: true,
+});
+const cerebras = new Cerebras({});
+
+async function main() {
+  const completion = await cerebras.chat.completions.create({
+    max_tokens: 1024,
+    model: "llama-3.1-8b",
+    messages: [
+      {
+        role: "user",
+        content: "How does a court case get to the Supreme Court?",
+      },
+    ],
+  });
+
+  console.log((completion.choices as any)[0].message.content);
+}
+
+main();

packages/sample-app/src/sample_cerebras.ts

@@ -41,6 +41,7 @@
     "@traceloop/instrumentation-anthropic": "^0.12.0",
     "@traceloop/instrumentation-azure": "^0.12.0",
     "@traceloop/instrumentation-bedrock": "^0.12.0",
+    "@traceloop/instrumentation-cerebras": "^0.1.0",
     "@traceloop/instrumentation-chromadb": "^0.12.0",
     "@traceloop/instrumentation-cohere": "^0.12.0",
     "@traceloop/instrumentation-langchain": "^0.12.0",
@@ -65,6 +66,7 @@
     "@anthropic-ai/sdk": "^0.32.1",
     "@aws-sdk/client-bedrock-runtime": "^3.709.0",
     "@azure/openai": "^1.0.0-beta.13",
+    "@cerebras/cerebras_cloud_sdk": "^1.29.0",
     "@google-cloud/aiplatform": "^3.34.0",
     "@google-cloud/vertexai": "^1.9.2",
     "@pinecone-database/pinecone": "^2.2.2",

packages/traceloop-sdk/package.json

@@ -17,6 +17,7 @@ import type * as VectorStoreModule from "@langchain/core/vectorstores";
 import type * as llamaindex from "llamaindex";
 import type * as chromadb from "chromadb";
 import type * as qdrant from "@qdrant/js-client-rest";
+import type * as cerebras from "@cerebras/cerebras_cloud_sdk";
 
 /**
  * Options for initializing the Traceloop SDK.
@@ -111,6 +112,7 @@ export interface InitializeOptions {
     llamaIndex?: typeof llamaindex;
     chromadb?: typeof chromadb;
     qdrant?: typeof qdrant;
+    cerebras?: typeof cerebras;
   };
 
   /**

packages/traceloop-sdk/src/lib/interfaces/initialize-options.interface.ts

@@ -38,6 +38,7 @@ import { LangChainInstrumentation } from "@traceloop/instrumentation-langchain";
 import { ChromaDBInstrumentation } from "@traceloop/instrumentation-chromadb";
 import { QdrantInstrumentation } from "@traceloop/instrumentation-qdrant";
 import { TogetherInstrumentation } from "@traceloop/instrumentation-together";
+import { CerebrasInstrumentation } from "@traceloop/instrumentation-cerebras";
 
 let _sdk: NodeSDK;
 let _spanProcessor: SimpleSpanProcessor | BatchSpanProcessor;
@@ -54,6 +55,7 @@ let pineconeInstrumentation: PineconeInstrumentation | undefined;
 let chromadbInstrumentation: ChromaDBInstrumentation | undefined;
 let qdrantInstrumentation: QdrantInstrumentation | undefined;
 let togetherInstrumentation: TogetherInstrumentation | undefined;
+let cerebrasInstrumentation: CerebrasInstrumentation | undefined;
 
 const instrumentations: Instrumentation[] = [];
 
@@ -111,6 +113,9 @@ export const initInstrumentations = () => {
 
   togetherInstrumentation = new TogetherInstrumentation({ exceptionLogger });
   instrumentations.push(togetherInstrumentation);
+
+  cerebrasInstrumentation = new CerebrasInstrumentation({ exceptionLogger });
+  instrumentations.push(cerebrasInstrumentation);
 };
 
 export const manuallyInitInstrumentations = (
@@ -218,6 +223,12 @@ export const manuallyInitInstrumentations = (
     instrumentations.push(togetherInstrumentation);
     togetherInstrumentation.manuallyInstrument(instrumentModules.together);
   }
+
+  if (instrumentModules?.cerebras) {
+    cerebrasInstrumentation = new CerebrasInstrumentation({ exceptionLogger });
+    instrumentations.push(cerebrasInstrumentation);
+    cerebrasInstrumentation.manuallyInstrument(instrumentModules.cerebras);
+  }
 };
 
 /**