fix(sdk): allow passing a function to the decorator

traceloop · Apr 1, 2024 · 9c967e3 · 9c967e3
1 parent 956bad4
commit 9c967e3
Show file tree

Hide file tree

Showing 4 changed files with 598 additions and 12 deletions.
diff --git a/packages/sample-app/src/sample_decorators.ts b/packages/sample-app/src/sample_decorators.ts
@@ -23,7 +23,9 @@ class SampleOpenAI {
     return chatCompletion.choices[0].message.content;
   }
 
-  @traceloop.workflow({ name: "sample_completion" })
+  @traceloop.workflow((thisArg) => ({
+    name: `sample_${(thisArg as SampleOpenAI).model}`,
+  }))
   async completion(jokeSubject: string) {
     const completion = await openai.completions.create({
       prompt: `Tell me a joke about ${jokeSubject}`,