Spanicin/agentic-ai

1import os, wandb2 3from wandb.sdk.data_types.trace_tree import Trace4 5WANDB_API_KEY = os.environ["WANDB_API_KEY"]6 7def trace_wandb(config,8                agent_option,9                prompt,10                completion,11                result,12                callback,13                err_msg,14                start_time_ms,15                end_time_ms):16    wandb.init(project = "openai-llm-agent")17 18    trace = Trace(19        kind = "LLM",20        name = "Real-Time Reasoning Application",21        status_code = "success" if (str(err_msg) == "") else "error",22        status_message = str(err_msg),23        inputs = {"prompt": prompt,24                  "agent_option": agent_option,25                  "config": str(config)26                 } if (str(err_msg) == "") else {},27        outputs = {"result": str(result),28                   "callback": str(callback),29                   "completion": str(completion)30                  } if (str(err_msg) == "") else {},31        start_time_ms = start_time_ms,32        end_time_ms = end_time_ms33    )34    35    trace.log("evaluation")36                    37    wandb.finish()