sync the names

SylphAI-Inc · Jun 30, 2024 · 632e1b1 · 632e1b1
1 parent 145c1b2
commit 632e1b1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lightrag/components/agent/__init__.py b/lightrag/components/agent/__init__.py
@@ -1,8 +1,8 @@
-from .react import DEFAULT_REACT_AGENT_SYSTEM_PROMPT, ReactAgent
+from .react import DEFAULT_REACT_AGENT_SYSTEM_PROMPT, ReActAgent
 from lightrag.utils.registry import EntityMapping
 
 __all__ = [
-    "ReactAgent",
+    "ReActAgent",
     "DEFAULT_REACT_AGENT_SYSTEM_PROMPT",
 ]
 

diff --git a/lightrag/components/agent/react.py b/lightrag/components/agent/react.py
@@ -164,7 +164,7 @@ def __init__(
             func=self._finish,
             answer="final answer: 'answer'",
         )
-        output_parser = JsonOutputParser(data_class=ouput_data_class, example=example)
+        output_parser = JsonOutputParser(data_class=ouput_data_class, examples=example)
         prompt_kwargs = {
             "tools": self.tool_manager.yaml_definitions,
             "output_format_str": output_parser.format_instructions(),
@@ -320,7 +320,7 @@ def _extra_repr(self) -> str:
 
 
 if __name__ == "__main__":
-    from components.model_client import GroqAPIClient
+    from lightrag.components.model_client import GroqAPIClient
     from lightrag.core.types import ModelClientType
     from lightrag.utils import setup_env  # noqa