chore: Updates Origin to use Enum value (#3375)

Co-authored-by: Kenneth S. <[email protected]>

Author: kensave

crates/chat-cli/src/api_client/mod.rs

@@ -365,7 +365,7 @@ impl ApiClient {
         let request = self
             .client
             .list_available_models()
-            .set_origin(Some(Origin::from("KIRO_CLI")))
+            .set_origin(Some(Origin::KiroCli))
             .set_profile_arn(self.profile.as_ref().map(|p| p.arn.clone()));
         let mut paginator = request.into_paginator().send();
 
@@ -458,7 +458,7 @@ impl ApiClient {
     ) -> Result<amzn_codewhisperer_client::operation::get_usage_limits::GetUsageLimitsOutput, ApiClientError> {
         self.client
             .get_usage_limits()
-            .set_origin(Some(amzn_codewhisperer_client::types::Origin::from("KIRO_CLI")))
+            .set_origin(Some(amzn_codewhisperer_client::types::Origin::KiroCli))
             .send()
             .await
             .map_err(ApiClientError::GetUsageLimitsError)
@@ -543,7 +543,7 @@ impl ApiClient {
             match client
                 .send_message()
                 .conversation_state(conversation_state)
-                .set_source(Some(QDeveloperOrigin::from("KIRO_CLI")))
+                .set_source(Some(QDeveloperOrigin::KiroCli))
                 .send()
                 .await
             {

crates/chat-cli/src/api_client/model.rs

@@ -940,8 +940,7 @@ impl From<UserInputMessage> for amzn_codewhisperer_streaming_client::types::User
             .set_user_input_message_context(value.user_input_message_context.map(Into::into))
             .set_user_intent(value.user_intent.map(Into::into))
             .set_model_id(value.model_id)
-            //TODO: Setup new origin.
-            .origin(amzn_codewhisperer_streaming_client::types::Origin::from("KIRO_CLI"))
+            .origin(amzn_codewhisperer_streaming_client::types::Origin::KiroCli)
             .build()
             .expect("Failed to build UserInputMessage")
     }
@@ -955,8 +954,7 @@ impl From<UserInputMessage> for amzn_qdeveloper_streaming_client::types::UserInp
             .set_user_input_message_context(value.user_input_message_context.map(Into::into))
             .set_user_intent(value.user_intent.map(Into::into))
             .set_model_id(value.model_id)
-            //TODO: Setup new origin.
-            .origin(amzn_qdeveloper_streaming_client::types::Origin::from("KIRO_CLI"))
+            .origin(amzn_qdeveloper_streaming_client::types::Origin::KiroCli)
             .build()
             .expect("Failed to build UserInputMessage")
     }