pretty

Author: nina-kollman

packages/sample-app/src/sample_associations.ts

@@ -112,7 +112,9 @@ async function customerServiceDemo() {
       });
 
       console.log("Customer 1 (cust-001):");
-      console.log(`Response: ${customer1Response.choices[0].message.content}\n`);
+      console.log(
+        `Response: ${customer1Response.choices[0].message.content}\n`,
+      );
 
       // Customer 2 - Update associations for new customer
       traceloop.Associations.set([
@@ -131,7 +133,9 @@ async function customerServiceDemo() {
       });
 
       console.log("Customer 2 (cust-002):");
-      console.log(`Response: ${customer2Response.choices[0].message.content}\n`);
+      console.log(
+        `Response: ${customer2Response.choices[0].message.content}\n`,
+      );
     },
   );
 }

packages/sample-app/src/sample_chatbot_interactive.ts

@@ -110,7 +110,9 @@ class InteractiveChatbot {
           parameters: z.object({
             expression: z
               .string()
-              .describe("The mathematical expression to evaluate (e.g., '2 + 2' or '10 * 5')"),
+              .describe(
+                "The mathematical expression to evaluate (e.g., '2 + 2' or '10 * 5')",
+              ),
           }),
           execute: async ({ expression }) => {
             try {
@@ -130,16 +132,25 @@ class InteractiveChatbot {
           description:
             "Get the current weather for a location. Use this when users ask about weather conditions.",
           parameters: z.object({
-            location: z.string().describe("The city and country, e.g., 'London, UK'"),
+            location: z
+              .string()
+              .describe("The city and country, e.g., 'London, UK'"),
           }),
           execute: async ({ location }) => {
             console.log(
               `\n${colors.yellow}🔧 Weather: Checking weather for ${location}${colors.reset}`,
             );
             // Simulated weather data
-            const weatherConditions = ["sunny", "cloudy", "rainy", "partly cloudy"];
+            const weatherConditions = [
+              "sunny",
+              "cloudy",
+              "rainy",
+              "partly cloudy",
+            ];
             const condition =
-              weatherConditions[Math.floor(Math.random() * weatherConditions.length)];
+              weatherConditions[
+                Math.floor(Math.random() * weatherConditions.length)
+              ];
             const temperature = Math.floor(Math.random() * 30) + 10; // 10-40°C
             return {
               location,

packages/traceloop-sdk/src/lib/tracing/associations.ts

@@ -17,7 +17,7 @@ export enum AssociationProperty {
  * Use this to check if a property should be set directly or with the TRACELOOP_ASSOCIATION_PROPERTIES prefix.
  */
 export const STANDARD_ASSOCIATION_PROPERTIES = new Set<string>(
-  Object.values(AssociationProperty)
+  Object.values(AssociationProperty),
 );
 
 /**
@@ -48,7 +48,9 @@ export class Associations {
     // Get current associations from context or create empty object
     const existingAssociations = otelContext
       .active()
-      .getValue(ASSOCATION_PROPERTIES_KEY) as Record<string, string> | undefined;
+      .getValue(ASSOCATION_PROPERTIES_KEY) as
+      | Record<string, string>
+      | undefined;
     const currentAssociations: Record<string, string> = existingAssociations
       ? { ...existingAssociations }
       : {};
@@ -65,7 +67,7 @@ export class Associations {
 
     // Set the new context as active using the context manager
     // This is the equivalent of Python's attach(set_value(...))
-    const contextManager = (otelContext as any)['_getContextManager']();
+    const contextManager = (otelContext as any)["_getContextManager"]();
     if (
       contextManager &&
       contextManager instanceof AsyncLocalStorageContextManager

packages/traceloop-sdk/test/associations.test.ts

@@ -111,10 +111,7 @@ describe("Test Associations API", () => {
     assert.ok(chatSpan);
 
     // Check that the association is set on both workflow and LLM spans
-    assert.strictEqual(
-      workflowSpan.attributes["conversation_id"],
-      "conv-123",
-    );
+    assert.strictEqual(workflowSpan.attributes["conversation_id"], "conv-123");
     assert.strictEqual(
       chatSpan.attributes[
         `${SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.conversation_id`
@@ -276,15 +273,6 @@ describe("Test Associations API", () => {
   });
 
   it("should propagate associations to all child spans", async () => {
-    @traceloop.task({ name: "subtask" })
-    async function subtask() {
-      const chatCompletion = await openai.chat.completions.create({
-        messages: [{ role: "user", content: "Child task message" }],
-        model: "gpt-3.5-turbo",
-      });
-      return chatCompletion.choices[0].message.content;
-    }
-
     const result = await traceloop.withWorkflow(
       { name: "test_child_propagation" },
       async () => {
@@ -295,7 +283,16 @@ describe("Test Associations API", () => {
         ]);
 
         // Call a child task
-        const taskResult = await subtask();
+        const taskResult = await traceloop.withTask(
+          { name: "subtask" },
+          async () => {
+            const chatCompletion = await openai.chat.completions.create({
+              messages: [{ role: "user", content: "Child task message" }],
+              model: "gpt-3.5-turbo",
+            });
+            return chatCompletion.choices[0].message.content;
+          },
+        );
 
         return taskResult;
       },
@@ -327,10 +324,16 @@ describe("Test Associations API", () => {
     );
     assert.strictEqual(workflowSpan.attributes["user_id"], "user-propagate");
 
-    assert.strictEqual(taskSpan.attributes["conversation_id"], "conv-propagate");
+    assert.strictEqual(
+      taskSpan.attributes["conversation_id"],
+      "conv-propagate",
+    );
     assert.strictEqual(taskSpan.attributes["user_id"], "user-propagate");
 
-    assert.strictEqual(chatSpan.attributes["conversation_id"], "conv-propagate");
+    assert.strictEqual(
+      chatSpan.attributes["conversation_id"],
+      "conv-propagate",
+    );
     assert.strictEqual(chatSpan.attributes["user_id"], "user-propagate");
   });