Update sherlock2.py

sherlock2.py

@@ -89,13 +89,13 @@ def search_and_scrape_wikipedia(keywords, max_topics_per_query=3, react_model='g
 
     **Thought 1:** I need to search Wikipedia for information related to "{question}".
 
-    **Action 1:** <search>{question}</search>
+    **Action 1:** 
 
     **Observation 1:** {observation}  # This will be filled in during the process
 
     # ... (Further Thought-Action-Observation steps as needed)
 
-    **Action N:** <finish>{answer}</finish>  # The final answer will be extracted from here
+    **Action N:**   # The final answer will be extracted from here
     """
 
     search_history = set()  # Keep track of explored topics to avoid redundancy
@@ -103,7 +103,8 @@ def search_and_scrape_wikipedia(keywords, max_topics_per_query=3, react_model='g
     react_model = genai.GenerativeModel(react_model)  # Initialize the generative model
 
     for query in keywords:
-        search_terms = wikipedia.search(query, results=max_topics_per_query, suggestion=False)  # Search Wikipedia
+        # Search Wikipedia (modified line)
+        search_terms = wikipedia.search(query, results=max_topics_per_query, suggestion=False, srsearch=query) 
 
         for search_term in search_terms:
             if search_term in search_history:
@@ -117,19 +118,20 @@ def search_and_scrape_wikipedia(keywords, max_topics_per_query=3, react_model='g
 
                 # Perform ReAct-based search and extraction
                 while True:
-                    response = react_model.generate_content([react_prompt], stop_sequences=["</finish>"])
+                    response = react_model.generate_content([react_prompt], stop_sequences=[""])
 
                     # Extract action and observation from the response
-                    action, observation = re.findall(r"<(.*?)>(.*?)</\1>", response.text)[-1]  # Get the last action and observation
+                    action, observation = re.findall(r"<(.*?)>(.*?)", response.text)[-1]  # Get the last action and observation
 
                     # Update the ReAct prompt with the observation
                     react_prompt = react_prompt.replace("{observation}", observation.strip(), 1)  # Replace only the first occurrence
 
                     if action == "finish":
                         answer = observation.strip()  # Extract the final answer
-                        break  # Exit the loop when </finish> is encountered
+                        break  # Exit the loop when  is encountered
 
-                page = wikipedia.page(search_term, auto_suggest=False)
+                # Get Wikipedia page and URL (modified line)
+                page = wikipedia.page(search_term, auto_suggest=False)  
                 url = page.url
 
                 additional_sources = []