fix: remove await from sync ctx.set_state() and fix Role enum usage

- ctx.set_state() and ctx.get_state() are synchronous in agent-framework
  1.1.1 (return None/value directly). Removing incorrect await that caused:
  TypeError: object NoneType can't be used in 'await' expression

- Role is a NewType (str alias) in 1.1.1, not an enum. Replace Role.USER
  and Role.ASSISTANT with string literals, and .role.value with .role

- Replace removed TextContent with Content.from_text() in retry utils

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: Prachig-Microsoft

src/ContentProcessorWorkflow/src/libs/agent_framework/azure_openai_response_retry.py

@@ -242,19 +242,14 @@ def _set_message_text(message: Any, new_text: str) -> Any:
         return out
 
     # agent_framework Message: .text is read-only, but .contents is a
-    # mutable list of content items (TextContent, DataContent, etc.).
-    # Replace all text-type items with a single TextContent carrying the
-    # truncated text.
+    # mutable list of content items (Content objects).
+    # Replace all items with a single Content carrying the truncated text.
     contents = getattr(message, "contents", None)
     if isinstance(contents, list):
         try:
-            # Dynamically import TextContent to avoid a hard dependency on
-            # a specific agent_framework version.
-            from agent_framework._types import (
-                TextContent,  # type: ignore[import-untyped]
-            )
+            from agent_framework._types import Content
 
-            message.contents = [TextContent(text=new_text)]
+            message.contents = [Content.from_text(new_text)]
             return message
         except Exception:
             # If import or mutation fails, try the generic path below.

src/ContentProcessorWorkflow/src/libs/agent_framework/groupchat_orchestrator.py

@@ -1042,7 +1042,7 @@ def _backfill_tool_usage_from_conversation(
         for msg in conversation:
             try:
                 role = getattr(msg, "role", None)
-                if role != Role.ASSISTANT:
+                if role != "assistant":
                     continue
 
                 agent_name = getattr(msg, "author_name", None) or "assistant"

src/ContentProcessorWorkflow/src/libs/agent_framework/middlewares.py

@@ -241,7 +241,7 @@ async def process(
 
         for i, message in enumerate(context.messages):
             content = message.text if message.text else str(message.contents)
-            logger.debug("  Message %d (%s): %s", i + 1, message.role.value, content)
+            logger.debug("  Message %d (%s): %s", i + 1, message.role, content)
 
         logger.debug(
             "[InputObserverMiddleware] Total messages: %d", len(context.messages)
@@ -252,7 +252,7 @@ async def process(
         modified_count = 0
 
         for message in context.messages:
-            if message.role == Role.USER and message.text:
+            if message.role == "user" and message.text:
                 original_text = message.text
                 updated_text = original_text
 

src/ContentProcessorWorkflow/src/steps/document_process/executor/document_process_executor.py

@@ -339,5 +339,5 @@ async def _on_poll(poll_data: dict) -> None:
             )
         )
 
-        await ctx.set_state("workflow_output", workflow_output)
+        ctx.set_state("workflow_output", workflow_output)
         await ctx.send_message(workflow_output)

src/ContentProcessorWorkflow/src/steps/gap_analysis/executor/gap_executor.py

@@ -142,7 +142,7 @@ async def handle_execute(
                 )
             )
 
-            await ctx.set_state("workflow_output", result)
+            ctx.set_state("workflow_output", result)
             await ctx.yield_output(result)
             return
 
@@ -205,7 +205,7 @@ async def handle_execute(
             Executor_Output(step_name="gap_analysis", output_data=gap_result)
         )
 
-        await ctx.set_state("workflow_output", result)
+        ctx.set_state("workflow_output", result)
         await ctx.yield_output(result)
 
     async def fetch_processed_result(self, process_id: str) -> dict | None:

src/ContentProcessorWorkflow/src/steps/rai/executor/rai_executor.py

@@ -118,7 +118,7 @@ async def handle_exectue(
                 Executor_Output(step_name="rai_analysis", output_data=rai_result)
             )
 
-            await ctx.set_state("workflow_output", result)
+            ctx.set_state("workflow_output", result)
             await ctx.send_message(result)
             return
 

src/ContentProcessorWorkflow/src/steps/summarize/executor/summarize_executor.py

@@ -123,7 +123,7 @@ async def handle_execute(
                 Executor_Output(step_name="summarizing", output_data=summarized_result)
             )
 
-            await ctx.set_state("workflow_output", result)
+            ctx.set_state("workflow_output", result)
             await ctx.send_message(result)
             return
 
@@ -204,7 +204,7 @@ async def handle_execute(
             Executor_Output(step_name="summarizing", output_data=summarized_result)
         )
 
-        await ctx.set_state("workflow_output", result)
+        ctx.set_state("workflow_output", result)
         await ctx.send_message(result)
 
     async def fetch_processed_steps_result(self, process_id: str) -> dict | None:

src/ContentProcessorWorkflow/tests/unit/libs/agent_framework/test_agent_builder.py

@@ -85,7 +85,7 @@ def test_build_delegates_to_chat_agent(self, mock_chat_agent):
         call_kwargs = mock_chat_agent.call_args
         assert call_kwargs.kwargs["name"] == "Bot"
         assert call_kwargs.kwargs["instructions"] == "Do stuff"
-        assert call_kwargs.kwargs["temperature"] == 0.5
+        assert call_kwargs.kwargs["default_options"]["temperature"] == 0.5
 
 
 # ── Static factory ───────────────────────────────────────────────────────────
@@ -107,7 +107,7 @@ def test_create_agent_delegates_to_chat_agent(self, mock_chat_agent):
         assert agent == "agent_instance"
         call_kwargs = mock_chat_agent.call_args
         assert call_kwargs.kwargs["name"] == "Bot"
-        assert call_kwargs.kwargs["temperature"] == 0.3
+        assert call_kwargs.kwargs["default_options"]["temperature"] == 0.3
 
 
 # ── with_kwargs ──────────────────────────────────────────────────────────────

src/ContentProcessorWorkflow/tests/unit/libs/agent_framework/test_azure_openai_response_retry_utils.py

@@ -6,7 +6,7 @@
 
 import pytest
 from agent_framework import Message
-from agent_framework._types import TextContent
+from agent_framework._types import Content
 
 from libs.agent_framework.azure_openai_response_retry import (
     ContextTrimConfig,
@@ -148,7 +148,7 @@ def test_message_replaces_contents(self) -> None:
         result = _set_message_text(m, "truncated")
         assert result.text == "truncated"
         assert len(result.contents) == 1
-        assert isinstance(result.contents[0], TextContent)
+        assert isinstance(result.contents[0], Content)
 
 
 class TestTrimMessagesWithMessage:

src/ContentProcessorWorkflow/tests/unit/libs/agent_framework/test_input_observer_middleware.py

@@ -7,7 +7,7 @@
 import asyncio
 from types import SimpleNamespace
 
-from agent_framework import Message, Role
+from agent_framework import Message
 
 from libs.agent_framework.middlewares import InputObserverMiddleware
 
@@ -16,7 +16,7 @@ def test_input_observer_middleware_replaces_user_text_when_configured() -> None:
     async def _run() -> None:
         ctx = SimpleNamespace(
             messages=[
-                Message(role=Role.USER, contents=["original"]),
+                Message(role="user", contents=["original"]),
             ]
         )
 
@@ -27,7 +27,7 @@ async def _next(_context):
 
         await mw.process(ctx, _next)
 
-        assert ctx.messages[0].role == Role.USER
+        assert ctx.messages[0].role == "user"
         assert ctx.messages[0].text == "replacement"
 
     asyncio.run(_run())