Deepak Sahu commited on
Commit
114d597
·
1 Parent(s): 4d16120

add logging info

Browse files
_agents/pandas_expression_generator/src/pandas_expression_generator/pandas_expression_generator_function.py CHANGED
@@ -129,6 +129,8 @@ async def pandas_expression_generator_function(
129
 
130
  try:
131
  ai_message: AIMessage = await llm_.ainvoke(prompt_)
 
 
132
  structured_text_str = extract_bracket_content(ai_message.content)
133
  output_message = str(expression_executor(structured_text_str, df))[:250]
134
  #
 
129
 
130
  try:
131
  ai_message: AIMessage = await llm_.ainvoke(prompt_)
132
+ logger.info("[Following Should just be structured list of string]")
133
+ logger.info(ai_message)
134
  structured_text_str = extract_bracket_content(ai_message.content)
135
  output_message = str(expression_executor(structured_text_str, df))[:250]
136
  #