Group traces

from keywordsai_tracing.decorators import workflow, task
from keywordsai_tracing.main import KeywordsAITelemetry
from keywordsai_tracing.contexts.span import keywordsai_span_attributes

k_tl = KeywordsAITelemetry()

@task(name="joke_creation")
def create_joke():
    # This is the joke creation task

@workflow(name="pirate_joke_generator")
def joke_workflow():
    # This is the joke creation workflow

@workflow(name="pirate_joke_plus_audience_reactions")
def pirate_joke_plus_audience():
    with keywordsai_span_attributes(
        keywordsai_params={
            "trace_group_identifier": "pirate_joke_id"
        }): 
        joke = joke_workflow() # This show case the basic workflow usage and compatibility with the OpenAI SDK
    return "python"

@workflow(name="pirate_joke_plus_audience_reactions")
def another_workflow():    
  with keywordsai_span_attributes(
        keywordsai_params={
            "trace_group_identifier": "pirate_joke_id" # same trace group identifier as the previous workflow
        }):
        # This is the another workflow
    return "python"

import { KeywordsAITelemetry } from "@keywordsai/tracing";

const keywordsAI = new KeywordsAITelemetry({
    // any config
});

async function testTask() {// any task or workflow
}

async function testWorkflow() {
  return await keywordsAI.withWorkflow({ name: "test" }, async () => {
    keywordsAI.withKeywordsAISpanAttributes(
      async () => {
        testTask();
      },
      {
        trace_group_identifier: "some_trace_group_identifier",
      }
    );
    return "test workflow";
  });
}

testWorkflow();

async function another_testWorkflow() {
  return await keywordsAI.withWorkflow({ name: "test_another_workflow" }, async () => {
    keywordsAI.withKeywordsAISpanAttributes(
      async () => {
     // any task or workflow
      },
      {
        trace_group_identifier: "some_trace_group_identifier", // same trace group identifier as the previous workflow
      }
    );
    return "test workflow no.2";
  });
}

another_testWorkflow();