From c878490140aa4e9d7b71c49214fb47f8618d403b Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 10 一月 2025 18:04:37 +0800
Subject: [PATCH] 初始化自研智能体问题
---
app/config/env_conf/default_agent_conf.json | 16 ++++++++++++----
main.py | 2 +-
app/service/knowledge.py | 2 +-
app/service/v2/initialize_data.py | 3 +--
4 files changed, 15 insertions(+), 8 deletions(-)
diff --git a/app/config/env_conf/default_agent_conf.json b/app/config/env_conf/default_agent_conf.json
index 70c35f7..27eb243 100644
--- a/app/config/env_conf/default_agent_conf.json
+++ b/app/config/env_conf/default_agent_conf.json
@@ -5,28 +5,36 @@
"name": "鎶ヨ〃鍚堝苟",
"description": "鎶ヨ〃鍚堝苟",
"icon": "intellFrame4",
- "agentType": "excelMerge"
+ "agentType": "excelMerge",
+ "dialogType": "3",
+ "mode": "agent-basic"
},
{
"id": "basic_excel_talk",
"name": "鏅鸿兘鏁版嵁",
"description": "鏅鸿兘鏁版嵁",
"icon": "intellFrame4",
- "agentType": "excelTalk"
+ "agentType": "excelTalk",
+ "dialogType": "3",
+ "mode": "agent-basic"
},
{
"id": "basic_question_talk",
"name": "鍑洪缁勫嵎",
"description": "鍑洪缁勫嵎",
"icon": "intellFrame4",
- "agentType": "questionTalk"
+ "agentType": "questionTalk",
+ "dialogType": "3",
+ "mode": "agent-basic"
},
{
"id": "basic_paper_talk",
"name": "鏂囨。鍑哄嵎",
"description": "鏂囨。鍑哄嵎",
"icon": "intellFrame4",
- "agentType": "paperTalk"
+ "agentType": "paperTalk",
+ "dialogType": "3",
+ "mode": "agent-basic"
}
],
"bs": []
diff --git a/app/service/knowledge.py b/app/service/knowledge.py
index 32e5762..7c40272 100644
--- a/app/service/knowledge.py
+++ b/app/service/knowledge.py
@@ -17,7 +17,7 @@
klg_list = [j.id for i in user.groups for j in i.knowledges]
query = query.filter(or_(KnowledgeModel.tenant_id == user_id, KnowledgeModel.id.in_(klg_list)))
if location:
- query = query.filter(KnowledgeModel.permission == "team")
+ query = query.filter(or_(KnowledgeModel.permission == "team", KnowledgeModel.tenant_id == user_id))
if keyword:
query = query.filter(KnowledgeModel.name.like('%{}%'.format(keyword)))
total = query.count()
diff --git a/app/service/v2/initialize_data.py b/app/service/v2/initialize_data.py
index 5d2043a..d6df64b 100644
--- a/app/service/v2/initialize_data.py
+++ b/app/service/v2/initialize_data.py
@@ -138,8 +138,7 @@
else:
try:
dialog = DialogModel(id=agent["id"], name=agent["name"], description=agent["description"],
- icon=agent["icon"], tenant_id=user.id if user else "", dialog_type="3",
- agent_id=agent["id"])
+ icon=agent["icon"], tenant_id=user.id if user else "", dialog_type=agent["dialogType"], mode=agent["mode"])
db.add(dialog)
db.commit()
db.refresh(dialog)
diff --git a/main.py b/main.py
index 9639f36..a37c34f 100644
--- a/main.py
+++ b/main.py
@@ -67,7 +67,7 @@
# )
# 鍒涘缓璋冨害鍣�
scheduler = BackgroundScheduler()
-scheduler.add_job(sync_agents_v2, 'interval', minutes=60, id="sync_resource_data")
+scheduler.add_job(sync_agents_v2, 'interval', minutes=10, id="sync_resource_data")
# scheduler.add_job(delete_file_after_delay, 'interval', minutes=10, id="delete_file_after_delay")
scheduler.start()
--
Gitblit v1.8.0