Update chain_app.py

chain_app.py

@@ -60,6 +60,14 @@ async def set_starters():
             )
         ]
 
+
+@cl.step(type="tool")
+async def tool():
+    # Simulate a running task
+    await cl.sleep(2)
+
+    return "Response from the tool!"
+
 @cl.set_chat_profiles
 async def chat_profile():
     return [
@@ -2587,14 +2595,17 @@ async def main(message: cl.Message):
         ).send()
 
     elif chat_profile == 'Yi-1.5-34B':
+        final_answer = await cl.Message(content="").send()
         client = InferenceClient(
             "01-ai/Yi-1.5-34B-Chat",
             token=f'{hf_token_llama_3_1}',
         )
+        tool_res = await tool()
         result = client.chat_completion(
         	messages=[{"role": "user", "content": message.content}],
             max_tokens=500,
         )
+        
         await cl.Message(
             content=result.choices[0].message.content,
         ).send()