Update app.py

app.py

@@ -58,7 +58,7 @@ def handle_vision_call(msg, image_history):
         image_history.clear()
         return stream
 
-@traceable(run_type="llm", name="gpt 4 turbo call")
+@traceable(run_type="llm", name="gpt 3 turbo call")
 async def gpt_call(message_history: list = []):
     client = OpenAI()
 
@@ -70,7 +70,7 @@ async def gpt_call(message_history: list = []):
     
     return stream
 
-
+@traceable(run_type="llm", name="gpt 4 turbo vision call")
 def gpt_vision_call(image_history: list = []):
     client = OpenAI()
   
@@ -92,6 +92,7 @@ def start_chat():
     cl.user_session.set("image_history", [{"role": "system", "content": "You are a helpful assistant."}])
 
 @cl.on_message
+@traceable(run_type="chain", name="message")
 async def on_message(msg: cl.Message):
     message_history = cl.user_session.get("message_history")
     image_history = cl.user_session.get("image_history")