display trace url

app.py

@@ -58,6 +58,9 @@ if "last_output" in st.session_state:
     text = st.session_state["last_text"]
     st.write(text)
 
+    url = output["url"]
+    st.info(f"The weave trace url: {url}", icon="ℹ️")
+
     # use the weave client to retrieve the call and attach feedback
     st.button(":thumbsup:",   on_click=lambda: weave_client.call(output['call_id']).feedback.add_reaction("👍"), key='up')
     st.button(":thumbsdown:", on_click=lambda: weave_client.call(output['call_id']).feedback.add_reaction("👎"), key='down')

rag/rag.py

@@ -150,7 +150,11 @@ class SimpleRAGPipeline(weave.Model):
     @weave.op()
     def predict(self, question: str):
         response = self.query_engine.query(question)
-        return {"response": response, 'call_id': weave.get_current_call().id}
+        return {
+            "response": response,
+            'call_id': weave.get_current_call().id,
+            "url": weave.get_current_call().ui_url,
+        }
 
 
 if __name__ == "__main__":