chore: remove duplicate and unused API code from the agent crate (#3342)

Author: brandonskiser

crates/agent/src/agent/agent_loop/model.rs

@@ -6,10 +6,6 @@ use std::sync::{
 use std::time::Duration;
 
 use futures::Stream;
-use serde::{
-    Deserialize,
-    Serialize,
-};
 use tokio::sync::mpsc;
 use tokio_stream::wrappers::ReceiverStream;
 use tokio_util::sync::CancellationToken;
@@ -27,7 +23,6 @@ use super::types::{
     Message,
     ToolSpec,
 };
-use crate::agent::rts::RtsModel;
 
 /// Represents a backend implementation for a converse stream compatible API.
 ///
@@ -51,59 +46,6 @@ pub trait Model: std::fmt::Debug + Send + Sync + 'static {
     }
 }
 
-/// The supported backends
-#[derive(Debug, Clone)]
-pub enum Models {
-    Rts(RtsModel),
-    Test(MockModel),
-}
-
-impl Models {
-    pub fn state(&self) -> ModelsState {
-        match self {
-            Models::Rts(v) => ModelsState::Rts {
-                conversation_id: Some(v.conversation_id().to_string()),
-                model_id: v.model_id().map(String::from),
-            },
-            Models::Test(_) => ModelsState::Test,
-        }
-    }
-}
-
-/// A serializable representation of the state contained within [Models].
-#[derive(Debug, Clone, Serialize, Deserialize)]
-pub enum ModelsState {
-    Rts {
-        conversation_id: Option<String>,
-        model_id: Option<String>,
-    },
-    Test,
-}
-
-impl Default for ModelsState {
-    fn default() -> Self {
-        Self::Rts {
-            conversation_id: None,
-            model_id: None,
-        }
-    }
-}
-
-impl Model for Models {
-    fn stream(
-        &self,
-        messages: Vec<Message>,
-        tool_specs: Option<Vec<ToolSpec>>,
-        system_prompt: Option<String>,
-        cancel_token: CancellationToken,
-    ) -> Pin<Box<dyn Stream<Item = StreamResult> + Send + 'static>> {
-        match self {
-            Models::Rts(rts_model) => rts_model.stream(messages, tool_specs, system_prompt, cancel_token),
-            Models::Test(test_model) => test_model.stream(messages, tool_specs, system_prompt, cancel_token),
-        }
-    }
-}
-
 #[derive(Debug, Clone)]
 pub struct MockModel {
     inner: Arc<Mutex<mock::Inner>>,

crates/agent/src/agent/agent_loop/types.rs

@@ -14,11 +14,6 @@ use serde_json::Map;
 use tracing::error;
 use uuid::Uuid;
 
-use crate::api_client::error::{
-    ApiClientError,
-    ConverseStreamError,
-};
-
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub enum StreamEvent {
@@ -74,12 +69,10 @@ impl StreamError {
         self
     }
 
-    /// Helper for downcasting the error source to [ConverseStreamError].
-    ///
-    /// Just defining this here for simplicity
-    pub fn as_rts_error(&self) -> Option<&ConverseStreamError> {
+    /// Helper for downcasting a [StreamErrorSource] to a concrete type.
+    pub fn as_concrete_error<T: StreamErrorSource>(&self) -> Option<&T> {
         if let Some(source) = &self.source {
-            (*source).as_any().downcast_ref::<ConverseStreamError>()
+            (*source).as_any().downcast_ref::<T>()
         } else {
             None
         }
@@ -164,18 +157,6 @@ pub trait StreamErrorSource: std::any::Any + std::error::Error + Send + Sync {
     fn as_any(&self) -> &dyn std::any::Any;
 }
 
-impl StreamErrorSource for ConverseStreamError {
-    fn as_any(&self) -> &dyn std::any::Any {
-        self
-    }
-}
-
-impl StreamErrorSource for ApiClientError {
-    fn as_any(&self) -> &dyn std::any::Any {
-        self
-    }
-}
-
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct Message {
@@ -573,7 +554,6 @@ mod tests {
     use std::str::FromStr;
 
     use super::*;
-    use crate::api_client::error::ConverseStreamErrorKind;
 
     macro_rules! test_ser_deser {
         ($ty:ident, $variant:expr, $text:expr) => {
@@ -585,19 +565,6 @@ mod tests {
         };
     }
 
-    #[test]
-    fn test_other_stream_err_downcasting() {
-        let err = StreamError::new(StreamErrorKind::Interrupted).with_source(Arc::new(ConverseStreamError::new(
-            ConverseStreamErrorKind::ModelOverloadedError,
-            None::<aws_smithy_types::error::operation::BuildError>, /* annoying type inference
-                                                                     * required */
-        )));
-        assert!(
-            err.as_rts_error()
-                .is_some_and(|r| matches!(r.kind, ConverseStreamErrorKind::ModelOverloadedError))
-        );
-    }
-
     #[test]
     fn test_image_format_ser_deser() {
         test_ser_deser!(ImageFormat, ImageFormat::Gif, "gif");

crates/agent/src/agent/mod.rs

@@ -4,7 +4,6 @@ pub mod consts;
 pub mod mcp;
 mod permissions;
 pub mod protocol;
-pub mod rts;
 pub mod task_executor;
 mod tool_utils;
 pub mod tools;