88 - Do not require workspace ID for creating spans

MagnivOrg · Aug 19, 2024 · fb19148 · fb19148
1 parent 10729f2
commit fb19148
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 15 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -57,34 +57,27 @@ export class PromptLayer {
   group: GroupManager;
   track: TrackManager;
   enableTracing: boolean;
-  workspaceId?: number;
   wrapWithSpan: typeof wrapWithSpan;
 
   constructor({
     apiKey = process.env.PROMPTLAYER_API_KEY,
     enableTracing = false,
-    workspaceId,
   }: ClientOptions = {}) {
     if (apiKey === undefined) {
       throw new Error(
         "PromptLayer API key not provided. Please set the PROMPTLAYER_API_KEY environment variable or pass the api_key parameter."
       );
     }
 
-    if (enableTracing && workspaceId === undefined) {
-      throw new Error("Please set a workspaceId to enable tracing.")
-    }
-
     this.apiKey = apiKey;
     this.enableTracing = enableTracing;
     this.templates = new TemplateManager(apiKey);
     this.group = new GroupManager(apiKey);
     this.track = new TrackManager(apiKey);
-    this.workspaceId = workspaceId;
     this.wrapWithSpan = wrapWithSpan;
 
-    if (enableTracing && workspaceId) {
-      setupTracing(enableTracing, workspaceId);
+    if (enableTracing) {
+      setupTracing(enableTracing);
     }
   }
 

diff --git a/src/span-exporter.ts b/src/span-exporter.ts
@@ -7,13 +7,11 @@ class PromptLayerSpanExporter implements SpanExporter {
   private apiKey: string | undefined;
   private enableTracing: boolean;
   private url: string;
-  private workspaceId: number;
 
-  constructor(enableTracing: boolean, workspaceId: number) {
+  constructor(enableTracing: boolean) {
     this.apiKey = process.env.PROMPTLAYER_API_KEY;
     this.enableTracing = enableTracing;
     this.url = `${URL_API_PROMPTLAYER}/spans-bulk`;
-    this.workspaceId = workspaceId;
   }
 
   private attributesToObject(attributes: Attributes | undefined): Record<string, any> {
@@ -92,7 +90,6 @@ class PromptLayerSpanExporter implements SpanExporter {
       },
       body: JSON.stringify({
         spans: requestData,
-        workspace_id: this.workspaceId,
       }),
     })
       .then(response => {

diff --git a/src/tracing.ts b/src/tracing.ts
@@ -7,9 +7,9 @@ export const getTracer = (name: string = 'promptlayer-tracer') => {
   return opentelemetry.trace.getTracer(name);
 }
 
-export const setupTracing = (enableTracing: boolean, workspaceId: number) => {
+export const setupTracing = (enableTracing: boolean) => {
   const provider = new NodeTracerProvider();
-  const exporter = new PromptLayerSpanExporter(enableTracing, workspaceId);
+  const exporter = new PromptLayerSpanExporter(enableTracing);
   const processor = new SimpleSpanProcessor(exporter);
   provider.addSpanProcessor(processor);
   provider.register();