فهرست منبع

Update api_server: record operation user

StrayWarrior 23 ساعت پیش
والد
کامیت
b612a162f0
1فایلهای تغییر یافته به همراه5 افزوده شده و 1 حذف شده
  1. 5 1
      pqai_agent_server/api_server.py

+ 5 - 1
pqai_agent_server/api_server.py

@@ -411,6 +411,7 @@ def save_native_agent_configuration():
     tools = json.dumps(req_data.get('tools', []))
     sub_agents = json.dumps(req_data.get('sub_agents', []))
     extra_params = req_data.get('extra_params', {})
+    operate_user = req_data.get('user', None)
     if isinstance(extra_params, dict):
         extra_params = json.dumps(extra_params)
     elif isinstance(extra_params, str):
@@ -439,6 +440,7 @@ def save_native_agent_configuration():
             agent.tools = tools
             agent.sub_agents = sub_agents
             agent.extra_params = extra_params
+            agent.update_user = operate_user
         else:
             agent = AgentConfiguration(
                 name=name,
@@ -449,7 +451,9 @@ def save_native_agent_configuration():
                 task_prompt=task_prompt,
                 tools=tools,
                 sub_agents=sub_agents,
-                extra_params=extra_params
+                extra_params=extra_params,
+                create_user=operate_user,
+                update_user=operate_user
             )
             session.add(agent)