Skip to content

Commit ac4bf30

Browse files
committed
refactor: update SQL queries to include embedding_model_id and adjust filters for knowledge_id
1 parent a34c089 commit ac4bf30

File tree

3 files changed

+8
-5
lines changed

3 files changed

+8
-5
lines changed

apps/knowledge/serializers/knowledge.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,8 +308,8 @@ def one(self):
308308
model=get_dynamics_model({'temp.id': models.CharField()})
309309
).filter(**{'temp.id': self.data.get("knowledge_id")}),
310310
'knowledge_custom_sql': QuerySet(
311-
model=get_dynamics_model({'knowledge.user_id': models.CharField()})
312-
).filter(**{'knowledge.user_id': self.data.get("user_id")}),
311+
model=get_dynamics_model({'knowledge.id': models.CharField()})
312+
).filter(**{'knowledge.id': self.data.get("knowledge_id")}),
313313
'folder_query_set': QuerySet(KnowledgeFolder)
314314
}
315315
if not workspace_manage:

apps/knowledge/sql/list_knowledge_user.sql

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
1212
"temp_knowledge".file_size_limit,
1313
"temp_knowledge".file_count_limit,
1414
"temp_knowledge"."scope",
15+
"temp_knowledge"."embedding_model_id"::text,
1516
"document_temp"."char_length",
1617
CASE
1718
WHEN
@@ -21,9 +22,8 @@ FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
2122
FROM (SELECT knowledge.*
2223
FROM knowledge knowledge ${knowledge_custom_sql}
2324
AND id in (select target
24-
from workspace_user_resource_permission
25-
${workspace_user_resource_permission_query_set}
26-
and 'VIEW' = any (permission_list))) temp_knowledge
25+
from workspace_user_resource_permission ${workspace_user_resource_permission_query_set}
26+
and 'VIEW' = any (permission_list))) temp_knowledge
2727
LEFT JOIN (SELECT "count"("id") AS document_count, "sum"("char_length") "char_length", knowledge_id
2828
FROM "document"
2929
GROUP BY knowledge_id) "document_temp" ON temp_knowledge."id" = "document_temp".knowledge_id
@@ -47,6 +47,7 @@ FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
4747
0 as file_size_limit,
4848
0 as file_count_limit,
4949
'WORKSPACE' as "scope",
50+
'' as embedding_model_id,
5051
0 as char_length,
5152
0 as application_mapping_count,
5253
0 as document_count

apps/knowledge/sql/list_knowledge_user_ee.sql

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
1212
"temp_knowledge".file_size_limit,
1313
"temp_knowledge".file_count_limit,
1414
"temp_knowledge"."scope",
15+
"temp_knowledge"."embedding_model_id"::text,
1516
"document_temp"."char_length",
1617
CASE
1718
WHEN
@@ -60,6 +61,7 @@ FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
6061
0 as file_size_limit,
6162
0 as file_count_limit,
6263
'WORKSPACE' as "scope",
64+
'' as embedding_model_id,
6365
0 as char_length,
6466
0 as application_mapping_count,
6567
0 as document_count

0 commit comments

Comments
 (0)