Skip to content

Commit 78a1391

Browse files
committed
feat(llm): update
1 parent 7e9635e commit 78a1391

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

src/retk/const/user_behavior_types.py

+3
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ class UserBehaviorTypeEnum(IntEnum):
1111
NODE_TRASHED_OPS = 4
1212
NODE_RESTORED_OPS = 5
1313
NODE_DELETED_OPS = 6
14+
EXTENDED_NODE_ACCEPTED = 14
15+
1416
# search, backend
1517
SEARCH_GLOBAL = 7
1618
SEARCH_AT = 8
@@ -20,6 +22,7 @@ class UserBehaviorTypeEnum(IntEnum):
2022
# view node, frontend
2123
VIEW_NODE_FROM_RECOMMENDATION = 11
2224
VIEW_NODE_FROM_LINKED_NODE = 12
25+
VIEW_EXTENDED_NODE = 13
2326

2427

2528
USER_BEHAVIOR_TYPE_MAP = {

src/retk/core/scheduler/tasks/extend_node.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@ async def async_deliver_unscheduled_extend_nodes() -> str:
4747
if code != const.CodeEnum.OK:
4848
logger.error(f"knowledge summary error: {code}")
4949
continue
50-
logger.debug(f"summary: {_summary}")
50+
oneline_s = _summary.replace('\n', '\n\n')
51+
logger.debug(f"summary: {oneline_s}")
5152
e0 = time.perf_counter()
5253
_extended, code = await knowledge.extend(
5354
llm_service=knowledge.LLM_SERVICES[item["extendService"]],
@@ -59,7 +60,8 @@ async def async_deliver_unscheduled_extend_nodes() -> str:
5960
if code != const.CodeEnum.OK:
6061
logger.error(f"knowledge extend error: {code}")
6162
continue
62-
logger.debug(f"extended: {_extended}")
63+
oneline_e = _extended.replace('\n', '\n\n')
64+
logger.debug(f"extended: {oneline_e}")
6365
ext = ExtendedNode(
6466
uid=item["uid"],
6567
sourceNid=item["nid"],

0 commit comments

Comments
 (0)