Update trace.py

trace.py

@@ -4,9 +4,6 @@ from wandb.sdk.data_types.trace_tree import Trace
 
 WANDB_API_KEY = os.environ["WANDB_API_KEY"]
 
-AGENT_LANGCHAIN  = "LangChain"
-AGENT_LLAMAINDEX = "LlamaIndex"
-
 def trace_wandb(config,
                 agent_option,
                 prompt,
@@ -17,20 +14,12 @@ def trace_wandb(config,
                 end_time_ms):
     wandb.init(project = "openai-llm-agent")
 
-    if (agent_option == AGENT_LANGCHAIN):
-        prompt_template = os.environ["LANGCHAIN_TEMPLATE"]
-    elif (agent_option == AGENT_LLAMAINDEX):
-        prompt_template = os.environ["LLAMAINDEX_TEMPLATE"]
-    else:
-        prompt_template = os.environ["TEMPLATE"]
-                    
     trace = Trace(
         kind = "LLM",
         name = "Real-Time Reasoning Application",
         status_code = "success" if (str(err_msg) == "") else "error",
         status_message = str(err_msg),
         inputs = {"prompt": prompt,
-                  "prompt_template": prompt_template,
                   "agent_option": agent_option,
                   "config": str(config)
                  } if (str(err_msg) == "") else {},