Make mypy happy

Author: pamelafox

src/backend/fastapi_app/rag_advanced.py

@@ -4,6 +4,7 @@
 
 from agents import (
     Agent,
+    ItemHelpers,
     ModelSettings,
     OpenAIChatCompletionsModel,
     Runner,
@@ -124,7 +125,8 @@ async def prepare_context(self) -> tuple[list[ItemPublic], list[ThoughtStep]]:
         thoughts = [
             ThoughtStep(
                 title="Prompt to generate search arguments",
-                description=[{"content": self.search_agent.instructions}] + run_results.input,
+                description=[{"content": self.query_prompt_template}]
+                + ItemHelpers.input_to_new_input_list(run_results.input),
                 props=self.model_for_thoughts,
             ),
             ThoughtStep(
@@ -163,7 +165,8 @@ async def answer(
                 + [
                     ThoughtStep(
                         title="Prompt to generate answer",
-                        description=[{"content": self.answer_agent.instructions}] + run_results.input,
+                        description=[{"content": self.answer_prompt_template}]
+                        + ItemHelpers.input_to_new_input_list(run_results.input),
                         props=self.model_for_thoughts,
                     ),
                 ],
@@ -178,7 +181,7 @@ async def answer_stream(
         run_results = Runner.run_streamed(
             self.answer_agent,
             input=self.chat_params.past_messages
-            + [{"content": self.prepare_rag_request(self.chat_params.original_user_query, items), "role": "user"}],
+            + [{"content": self.prepare_rag_request(self.chat_params.original_user_query, items), "role": "user"}],  # noqa
         )
 
         yield RetrievalResponseDelta(
@@ -188,7 +191,8 @@ async def answer_stream(
                 + [
                     ThoughtStep(
                         title="Prompt to generate answer",
-                        description=[{"content": self.answer_agent.instructions}] + run_results.input,
+                        description=[{"content": self.answer_prompt_template}]
+                        + ItemHelpers.input_to_new_input_list(run_results.input),
                         props=self.model_for_thoughts,
                     ),
                 ],

src/backend/fastapi_app/rag_simple.py

@@ -1,7 +1,7 @@
 from collections.abc import AsyncGenerator
 from typing import Optional, Union
 
-from agents import Agent, ModelSettings, OpenAIChatCompletionsModel, Runner, set_tracing_disabled
+from agents import Agent, ItemHelpers, ModelSettings, OpenAIChatCompletionsModel, Runner, set_tracing_disabled
 from openai import AsyncAzureOpenAI, AsyncOpenAI
 from openai.types.responses import ResponseInputItemParam, ResponseTextDeltaEvent
 
@@ -98,7 +98,8 @@ async def answer(
                 + [
                     ThoughtStep(
                         title="Prompt to generate answer",
-                        description=[{"content": self.answer_agent.instructions}] + run_results.input,
+                        description=[{"content": self.answer_prompt_template}]
+                        + ItemHelpers.input_to_new_input_list(run_results.input),
                         props=self.model_for_thoughts,
                     ),
                 ],
@@ -123,7 +124,8 @@ async def answer_stream(
                 + [
                     ThoughtStep(
                         title="Prompt to generate answer",
-                        description=[{"content": self.answer_agent.instructions}] + run_results.input,
+                        description=[{"content": self.answer_agent.instructions}]
+                        + ItemHelpers.input_to_new_input_list(run_results.input),
                         props=self.model_for_thoughts,
                     ),
                 ],