Merge pull request #40 from stakpak/fix/infinite-loop

Fix: Infinite loop

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "stakpak"
-version = "0.1.40"
+version = "0.1.41"
 edition = "2021"
 
 [dependencies]

src/client/models.rs

@@ -568,13 +568,18 @@ pub struct GetDockerfileTemplateArgs {
     pub runtime_version: Option<String>,
 }
 
-#[derive(Debug, Deserialize, Serialize, Clone, PartialEq)]
+#[derive(Debug, Deserialize, Serialize, Clone)]
 pub struct RunAgentInput {
     pub checkpoint_id: Uuid,
-
     pub input: AgentInput,
 }
 
+impl PartialEq for RunAgentInput {
+    fn eq(&self, other: &Self) -> bool {
+        self.input == other.input
+    }
+}
+
 #[derive(Debug, Deserialize, Serialize, Clone)]
 pub struct RunAgentOutput {
     pub checkpoint: AgentCheckpointListItem,

src/commands/agent/mod.rs

@@ -125,10 +125,11 @@ impl AgentCommands {
                     get_or_create_session(&client, agent_id, checkpoint_id, Some(input.clone()))
                         .await?;
 
-                if let Some(flow_ref) = &session.flow_ref {
+                // TODO: Better way to allow sync for specific agents
+                if matches!(agent_id, AgentID::KevinV1) && session.flow_ref.is_some() {
                     let config_clone = config.clone();
                     let client_clone = Client::new(&config_clone).map_err(|e| e.to_string())?;
-                    let flow_ref = flow_ref.clone();
+                    let flow_ref = session.flow_ref.clone().unwrap();
                     tokio::spawn(async move {
                         flow::sync(&config_clone, &client_clone, &flow_ref, None).await
                     });