Merge pull request #180 from lijingcheng2021/react_pro_v1

Refactor: Remove __init__ method and rewrite search functionality for improved modularity

Author: zhang_lu

omagent-core/src/omagent_core/advanced_components/workflow/react/agent/wiki_search/wiki_search.py

@@ -1,17 +1,27 @@
 from omagent_core.engine.worker.base import BaseWorker
 from omagent_core.utils.registry import registry
 from omagent_core.utils.logger import logging
-from langchain.agents.react.base import DocstoreExplorer
-from langchain_community.docstore.wikipedia import Wikipedia
+import wikipedia
 from pydantic import Field
 
+class WikipediaSearcher:
+    """Simple Wikipedia search implementation that mimics DocstoreExplorer"""
+    
+    def search(self, query: str) -> str:
+        """Search Wikipedia and return the page content"""
+        try:
+            page_content = wikipedia.page(query).content
+            return page_content
+        except wikipedia.PageError:
+            return f"Could not find [{query}]. Similar: {wikipedia.search(query)}"
+        except wikipedia.DisambiguationError:
+            return f"Could not find [{query}]. Similar: {wikipedia.search(query)}"
+
 @registry.register_worker()
 class WikiSearch(BaseWorker):
     """Wiki Search worker for React workflow"""
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.docstore = DocstoreExplorer(Wikipedia())
+    wiki_searcher: WikipediaSearcher = Field(default_factory=WikipediaSearcher)
 
     def _run(self, action_output: str, *args, **kwargs):
         """Execute search or lookup based on action output"""
@@ -60,7 +70,7 @@ def _handle_search(self, action_output: str) -> str:
             return action_output
 
         try:
-            result = self.docstore.search(search_term)
+            result = self.wiki_searcher.search(search_term)
             if result:
                 result_text = result.strip('\n').strip().replace('\n', '')
 

omagent-core/src/omagent_core/advanced_components/workflow/react_pro/agent/action/action.py

@@ -21,9 +21,6 @@ class Action(BaseLLMBackend, BaseWorker):
         ]
     )
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
     def _run(self, *args, **kwargs):
         """Process the query using ReAct approach"""
         # Get context and other states from STM