优化react

Author: shell-nlp

gpt_server/model_handler/react/prompts/__init__.py

@@ -2,7 +2,7 @@
 import json
 import uuid
 
-from gpt_server.model_handler.react.prompt import (
+from gpt_server.model_handler.react.v0.prompt import (
     GLM4_TOOL_PROMPT,
     TOOL_SUFFIX_PROMPT,
 )

gpt_server/model_handler/react/v0/__init__.py

@@ -3,10 +3,10 @@
 import json
 import uuid
 
-from gpt_server.model_handler.react.prompt import (
+from gpt_server.model_handler.react.v0.prompt import (
     TOOL_SUFFIX_PROMPT,
 )
-from gpt_server.model_handler.react.prompts.qwen_prompt import (
+from gpt_server.model_handler.react.v0.prompts.qwen_prompt import (
     TOOL_SYSTEM_PROMPT_CN,
     TOOl_CHOICE_SYSTEM_PROMPT_CN,
     TOOL_CHOICE_SUFFIX_PROMPT,

…ver/model_handler/react/chatglm_react.py …/model_handler/react/v0/chatglm_react.pygpt_server/model_handler/react/chatglm_react.py renamed to gpt_server/model_handler/react/v0/chatglm_react.py gpt_server/model_handler/react/chatglm_react.py renamed to gpt_server/model_handler/react/v0/chatglm_react.py

@@ -2,7 +2,7 @@
 import json
 import uuid
 
-from gpt_server.model_handler.react.prompt import (
+from gpt_server.model_handler.react.v0.prompt import (
     GLM4_TOOL_PROMPT,
     TOOL_SUFFIX_PROMPT,
 )

gpt_server/model_handler/react/prompt.py …_server/model_handler/react/v0/prompt.pygpt_server/model_handler/react/prompt.py renamed to gpt_server/model_handler/react/v0/prompt.py gpt_server/model_handler/react/prompt.py renamed to gpt_server/model_handler/react/v0/prompt.py

@@ -0,0 +1,16 @@
+from pydantic import BaseModel, Field
+from typing import Union
+
+
+class Action(BaseModel):
+    Action: str = Field(description="工具名称，必须是 [{tool_names}] 之一")
+    Action_Input: str = Field(description="工具输入, 值必须使用 json 格式")
+
+
+class Answer(BaseModel):
+    Final_Answer: str = Field(description="问题的最终回答")
+
+
+class React(BaseModel):
+    Thought: str = Field(description="你应该时刻思考自己该做什么")
+    Reason: Union[Action, Answer]