fix: metrics on posthog (#6024)

Co-authored-by: Zane Staggs <zane@squareup.com>
Co-authored-by: Zane <75694352+zanesq@users.noreply.github.com>

Author: michaelneale

Cargo.lock

@@ -248,6 +248,8 @@ pub struct SessionSettings {
 }
 
 pub async fn build_session(session_config: SessionBuilderConfig) -> CliSession {
+    goose::posthog::set_session_context("cli", session_config.resume);
+
     let config = Config::global();
 
     let (saved_provider, saved_model_config) = if session_config.resume {

crates/goose-cli/src/session/builder.rs

@@ -105,6 +105,8 @@ async fn start_agent(
     State(state): State<Arc<AppState>>,
     Json(payload): Json<StartAgentRequest>,
 ) -> Result<Json<Session>, ErrorResponse> {
+    goose::posthog::set_session_context("desktop", false);
+
     let StartAgentRequest {
         working_dir,
         recipe,
@@ -197,6 +199,8 @@ async fn resume_agent(
     State(state): State<Arc<AppState>>,
     Json(payload): Json<ResumeAgentRequest>,
 ) -> Result<Json<Session>, ErrorResponse> {
+    goose::posthog::set_session_context("desktop", true);
+
     let session = SessionManager::get_session(&payload.session_id, true)
         .await
         .map_err(|err| {

crates/goose-server/src/routes/agent.rs

@@ -111,6 +111,7 @@ shellexpand = "3.1.1"
 indexmap = "2.12.0"
 ignore = "0.4.25"
 which = "8.0.0"
+posthog-rs = "0.3.7"
 
 
 [target.'cfg(target_os = "windows")'.dependencies]

crates/goose/Cargo.toml

@@ -1175,7 +1175,8 @@ impl Agent {
                                 no_tools_called = false;
                             }
                         }
-                        Err(ProviderError::ContextLengthExceeded(_error_msg)) => {
+                        Err(ref provider_err @ ProviderError::ContextLengthExceeded(_)) => {
+                            crate::posthog::emit_error(provider_err.telemetry_type());
                             yield AgentEvent::Message(
                                 Message::assistant().with_system_notification(
                                     SystemNotificationType::InlineMessage,
@@ -1209,11 +1210,12 @@ impl Agent {
                                 }
                             }
                         }
-                        Err(e) => {
-                            error!("Error: {}", e);
+                        Err(ref provider_err) => {
+                            crate::posthog::emit_error(provider_err.telemetry_type());
+                            error!("Error: {}", provider_err);
                             yield AgentEvent::Message(
                                 Message::assistant().with_text(
-                                    format!("Ran into this error: {e}.\n\nPlease retry if you think this is a transient or recoverable error.")
+                                    format!("Ran into this error: {provider_err}.\n\nPlease retry if you think this is a transient or recoverable error.")
                                 )
                             );
                             break;

crates/goose/src/agents/agent.rs

@@ -38,9 +38,9 @@ impl ToolRouteManager {
 
     pub async fn record_tool_requests(&self, requests: &[ToolRequest]) {
         let selector = self.router_tool_selector.lock().await.clone();
-        if let Some(selector) = selector {
-            for request in requests {
-                if let Ok(tool_call) = &request.tool_call {
+        for request in requests {
+            if let Ok(tool_call) = &request.tool_call {
+                if let Some(ref selector) = selector {
                     if let Err(e) = selector.record_tool_call(&tool_call.name).await {
                         error!("Failed to record tool call: {}", e);
                     }

crates/goose/src/agents/tool_route_manager.rs

@@ -9,6 +9,7 @@ pub mod mcp_utils;
 pub mod model;
 pub mod oauth;
 pub mod permission;
+pub mod posthog;
 pub mod prompt_template;
 pub mod providers;
 pub mod recipe;