mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-14 05:31:40 +00:00
Gpts app v0.4 (#1170)
This commit is contained in:
@@ -94,10 +94,10 @@ class MetaDbGptsMessageMemory(GptsMessageMemory):
|
||||
conv_id: str,
|
||||
agent1: str,
|
||||
agent2: str,
|
||||
current_gogal: Optional[str] = None,
|
||||
current_goal: Optional[str] = None,
|
||||
) -> Optional[List[GptsMessage]]:
|
||||
db_results = self.gpts_message.get_between_agents(
|
||||
conv_id, agent1, agent2, current_gogal
|
||||
conv_id, agent1, agent2, current_goal
|
||||
)
|
||||
results = []
|
||||
db_results = sorted(db_results, key=lambda x: x.rounds)
|
||||
|
@@ -39,7 +39,7 @@ class GptsMessagesEntity(Model):
|
||||
content = Column(
|
||||
Text(length=2**31 - 1), nullable=True, comment="Content of the speech"
|
||||
)
|
||||
current_gogal = Column(
|
||||
current_goal = Column(
|
||||
Text, nullable=True, comment="The target corresponding to the current message"
|
||||
)
|
||||
context = Column(Text, nullable=True, comment="Current conversation context")
|
||||
@@ -78,7 +78,7 @@ class GptsMessagesDao(BaseDao):
|
||||
model_name=entity.get("model_name", None),
|
||||
context=entity.get("context", None),
|
||||
rounds=entity.get("rounds", None),
|
||||
current_gogal=entity.get("current_gogal", None),
|
||||
current_goal=entity.get("current_goal", None),
|
||||
review_info=entity.get("review_info", None),
|
||||
action_report=entity.get("action_report", None),
|
||||
)
|
||||
@@ -120,7 +120,7 @@ class GptsMessagesDao(BaseDao):
|
||||
conv_id: str,
|
||||
agent1: str,
|
||||
agent2: str,
|
||||
current_gogal: Optional[str] = None,
|
||||
current_goal: Optional[str] = None,
|
||||
) -> Optional[List[GptsMessagesEntity]]:
|
||||
session = self.get_raw_session()
|
||||
gpts_messages = session.query(GptsMessagesEntity)
|
||||
@@ -139,9 +139,9 @@ class GptsMessagesDao(BaseDao):
|
||||
),
|
||||
)
|
||||
)
|
||||
if current_gogal:
|
||||
if current_goal:
|
||||
gpts_messages = gpts_messages.filter(
|
||||
GptsMessagesEntity.current_gogal == current_gogal
|
||||
GptsMessagesEntity.current_goal == current_goal
|
||||
)
|
||||
result = gpts_messages.order_by(GptsMessagesEntity.rounds).all()
|
||||
session.close()
|
||||
|
@@ -48,7 +48,7 @@ class AgentOperator(
|
||||
now_rely_messages: List[Dict] = []
|
||||
|
||||
# Isolate the message delivery mechanism and pass it to the operator
|
||||
input_value.message["current_gogal"] = (
|
||||
input_value.message["current_goal"] = (
|
||||
f"[{self._agent.name if self._agent.name else self._agent.profile}]:"
|
||||
+ input_value.message["content"]
|
||||
)
|
||||
@@ -139,14 +139,14 @@ class AwelAgentOperator(
|
||||
agent = await self.get_agent(input_value)
|
||||
if agent.fixed_subgoal and len(agent.fixed_subgoal) > 0:
|
||||
# Isolate the message delivery mechanism and pass it to the operator
|
||||
input_value.message["current_gogal"] = (
|
||||
input_value.message["current_goal"] = (
|
||||
f"[{agent.name if agent.name else agent.profile}]:"
|
||||
+ agent.fixed_subgoal
|
||||
)
|
||||
now_message["content"] = agent.fixed_subgoal
|
||||
else:
|
||||
# Isolate the message delivery mechanism and pass it to the operator
|
||||
input_value.message["current_gogal"] = (
|
||||
input_value.message["current_goal"] = (
|
||||
f"[{agent.name if agent.name else agent.profile}]:"
|
||||
+ input_value.message["content"]
|
||||
)
|
||||
|
@@ -45,7 +45,7 @@ class AwelLayoutChatManager(ManagerAgent):
|
||||
start_message_context: AgentGenerateContext = AgentGenerateContext(
|
||||
message={
|
||||
"content": message,
|
||||
"current_gogal": message,
|
||||
"current_goal": message,
|
||||
},
|
||||
sender=self,
|
||||
reviewer=reviewer,
|
||||
|
@@ -58,7 +58,7 @@ class AwelLayoutChatNewManager(ManagerAgent):
|
||||
start_message_context: AgentGenerateContext = AgentGenerateContext(
|
||||
message={
|
||||
"content": message,
|
||||
"current_gogal": message,
|
||||
"current_goal": message,
|
||||
},
|
||||
sender=self,
|
||||
reviewer=reviewer,
|
||||
|
@@ -161,7 +161,7 @@ class AutoPlanChatManager(ManagerAgent):
|
||||
now_plan: GptsPlan = todo_plans[0]
|
||||
current_goal_message = {
|
||||
"content": now_plan.sub_task_content,
|
||||
"current_gogal": now_plan.sub_task_content,
|
||||
"current_goal": now_plan.sub_task_content,
|
||||
"context": {
|
||||
"plan_task": now_plan.sub_task_content,
|
||||
"plan_task_num": now_plan.sub_task_num,
|
||||
|
Reference in New Issue
Block a user