feat: update to output text in innerinput

Author: iambenkay

llm/openai/src/client.rs

@@ -195,14 +195,15 @@ pub enum InputItem {
 pub enum InnerInput {
     TextInput(String),
     List(Vec<InnerInputItem>),
-    OutputList(Vec<OutputMessageContent>),
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(tag = "type")]
 pub enum InnerInputItem {
     #[serde(rename = "input_text")]
     TextInput { text: String },
+    #[serde(rename = "output_text")]
+    TextOutput { text: String },
     #[serde(rename = "input_image")]
     ImageInput {
         image_url: String,

llm/openai/src/conversions.rs

@@ -43,12 +43,7 @@ pub fn create_request(
 pub fn messages_to_input_items(messages: Vec<Message>) -> Vec<InputItem> {
     let mut items = Vec::new();
     for message in messages {
-        let item = match message.role {
-            Role::Assistant => llm_message_to_assistant_message(message),
-            _ => llm_message_to_other_message(message),
-        };
-
-        items.push(item);
+        items.push(llm_message_to_openai_message(message));
     }
     items
 }
@@ -118,11 +113,15 @@ pub fn to_openai_role_name(role: Role) -> &'static str {
     }
 }
 
-pub fn llm_message_to_other_message(message: Message) -> InputItem {
+pub fn llm_message_to_openai_message(message: Message) -> InputItem {
     let mut items = Vec::new();
+
     for content_part in message.content {
         let item = match content_part {
-            ContentPart::Text(msg) => InnerInputItem::TextInput { text: msg },
+            ContentPart::Text(msg) => match message.role {
+                Role::Assistant => InnerInputItem::TextOutput { text: msg },
+                _ => InnerInputItem::TextInput { text: msg },
+            },
             ContentPart::Image(image_reference) => match image_reference {
                 ImageReference::Url(image_url) => InnerInputItem::ImageInput {
                     image_url: image_url.url,
@@ -159,21 +158,6 @@ pub fn llm_message_to_other_message(message: Message) -> InputItem {
     }
 }
 
-pub fn llm_message_to_assistant_message(message: Message) -> InputItem {
-    let mut items = Vec::new();
-
-    for content_part in message.content {
-        if let ContentPart::Text(msg) = content_part {
-            items.push(OutputMessageContent::Text { text: msg });
-        }
-    }
-
-    InputItem::InputMessage {
-        role: to_openai_role_name(message.role).to_string(),
-        content: InnerInput::OutputList(items),
-    }
-}
-
 pub fn parse_error_code(code: String) -> ErrorCode {
     if let Some(code) = <u16 as FromStr>::from_str(&code)
         .ok()

test/components-rust/test-llm/src/lib.rs

@@ -621,8 +621,6 @@ impl Guest for Component {
 
         let mut result = String::new();
 
-        let name = std::env::var("GOLEM_WORKER_NAME").unwrap();
-
         loop {
             match utils::consume_next_event(&stream) {
                 Some(delta) => {