Update main.py

main.py

@@ -376,63 +376,6 @@ class FollowupQueryModel(BaseModel):
 
 import re
 
-def parse_followup_response(input_text):
-    # Define patterns for response and interact
-    response_pattern = re.compile(r'<response>(.*?)<\/response>', re.DOTALL)
-    interact_pattern = re.compile(r'<interact>(.*?)<\/interact>', re.DOTALL)
-
-    # Find all matches for response and interact
-    response_matches = response_pattern.finditer(input_text)
-    interact_matches = interact_pattern.finditer(input_text)
-
-    # Initialize variables to keep track of the position
-    last_end = 0
-    combined_response = ""
-    parsed_interacts = []
-
-    # Combine responses and capture everything in between
-    for response_match in response_matches:
-        # Capture text before the current response tag
-        combined_response += input_text[last_end:response_match.start()].strip() + "\n"
-        # Add the response content
-        combined_response += response_match.group(1).strip() + "\n"
-        # Update the last end position
-        last_end = response_match.end()
-
-    # Check for interacts and parse them
-    for interact_match in interact_matches:
-        # Capture text before the current interact tag
-        combined_response += input_text[last_end:interact_match.start()].strip() + "\n"
-        # Process the interact block
-        interact_text = interact_match.group(1).strip()
-        if interact_text:
-            # Split by "text:" to separate each question block
-            question_blocks = interact_text.split("- text:")
-            
-            # Loop through each block and extract the question and its options
-            for block in question_blocks[1:]:
-                # Extract the question using regex (up to the "options:" part)
-                question_match = re.search(r'^(.*?)\s*options:', block, re.DOTALL)
-                if question_match:
-                    question = question_match.group(1).strip().strip('"')
-                    
-                    # Extract the options using regex
-                    options_match = re.search(r'options:\s*(.*?)$', block, re.DOTALL)
-                    if options_match:
-                        options = [option.strip().strip('"') for option in options_match.group(1).split('-') if option.strip()]
-                        
-                        # Add the parsed question and options to the list
-                        parsed_interacts.append({'question': question, 'options': options})
-        # Update the last end position
-        last_end = interact_match.end()
-
-    # Capture any remaining text after the last tag
-    combined_response += input_text[last_end:].strip()
-
-    return combined_response.strip(), parsed_interacts
-
-import re
-
 def parse_followup_and_tools(input_text):
     # Remove extra brackets and excess quotes
     cleaned_text = re.sub(r'\[|\]|"+', ' ', input_text)
@@ -498,7 +441,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
             yield content
 
         logger.info(f"LLM RAW response for query: {query.query}: {full_response}")
-        response_content, interact = parse_followup_response(full_response)
+        response_content, interact,tools = parse_followup_and_tools(full_response)
         
         result = {
             "response": response_content,
@@ -541,7 +484,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
             yield content
 
         logger.info(f"LLM RAW response for query: {query.query}: {full_response}")
-        response_content, interact = parse_followup_response(full_response)
+        response_content, interact,tools = parse_followup_and_tools(full_response)
         
         result = {
             "clarification": interact