@@ -113,18 +113,18 @@ def _prepare_for_generation(self, state: WillaChatbotState) -> dict[str, list[An
113113 """Prepare the current and past messages for response generation."""
114114 messages = state ["messages" ]
115115 summarized_conversation = state .get ("summarized_messages" , messages )
116-
116+
117117 if not any (isinstance (msg , HumanMessage ) for msg in messages ):
118118 return {"messages" : [AIMessage (content = "I'm sorry, I didn't receive a question." )]}
119-
119+
120120 prompt = get_langfuse_prompt ()
121121 system_messages = prompt .invoke ({})
122-
122+
123123 if hasattr (system_messages , "messages" ):
124124 all_messages = summarized_conversation + system_messages .messages
125125 else :
126126 all_messages = summarized_conversation + [system_messages ]
127-
127+
128128 return {"messages" : all_messages }
129129
130130 def _generate_response (self , state : WillaChatbotState ) -> dict [str , list [AnyMessage ]]:
@@ -145,7 +145,7 @@ def _generate_response(self, state: WillaChatbotState) -> dict[str, list[AnyMess
145145
146146 # Create clean response content
147147 response_content = str (response .content ) if hasattr (response , 'content' ) else str (response )
148-
148+
149149 response_messages : list [AnyMessage ] = [AIMessage (content = response_content ),
150150 ChatMessage (content = tind_metadata , role = 'TIND' ,
151151 response_metadata = {'tind' : True })]
0 commit comments