IlinaEngine 0.8.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,2 @@
1
+ from .engine import Engine
2
+ from .type import NodeEvent, NodeEventTypes, IlinaMessage, IlinaToolCall
@@ -0,0 +1,32 @@
1
+ from pydantic import BaseModel
2
+
3
+ class IlinaConfig(BaseModel):
4
+ """ 文件夹下的 .ilinaconfig 内的数据模型 """
5
+ workpath: str|None = None
6
+ open_or_alarm: bool = False # 如果是 True 倾向于打开文件,如果是 False 倾向于发送通知
7
+
8
+ class MCPConfig(BaseModel):
9
+ command: str
10
+ args: list[str]
11
+
12
+ class ModelConfig(BaseModel):
13
+ base_url: str
14
+ api_key: str
15
+ model_name: str
16
+
17
+ class AIConfig(BaseModel):
18
+ """ 模型配置 """
19
+ main_model: ModelConfig = ModelConfig(
20
+ base_url='http://localhost:11434/v1/',
21
+ api_key='sk-xxx',
22
+ model_name='qwen3-vl:4b'
23
+ )
24
+
25
+ sub_model: ModelConfig = ModelConfig(
26
+ base_url='http://localhost:11434/v1/',
27
+ api_key='sk-xxx',
28
+ model_name='qwen3-vl:4b'
29
+ )
30
+
31
+ mcps: dict[str, MCPConfig] = {}
32
+ default_system_prompt_template: str = '当前工作目录为 {{workpath}},当你编辑文件之后,你应该{{open_or_alarm}}'
@@ -0,0 +1,25 @@
1
+ # Ilina Message
2
+ from typing import Literal
3
+ from pydantic import BaseModel
4
+
5
+ class IlinaToolDefinition(BaseModel):
6
+ """ 工具定义 """
7
+ name: str # 工具名称
8
+ description: str # 工具说明
9
+ arguments: dict[str, object] # 参数的 JSON Schema
10
+
11
+ class IlinaToolCall(BaseModel):
12
+ """ 工具调用 """
13
+ name: str = ''
14
+ arguments: str = ''
15
+ tool_call_id: str = ''
16
+
17
+ class IlinaMessage(BaseModel):
18
+ """ 对话消息 """
19
+ role: Literal['user', 'assistant', 'system', 'tool', 'error']
20
+ content: str = ''
21
+ reasoning_content: str = '' # 仅在 assistant 中使用
22
+ tool_calls: list[IlinaToolCall] = [] # 仅在 assistant 中使用
23
+ tool_call_id: str = '' # 仅在 tool 中使用
24
+ tool_name: str = '' # 保存工具名,仅在 tool 中使用
25
+ # tool_readable_content: str = '' # 返回一个人类可读的消息,如果指定了这一项,那么渲染时建议用这个替换content
@@ -0,0 +1,222 @@
1
+ # 调用 API
2
+
3
+ from enum import Enum
4
+ from typing import Generator
5
+ from logging import getLogger
6
+ from dataclasses import dataclass
7
+ from FovesConfig import ConfigLoader
8
+
9
+
10
+ # from tools import Toolset
11
+ from .tree import Node
12
+ from .sync_mcp import MCPLoader
13
+ from ._config_models import AIConfig
14
+ from ._ilina_message import IlinaMessage, IlinaToolCall
15
+ from openai import OpenAI
16
+ from openai.types.chat import (
17
+ ChatCompletionAssistantMessageParam,
18
+ ChatCompletionMessageFunctionToolCallParam,
19
+ ChatCompletionUserMessageParam,
20
+ ChatCompletionSystemMessageParam,
21
+ ChatCompletionToolMessageParam,
22
+
23
+ ChatCompletionMessageParam,
24
+ ChatCompletionChunk,
25
+ ChatCompletion,
26
+ )
27
+
28
+ class NodeEventTypes(str, Enum):
29
+ CREATED = 'CREATED'
30
+ UPDATED = 'UPDATED'
31
+ FINISNED = 'FINISHED'
32
+ ERROR = 'ERROR'
33
+
34
+ @dataclass
35
+ class NodeEvent:
36
+ """ 节点发生变化的事件 """
37
+ node: Node
38
+ _type: NodeEventTypes
39
+
40
+ class OpenAIClient:
41
+ def __init__(self, is_main_model: bool, mcp_loader: MCPLoader):
42
+ with ConfigLoader('./configs/ai.json', AIConfig) as config:
43
+ if is_main_model:
44
+ modelcfg = config.main_model
45
+ else:
46
+ modelcfg = config.sub_model
47
+ self.client = OpenAI(base_url=modelcfg.base_url, api_key=modelcfg.api_key)
48
+ self.model = modelcfg.model_name
49
+ # self.toolset = Toolset()
50
+ self.mcp_loader = mcp_loader
51
+ self.log = getLogger(f"Model_{self.model}")
52
+
53
+ def ilina_to_openai(self, ilina: IlinaMessage) -> ChatCompletionMessageParam|None:
54
+ """ 将 IlinaMessage 转化为 ChatCompletionMessageParam, 会过滤掉 error 类型"""
55
+ if ilina.role == 'assistant':
56
+ tool_calls = []
57
+ for tool_call in ilina.tool_calls:
58
+ tool_calls.append(ChatCompletionMessageFunctionToolCallParam(
59
+ type='function',
60
+ id=tool_call.tool_call_id,
61
+ function={
62
+ 'name': tool_call.name,
63
+ 'arguments': tool_call.arguments
64
+ }
65
+ ))
66
+
67
+ if len(tool_calls) > 0:
68
+ return ChatCompletionAssistantMessageParam(
69
+ role='assistant',
70
+ content=ilina.content,
71
+ tool_calls=tool_calls
72
+ )
73
+ else:
74
+ return ChatCompletionAssistantMessageParam(
75
+ role='assistant',
76
+ content=ilina.content,
77
+ )
78
+
79
+ elif ilina.role == 'user':
80
+ return ChatCompletionUserMessageParam(
81
+ role='user',
82
+ content=ilina.content
83
+ )
84
+
85
+ elif ilina.role == 'system':
86
+ return ChatCompletionSystemMessageParam(
87
+ role='system',
88
+ content=ilina.content
89
+ )
90
+
91
+ elif ilina.role == 'tool':
92
+ return ChatCompletionToolMessageParam(
93
+ role='tool',
94
+ content=ilina.content,
95
+ tool_call_id=ilina.tool_call_id
96
+ )
97
+
98
+ elif ilina.role == 'error':
99
+ return None
100
+
101
+ def chat(self, messages: list[IlinaMessage]) -> Generator[NodeEvent, None, None]:
102
+ """ 调用模型,会首先用生成器返回流失输出结果,最后return合并的Node """
103
+ new_messages: list[IlinaMessage] = [] # 存储本轮调用生成的消息
104
+ stop_reason: str = '' # 存储停止原因
105
+ self.log.info('开始调用模型')
106
+ while stop_reason != 'stop':
107
+ # 发起请求
108
+ self.log.info('发起 API 请求')
109
+ try:
110
+ # 首先将 messages 转换成 openai 格式
111
+ openai_messages = []
112
+ for item in map(self.ilina_to_openai, messages + new_messages):
113
+ if item is not None:
114
+ openai_messages.append(item)
115
+
116
+ # 发起调用
117
+ res = self.client.chat.completions.create(
118
+ messages=openai_messages,
119
+ model=self.model,
120
+ tools=self.mcp_loader.get_list_openai(),
121
+ stream=True
122
+ )
123
+ except Exception as e:
124
+ yield NodeEvent(Node(IlinaMessage(role='error', content=f'{e}')), NodeEventTypes.ERROR)
125
+ return
126
+
127
+ assistant_node = Node(IlinaMessage(role='assistant'))
128
+ # 传递节点开始事件
129
+ yield NodeEvent(assistant_node, NodeEventTypes.CREATED)
130
+
131
+ # 逐帧解析
132
+ tool_call_nodes: dict[str, Node] = {} # 保存工具 ID 对应的节点
133
+ for chunk in res:
134
+ chunk: ChatCompletionChunk
135
+ delta = chunk.choices[0].delta
136
+
137
+ # 保存流式信息
138
+ if delta.content:
139
+ assistant_node.message.content += delta.content
140
+ yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
141
+
142
+ # 保存思考信息
143
+ reasoning_content_delta = ''
144
+ try:
145
+ self.log.debug(delta)
146
+ if hasattr(delta, 'reasoning_content'): # Deepseek:用 reasoning_content 输出思考流
147
+ reasoning_content_delta = delta.reasoning_content or '' # pyright: ignore[reportAttributeAccessIssue]
148
+ elif hasattr(delta, 'reasoning'): # Ollama: 用 reasoning 输出思考流
149
+ reasoning_content_delta = delta.reasoning or '' # pyright: ignore[reportAttributeAccessIssue]
150
+ except AttributeError: # 如果获取不到,也设置成空
151
+ reasoning_content_delta = ''
152
+
153
+ assistant_node.message.reasoning_content += reasoning_content_delta
154
+ yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
155
+
156
+ # 保存工具调用
157
+ if delta.tool_calls: # 如果工具调用不为 None
158
+ for tool_call in delta.tool_calls: # 对于工具调用列表中的每个工具
159
+ try: # 尝试直接追加到现有索引处
160
+ if tool_call.id: # 覆写 ID
161
+ assistant_node.message.tool_calls[tool_call.index].tool_call_id = tool_call.id
162
+
163
+ if tool_call.function: # 更新 function
164
+ if tool_call.function.name: # 覆写 name
165
+ assistant_node.message.tool_calls[tool_call.index].name = tool_call.function.name
166
+ if tool_call.id:
167
+ tool_call_nodes[tool_call.id].message.tool_name = tool_call.function.name
168
+ if tool_call.function.arguments: # 追加 arguments
169
+ assistant_node.message.tool_calls[tool_call.index].arguments += tool_call.function.arguments
170
+
171
+ yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
172
+
173
+ except IndexError: # 不存在则新增
174
+ assert tool_call.id is not None
175
+ assert tool_call.function is not None
176
+ assert tool_call.function.name is not None
177
+ assert tool_call.function.arguments is not None
178
+ assistant_node.message.tool_calls.append(IlinaToolCall(
179
+ tool_call_id=tool_call.id,
180
+ name=tool_call.function.name,
181
+ arguments=tool_call.function.arguments
182
+ ))
183
+ tool_call_node = Node(IlinaMessage(role='tool', tool_call_id=tool_call.id, tool_name=tool_call.function.name))
184
+ tool_call_nodes[tool_call.id] = tool_call_node
185
+ yield NodeEvent(tool_call_node, NodeEventTypes.CREATED)
186
+ yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
187
+
188
+ # 更新停止原因
189
+ if chunk.choices[0].finish_reason:
190
+ stop_reason = chunk.choices[0].finish_reason
191
+ self.log.info(f'停止原因:{stop_reason}')
192
+
193
+ # 添加助手信息
194
+ new_messages.append(assistant_node.message)
195
+ yield NodeEvent(assistant_node, NodeEventTypes.FINISNED)
196
+
197
+ # 调用并添加工具信息
198
+ # 返回流式传输的工具块
199
+ for call in assistant_node.message.tool_calls:
200
+ self.log.info(f'调用工具 {call.name}')
201
+ result = self.mcp_loader.call(call)
202
+ self.log.info(f'工具返回 {result}')
203
+ # 根据 ID 获取对应的节点并进行修改和发送
204
+ tool_call_node = tool_call_nodes[call.tool_call_id]
205
+ tool_call_node.message.content = result
206
+ # 向新消息列表中添加工具消息
207
+ new_messages.append(tool_call_node.message)
208
+ yield NodeEvent(tool_call_node, NodeEventTypes.UPDATED)
209
+ yield NodeEvent(tool_call_node, NodeEventTypes.FINISNED)
210
+
211
+ self.log.debug(f'当前的new_messages:\n{'\n'.join([str(m) for m in new_messages])}')
212
+
213
+ def once(self, sysprompt: str, user_input: str) -> str|None:
214
+ """ 进行一次调用 """
215
+ res: ChatCompletion = self.client.chat.completions.create(
216
+ messages=[
217
+ {'role': 'system', 'content': sysprompt},
218
+ {'role': 'user', 'content': user_input}
219
+ ],
220
+ model=self.model,
221
+ )
222
+ return res.choices[0].message.content
IlinaEngine/engine.py ADDED
@@ -0,0 +1,147 @@
1
+ # 引擎层
2
+ import logging
3
+ from uuid import UUID
4
+ from typing import Generator
5
+
6
+ from FovesLog import LoggedTask
7
+ from .tree import Tree, Node
8
+ from .call_openai import *
9
+ from ._ilina_message import IlinaMessage
10
+ from .exceptions import *
11
+
12
+ class Engine:
13
+ def __init__(self, filename: str) -> None:
14
+ self.log = logging.getLogger('对话引擎')
15
+ self.log.setLevel(logging.INFO)
16
+
17
+ with LoggedTask('初始化', logger=self.log) as task:
18
+ self.tree = Tree(filename)
19
+ task.checkpoint(f'建立文件树')
20
+ self.mcp_loader = MCPLoader()
21
+ task.checkpoint(f'建立MCP工具')
22
+ self.main_model = OpenAIClient(True, self.mcp_loader)
23
+
24
+ @property
25
+ def workpath(self) -> str:
26
+ """ 获取工作目录 """
27
+ return str(self.tree.workpath)
28
+
29
+ @property
30
+ def name(self) -> str:
31
+ """ 获取树的名字 """
32
+ return self.tree.name
33
+
34
+ @name.setter
35
+ def name(self, value: str):
36
+ self.tree.name = value
37
+
38
+ def set_name(self, new_name: str):
39
+ """ 设置名字的函数版本,适用于某些无法使用属性赋值的情况
40
+
41
+ 说的就是你, lambda 语句
42
+ """
43
+ self.name = new_name
44
+
45
+ @property
46
+ def message_list(self) -> tuple[list[UUID], list[IlinaMessage]]:
47
+ """ 获取当前的消息链 """
48
+ return self.tree.root_node._to_message_list()
49
+
50
+ @property
51
+ def readonly_root_node(self) -> Node:
52
+ """ 获取根节点。注意:只应该用来获取树结构 """
53
+ return self.tree.root_node
54
+
55
+ @property
56
+ def readonly_leaves(self) -> list[Node]:
57
+ """ 获取所有叶子节点。注意:只应该用来获取树结构 """
58
+ return self.tree.root_node._get_leaves()
59
+
60
+ @property
61
+ def readonly_now_node(self) -> UUID:
62
+ """ 获取当前最新节点的 UUID """
63
+ return self.tree.root_node._get_now().uuid
64
+
65
+ def get_message_by_uuid(self, uuid: UUID) -> IlinaMessage:
66
+ """ 通过 UUID 获取消息内容 """
67
+ return self.tree.uuid_to_node_table[uuid].message
68
+
69
+ def get_parent(self, uuid: UUID) -> UUID:
70
+ """ 获取某个节点的父节点。对于根节点和其他找不到父节点的节点会报错 """
71
+ parent = self.tree.root_node._get_parent(uuid)
72
+ if parent:
73
+ return parent
74
+ else:
75
+ raise ParentNotFoundError(uuid)
76
+
77
+ def edit_node(self, target: UUID, new_message: IlinaMessage) -> UUID:
78
+ """ 修改节点内容,不会实际修改,而是作为父节点的新子节点插入,会返回新节点的 UUID """
79
+ new_node = Node(new_message)
80
+ with self.tree as tree:
81
+ tree.insert(new_node, self.get_parent(target))
82
+ return new_node.uuid
83
+
84
+ def invoke(self, start_from: UUID|None=None) -> Generator[NodeEvent, bool, None]:
85
+ """ 将当前的对话发送给主模型,并获取其回复,默认从最新分支开始
86
+ 并会自动将 NodeFinished 事件的节点添加到节点树中
87
+ 如果指定了 start_from,会从那里截断并插入节点。
88
+ 如果 start_from 不在当前分支,会首先切换过去。
89
+ """
90
+ self.log.info(f'调用 AI')
91
+
92
+ # 如果未指定 start_from,就设置到当前的末尾
93
+ if start_from is None:
94
+ start_from = self.readonly_now_node
95
+
96
+ # 如果不在当前分支,就切换过去
97
+ uuids, messages = self.message_list
98
+ if start_from not in uuids:
99
+ self.move_to_node(start_from)
100
+ uuids, messages = self.message_list
101
+
102
+ # 如果节点类型是 assistant,就获从其父节点出截断,否则从节点处截断
103
+ if self.get_message_by_uuid(start_from).role == 'assistant':
104
+ messages = messages[:uuids.index(start_from)]
105
+ append_point_uuid = self.get_parent(start_from)
106
+ else:
107
+ messages = messages[:uuids.index(start_from) + 1]
108
+ append_point_uuid = start_from
109
+
110
+ # 调用并进行处理
111
+ for event in self.main_model.chat(messages):
112
+ stop: bool|None = yield event
113
+
114
+ if event._type == NodeEventTypes.ERROR:
115
+ stop = True
116
+ elif event._type == NodeEventTypes.FINISNED:
117
+ with self.tree as tree:
118
+ tree.insert(event.node, append_point_uuid)
119
+ append_point_uuid = event.node.uuid
120
+
121
+ if stop:
122
+ return
123
+
124
+ def send(self, message: IlinaMessage) -> UUID:
125
+ """ 将消息插入到当前节点之后,会返回新节点的 UUID """
126
+ self.log.info(f'插入用户消息...')
127
+ new_node = Node(message)
128
+ with self.tree as tree:
129
+ if not tree.insert(new_node, tree.now_node.uuid):
130
+ raise NodeNotFoundError(tree.now_node.uuid)
131
+ return new_node.uuid
132
+
133
+
134
+ def move_to_node(self, uuid: UUID):
135
+ """ 将指针设置到指定 UUID 所在分支的末尾 """
136
+ self.log.info(f'移动指针到 {uuid}')
137
+ with self.tree as tree:
138
+ if not tree.set_pointer(uuid):
139
+ raise NodeNotFoundError(uuid)
140
+
141
+
142
+ def delete_node(self, uuid: UUID):
143
+ """ 删除指定 UUID 的节点 """
144
+ self.log.info(f'删除节点: {uuid}')
145
+ with self.tree as root_node:
146
+ if not root_node.delete(uuid):
147
+ raise NodeNotFoundError(uuid)
@@ -0,0 +1,23 @@
1
+ # 自定义异常
2
+
3
+ from uuid import UUID
4
+
5
+ class NodeNotFoundError(Exception):
6
+ """ 未找到指定 UUID 所对应的节点 """
7
+ def __init__(self, uuid: UUID, *args: object) -> None:
8
+ message = f'未找到 UUID 为 {uuid} 的节点'
9
+ super().__init__(message, *args)
10
+ self.message = message
11
+
12
+ def __str__(self):
13
+ return self.message
14
+
15
+ class ParentNotFoundError(Exception):
16
+ """ 未找到指定 UUID 所对应的父节点 """
17
+ def __init__(self, uuid: UUID, *args: object) -> None:
18
+ message = f'未找到 UUID 为 {uuid} 的节点的父节点'
19
+ super().__init__(message, *args)
20
+ self.message = message
21
+
22
+ def __str__(self):
23
+ return self.message
@@ -0,0 +1,173 @@
1
+ # sync_mcp.py
2
+ import json
3
+ import asyncio
4
+ import threading
5
+ import logging
6
+
7
+ from mcp import ClientSession, StdioServerParameters
8
+ from mcp.client.stdio import stdio_client
9
+ from openai.types.chat import ChatCompletionFunctionToolParam
10
+ from openai.types.shared_params import FunctionDefinition
11
+ from FovesConfig import ConfigLoader
12
+
13
+ from ._ilina_message import IlinaToolDefinition, IlinaToolCall
14
+ from ._config_models import AIConfig
15
+ from FovesLog import LoggedTask
16
+
17
+ class SyncMcpClient:
18
+ """
19
+ 同步 MCP 客户端 — 内部维护一个后台事件循环,对外暴露同步方法。
20
+ 可以直接嵌入同步的模型调用循环。
21
+ """
22
+
23
+ def __init__(self, name: str):
24
+ self._loop: asyncio.AbstractEventLoop|None = None
25
+ self._thread: threading.Thread|None = None
26
+ self._session: ClientSession|None = None
27
+ self._ready = threading.Event()
28
+ self._ctx = None
29
+ self.name = name
30
+ self.log = logging.getLogger('MCP Client')
31
+ logging.getLogger('mcp').setLevel(logging.DEBUG)
32
+
33
+ # ── 连接 ──────────────────────────────────
34
+ def connect(self, command: str, args: list[str]):
35
+ """
36
+ 启动后台线程,连接 MCP Server。
37
+ 调用会阻塞直到握手完成。
38
+ """
39
+
40
+ async def _connect():
41
+ server_params = StdioServerParameters(
42
+ command=command,
43
+ args=args,
44
+ env={"PYTHONUNBUFFERED": "1"}, # 禁用 stdout 缓冲,确保 JSON-RPC 响应即时发送
45
+ )
46
+ # 注意:不能用 async with,因为要跨调用保持连接
47
+ self._ctx = stdio_client(server_params)
48
+ read, write = await self._ctx.__aenter__()
49
+ self._session = ClientSession(read, write)
50
+ await self._session.__aenter__()
51
+ await self._session.initialize()
52
+ self._ready.set() # 通知主线程:好了
53
+
54
+ def _run_loop():
55
+ self._loop = asyncio.new_event_loop()
56
+ asyncio.set_event_loop(self._loop)
57
+ self._loop.run_until_complete(_connect())
58
+ # 连接完成后,事件循环继续跑,处理后续调用
59
+ self._loop.run_forever()
60
+
61
+ self._thread = threading.Thread(target=_run_loop, daemon=True)
62
+ self._thread.start()
63
+ self._ready.wait() # 阻塞直到连接完成
64
+
65
+ # ── 工具列表 ──────────────────────────────
66
+ def list_tools(self) -> list[IlinaToolDefinition]:
67
+ """同步获取工具列表"""
68
+ async def _list():
69
+ assert self._session is not None
70
+ result = await self._session.list_tools()
71
+ return [
72
+ IlinaToolDefinition(
73
+ name=self.name + '_' + t.name,
74
+ description=t.description or '',
75
+ arguments=t.inputSchema
76
+ ) for t in result.tools
77
+ ]
78
+ return self._run(_list())
79
+
80
+ # ── 调用工具 ★ 你最常用的 ─────────────────
81
+ def call_tool(self, name: str, arguments: dict) -> str:
82
+ """
83
+ 同步调用 MCP 工具,返回文本结果。
84
+ 直接嵌入你的工具调用循环。
85
+ """
86
+ self.log.info(f'正在调用工具:{name}, args:\n{arguments}')
87
+ # 去除 MCP 服务器名的前缀
88
+ name = name[len(self.name) + 1:]
89
+ async def _call():
90
+ assert self._session is not None
91
+
92
+ self.log.debug(f'调用工具')
93
+ result = self._session.call_tool(name, arguments)
94
+ self.log.debug(f'await 等待 result')
95
+ result = await result
96
+ # [2026/06/07] 无论如何都不要把上面的调用方式改成下面这种。会卡住。
97
+ # result = await self._session.call_tool(name, arguments)
98
+ self.log.debug(f'{result=}')
99
+
100
+ # 优先结构化输出
101
+ if result.structuredContent:
102
+ return str(result.structuredContent)
103
+ # 降级:拼接文本
104
+ self.log.warning(f'降级了消息输出,使用拼接文本')
105
+ parts = []
106
+ for block in result.content:
107
+ if hasattr(block, "text"):
108
+ parts.append(block.text) # pyright: ignore[reportAttributeAccessIssue]
109
+
110
+ self.log.debug(f'return={"\\n".join(parts) if parts else str(result.content)}')
111
+ return "\n".join(parts) if parts else str(result.content)
112
+
113
+ return self._run(_call())
114
+
115
+ # ── 关闭 ──────────────────────────────────
116
+ def close(self):
117
+ """清理资源"""
118
+ async def _close():
119
+ if self._session:
120
+ await self._session.__aexit__(None, None, None)
121
+ if self._ctx:
122
+ await self._ctx.__aexit__(None, None, None) # pyright: ignore[reportGeneralTypeIssues]
123
+
124
+ if self._loop and self._loop.is_running():
125
+ self._loop.call_soon_threadsafe(self._loop.stop)
126
+ if self._thread:
127
+ self._thread.join(timeout=5)
128
+
129
+ # ── 内部 ──────────────────────────────────
130
+ def _run(self, coro):
131
+ """把协程丢到后台事件循环里跑,阻塞等结果"""
132
+ assert self._loop is not None
133
+ fut = asyncio.run_coroutine_threadsafe(coro, self._loop)
134
+ return fut.result() # 阻塞直到完成
135
+
136
+
137
+ class MCPLoader:
138
+ ''' 负责管理和调用 MCP 工具 '''
139
+ def __init__(self):
140
+ self.log = logging.getLogger(f'MCP Loader')
141
+ self.log.setLevel(logging.INFO)
142
+ # 从配置中读取MCP工具
143
+ self.clients: dict[str, SyncMcpClient] = {}
144
+ with ConfigLoader('./configs/ai.json', AIConfig) as config:
145
+ with LoggedTask('加载 MCP 服务', logger=self.log) as task:
146
+ try:
147
+ for mcp_name in config.mcps:
148
+ self.clients[mcp_name] = SyncMcpClient(mcp_name)
149
+ self.clients[mcp_name].connect(config.mcps[mcp_name].command, config.mcps[mcp_name].args)
150
+ except TypeError:
151
+ self.clients = {}
152
+
153
+ def get_list_openai(self) -> list[ChatCompletionFunctionToolParam]:
154
+ """ 返回可以传给OpenAI模型调用的列表 """
155
+ total: list[ChatCompletionFunctionToolParam] = []
156
+ for client in self.clients.values():
157
+ ilina_tools = client.list_tools()
158
+ for tool in ilina_tools:
159
+ total.append(ChatCompletionFunctionToolParam(
160
+ type='function',
161
+ function=FunctionDefinition(
162
+ name=tool.name,
163
+ description=tool.description,
164
+ parameters=tool.arguments,
165
+ )))
166
+ return total
167
+
168
+ def call(self, call: IlinaToolCall) -> str:
169
+ """ 调用MCP工具 """
170
+ for client_name in self.clients:
171
+ if call.name.startswith(client_name):
172
+ return self.clients[client_name].call_tool(call.name, json.loads(call.arguments))
173
+ return f'未找到工具“{call.name}”,请检查名称'
@@ -0,0 +1,19 @@
1
+ # 负责处理系统提示相关的内容
2
+ import os
3
+ from FovesConfig import ConfigLoader
4
+ from ._config_models import AIConfig
5
+ from ._ilina_message import IlinaMessage
6
+
7
+ def load_default_sysprompt(replace_dict: dict[str, str]) -> IlinaMessage:
8
+ """ 警告:尽量减少向系统提示里放置的东西。 """
9
+ with ConfigLoader('./configs/ai.json', AIConfig) as config:
10
+ if os.path.exists(config.default_system_prompt_template):
11
+ with open(config.default_system_prompt_template, 'r', encoding='utf-8') as f:
12
+ prompt = f.read()
13
+ else:
14
+ prompt = config.default_system_prompt_template
15
+
16
+ for key in replace_dict:
17
+ prompt = prompt.replace('{{'+ key +'}}', replace_dict[key])
18
+
19
+ return IlinaMessage(role='system', content=prompt)
IlinaEngine/tools.py ADDED
@@ -0,0 +1,72 @@
1
+ # 管理工具和技能。
2
+ import json
3
+ import inspect
4
+ from pydantic import create_model
5
+ from typing import Callable, get_type_hints
6
+ from ._ilina_message import IlinaMessage
7
+
8
+ def func_to_openai_tool(func, description: str|None = None):
9
+ """
10
+ 把一个 Python 函数直接变成 OpenAI tools 列表中的一项。
11
+
12
+ OpenAI tools 格式:
13
+ {
14
+ "type": "function",
15
+ "function": {
16
+ "name": "...",
17
+ "description": "...",
18
+ "parameters": { ... JSON Schema ... }
19
+ }
20
+ }
21
+ """
22
+ sig = inspect.signature(func)
23
+ hints = get_type_hints(func)
24
+
25
+ # 工具描述:显式传入 > docstring 第一段
26
+ desc = description
27
+ if desc is None and func.__doc__:
28
+ desc = func.__doc__.split("\n\n")[0].strip()
29
+
30
+ # 构建参数字段
31
+ fields = {}
32
+ for name, param in sig.parameters.items():
33
+ if name in ("self", "cls"):
34
+ continue
35
+ py_type = hints.get(name, str)
36
+ default = ... if param.default is inspect.Parameter.empty else param.default
37
+ fields[name] = (py_type, default)
38
+
39
+ Model = create_model(f"{func.__name__}_params", **fields)
40
+ params_schema = Model.model_json_schema()
41
+
42
+ return {
43
+ "type": "function",
44
+ "function": {
45
+ "name": func.__name__,
46
+ "description": desc or "",
47
+ "parameters": params_schema,
48
+ }
49
+ }
50
+
51
+ class Toolset:
52
+ def __init__(self):
53
+ self.tool_functions: dict[str, Callable] = {} # 键是工具名称,值是工具函数。
54
+ self.tool_explains: dict[str, dict] = {} # 键是工具名称,值是工具说明。
55
+
56
+ def to_toollist(self) -> list[dict]:
57
+ return list(self.tool_explains.values())
58
+
59
+ def add_tool(self, func: Callable):
60
+ tool_name = func.__name__
61
+ self.tool_functions[tool_name] = func
62
+ self.tool_explains[tool_name] = func_to_openai_tool(func)
63
+
64
+ def call_tool(self, tools: list[list[str]]) -> list[IlinaMessage]:
65
+ results = []
66
+ for tool in tools:
67
+ name, param, call_id = tool
68
+ func = self.tool_functions[name]
69
+ param_dict = json.loads(param)
70
+ result = func(**param_dict)
71
+ results.append(IlinaMessage(role='tool', content=result, tool_call_id=call_id))
72
+ return results
IlinaEngine/tree.py ADDED
@@ -0,0 +1,344 @@
1
+ import os
2
+ import json
3
+ import copy
4
+ import time
5
+ import logging
6
+ from uuid import UUID, uuid4
7
+ from typing import TypedDict, Required, get_type_hints, Generator
8
+ from pathlib import Path
9
+ from pydantic import ValidationError
10
+ from FovesLog import LoggedTask
11
+ from FovesConfig import ConfigLoader
12
+
13
+ from .sysprompt import load_default_sysprompt
14
+ from ._ilina_message import IlinaMessage
15
+ from ._config_models import IlinaConfig
16
+
17
+ class Node:
18
+ def __init__(self,
19
+ message: IlinaMessage,
20
+ uuid: UUID|None = None,
21
+ pointer: UUID|None = None,
22
+ children: list["Node"]|None = None,
23
+ ):
24
+
25
+ self.message: IlinaMessage = copy.deepcopy(message)
26
+ self.uuid: UUID = uuid or uuid4()
27
+ self.pointer: UUID|None = pointer or None
28
+ self.children: list[Node] = copy.deepcopy(children) or []
29
+
30
+ def _get_pointed_child(self) -> "Node|None":
31
+ """ 返回指向的节点 """
32
+ for child in self.children:
33
+ if child.uuid == self.pointer:
34
+ return child
35
+ return None
36
+
37
+ def _get_leaves(self) -> list["Node"]:
38
+ """ 返回所有的叶子节点 """
39
+ if len(self.children) == 0:
40
+ return [self]
41
+ else:
42
+ leaves = []
43
+ for child in self.children:
44
+ leaves += child._get_leaves()
45
+ return leaves
46
+
47
+ def _insert(self, new_node: "Node", uuid: UUID) -> bool:
48
+ """ 在 uuid 的节点新增一个子节点 new_node,会自动向下搜索。如果没有找到,就会返回 False"""
49
+ if self.uuid == uuid:
50
+ self.children.append(new_node)
51
+ self.pointer = new_node.uuid
52
+ return True
53
+ else:
54
+ for child in self.children:
55
+ if child._insert(new_node, uuid):
56
+ return True
57
+ return False
58
+
59
+ def _set_pointer(self, uuid: UUID) -> bool:
60
+ """ 设置指针到某个 UUID 的节点,会自动向下搜索。如果没有找到,就会返回 False,会自动设置整个链路的指针 """
61
+ if self.uuid == uuid:
62
+ return True
63
+ else:
64
+ for child in self.children:
65
+ if child._set_pointer(uuid):
66
+ self.pointer = child.uuid
67
+ return True
68
+ return False
69
+
70
+ def _delete(self, uuid: UUID) -> bool:
71
+ """ 删除某个 UUID 的节点,会自动向下搜索。如果没有找到,就会返回 False """
72
+ length = len(self.children)
73
+ self.children = list(filter(lambda child: child.uuid != uuid, self.children))
74
+ # 移动指针
75
+ if self.pointer not in self.children:
76
+ if len(self.children) == 0:
77
+ self.pointer = None
78
+ else:
79
+ self.pointer = self.children[0].uuid
80
+
81
+ if len(self.children) < length:
82
+ return True
83
+ else:
84
+ for child in self.children:
85
+ if child._delete(uuid):
86
+ return True
87
+ return False
88
+
89
+ def _get_parent(self, uuid: UUID) -> UUID|None:
90
+ """ 返回指定节点的父节点的 UUID,None 表示未找到 """
91
+ for child in self.children:
92
+ if child.uuid == uuid:
93
+ return self.uuid
94
+ else:
95
+ result = child._get_parent(uuid)
96
+ if result:
97
+ return result
98
+
99
+ def _get_now(self) -> 'Node':
100
+ """ 返回当前指向的最深的节点 """
101
+ for child in self.children:
102
+ if child.uuid == self.pointer:
103
+ return child._get_now()
104
+ return self
105
+
106
+ def walk(self) -> Generator["Node", None, None]:
107
+ yield self
108
+ for child in self.children:
109
+ yield from child.walk()
110
+
111
+ def __hash__(self) -> int:
112
+ return hash(self.uuid)
113
+
114
+ def __eq__(self, value: object) -> bool:
115
+ return (isinstance(value, Node) and self.uuid == value.uuid)
116
+
117
+ def __str__(self) -> str:
118
+ s = f'[{self.message.role}]: {self.message.content}\n'
119
+ for child in self.children:
120
+ if child.uuid == self.pointer:
121
+ s += str(child)
122
+ return s
123
+
124
+ def __repr__(self) -> str:
125
+ role = self.message.role
126
+ content = self.message.content
127
+ if isinstance(content, list):
128
+ content = "[...]"
129
+ preview = str(content)[:20].replace('\n', '\\n') + ("..." if len(str(content)) > 20 else "")
130
+ result = f'Node [{role}] {preview} ({str(self.uuid)})'
131
+ if not self.children:
132
+ return result
133
+ for i, child in enumerate(self.children):
134
+ is_last = (i == len(self.children) - 1)
135
+ is_pointer = (child.uuid == self.pointer)
136
+ branch = '└── ' if is_last else '├── '
137
+ indent = ' ' if is_last else '│ '
138
+ marker = '▶ ' if is_pointer else ''
139
+ child_lines = repr(child).split('\n')
140
+ result += '\n' + branch + marker + child_lines[0]
141
+ for line in child_lines[1:]:
142
+ result += '\n' + indent + line
143
+ return result
144
+
145
+ def _to_message_list(self) -> tuple[list[UUID], list[IlinaMessage]]:
146
+ """ 将现在的列表转化为可以传递给 AI 的消息列表 """
147
+ uuids = [self.uuid]
148
+ messages = [self.message]
149
+ for child in self.children:
150
+ if child.uuid == self.pointer:
151
+ new_uuid, new_messages = child._to_message_list()
152
+ uuids += new_uuid
153
+ messages += new_messages
154
+ return (uuids, messages)
155
+
156
+ def _to_list(self) -> list['Node']:
157
+ """ 获取当前所指向的节点链 """
158
+ s: list[Node] = [self]
159
+ for child in self.children:
160
+ if child.uuid == self.pointer:
161
+ s += child._to_list()
162
+ return s
163
+
164
+ def node_json_dump(obj):
165
+ if isinstance(obj, Node):
166
+ return {
167
+ "message": obj.message.model_dump(),
168
+ "uuid": str(obj.uuid),
169
+ "pointer": str(obj.pointer) if obj.pointer else None,
170
+ "children": [node_json_dump(child) for child in obj.children],
171
+ }
172
+ else:
173
+ return obj
174
+
175
+ def node_json_load(obj):
176
+ if isinstance(obj, dict):
177
+ if "message" in obj:
178
+ return Node(
179
+ message=IlinaMessage(**obj["message"]),
180
+ uuid=UUID(obj["uuid"]),
181
+ pointer=UUID(obj["pointer"]) if obj["pointer"] else None,
182
+ children=[node_json_load(child) for child in obj["children"]], # pyright: ignore[reportArgumentType]
183
+ )
184
+ else:
185
+ return obj
186
+ else:
187
+ return obj
188
+
189
+ class SaveData(TypedDict): # 这是保存的文件里面的内容
190
+ create_time: Required[int|float]
191
+ root_node: Required[Node]
192
+
193
+ class Tree:
194
+ """ 对话树,会同步到文件 """
195
+ def __init__(self, full_path: str|Path): # 创建新的树,或者从文件加载树。
196
+ self.log = logging.getLogger(f'对话树')
197
+ self.log.setLevel(logging.INFO)
198
+ if isinstance(full_path, str):
199
+ full_path = Path(full_path)
200
+ self._update_paths(full_path)
201
+ self.uuid_to_node_table: dict[UUID, Node] = {}
202
+ self._load()
203
+
204
+ def _update_paths(self, new_path: Path):
205
+ """ 修改树中的各种路径 """
206
+ self.full_path: Path = new_path
207
+ self.workpath: Path = new_path.parent
208
+ self.file_name: str = new_path.stem
209
+
210
+ @property
211
+ def name(self) -> str:
212
+ return self.file_name
213
+
214
+ @name.setter
215
+ def name(self, value: str):
216
+ new_path = self.full_path.with_stem(value)
217
+ if self.full_path.is_file():
218
+ self.full_path.rename(new_path)
219
+ self._update_paths(new_path)
220
+ self._save()
221
+
222
+ @property
223
+ def root_node(self) -> Node:
224
+ return self.save_data["root_node"]
225
+
226
+ def insert(self, new_node: "Node", uuid: UUID) -> bool:
227
+ """ 会同步更新 UUID 到节点映射表的 insert
228
+ 不要直接调用根节点的 _insert
229
+ """
230
+ if self.root_node._insert(new_node, uuid):
231
+ self.uuid_to_node_table[new_node.uuid] = new_node
232
+ return True
233
+ else:
234
+ return False
235
+
236
+ def delete(self, uuid: UUID) -> bool:
237
+ """ 会同步更新 UUID 到节点映射表的删除
238
+ 不要直接调用根节点的 _delete
239
+ 如果没找到返回 False
240
+
241
+ """
242
+ if self.root_node._delete(uuid):
243
+ del self.uuid_to_node_table[uuid]
244
+ return True
245
+ else:
246
+ return False
247
+
248
+ @property
249
+ def now_node(self) -> Node:
250
+ """ 获取当前指向的叶子节点 """
251
+ return self.root_node._get_now()
252
+
253
+ def set_pointer(self, uuid: UUID) -> bool:
254
+ """ 设置pointer """
255
+ return self.root_node._set_pointer(uuid)
256
+
257
+ def _save(self) -> None:
258
+ """ 将 save_data 同步到文件 """
259
+ with LoggedTask('将树写入到文件', logger=self.log):
260
+ os.makedirs(self.full_path.parent, exist_ok=True)
261
+ with open(self.full_path, 'w', encoding='utf-8') as f:
262
+ json.dump(self.save_data, f, default=node_json_dump, indent=4, ensure_ascii=False)
263
+
264
+ def _get_default_save_data(self) -> SaveData:
265
+ self._load_floder_config()
266
+ config_name = self.full_path.parent / '.ilinaconfig'
267
+ if config_name.exists() and config_name.is_file():
268
+ config = ConfigLoader(config_name, IlinaConfig).readonly() # 只读
269
+ else:
270
+ config = IlinaConfig()
271
+ return SaveData( # 创建初始信息
272
+ create_time=time.time(),
273
+ root_node=Node(message=load_default_sysprompt(
274
+ replace_dict={
275
+ 'workpath': str(self.workpath),
276
+ 'open_or_alarm': '打开那个文件且不发送通知' if config.open_or_alarm else '不打开那个文件而是发送通知'
277
+ })))
278
+
279
+ def _load(self) -> None:
280
+ """ 从文件同步 save_data """
281
+ with LoggedTask('从文件加载节点树', logger=self.log) as task:
282
+ try:
283
+ with open(self.full_path, 'r', encoding='utf-8') as f:
284
+ data: SaveData = json.load(f, object_hook=node_json_load)
285
+ if data == {}:
286
+ self.save_data: SaveData = self._get_default_save_data()
287
+ self._save()
288
+ return
289
+ except FileNotFoundError: # 如果文件不存在,就用现有信息保存
290
+ self.save_data: SaveData = self._get_default_save_data()
291
+ self._save()
292
+ return
293
+ except json.JSONDecodeError:
294
+ with open(self.full_path, 'r', encoding='utf-8') as f:
295
+ if f.read().strip() == '':
296
+ self.save_data: SaveData = self._get_default_save_data()
297
+ self._save()
298
+ return
299
+ raise ValueError(f'文件 "{self.full_path}" 的格式错误:不是有效的 JSON')
300
+ except ValidationError:
301
+ raise ValueError(f'文件 "{self.full_path}" 的格式错误:不是有效的 JSON')
302
+ task.checkpoint(f'读取完成')
303
+
304
+ # 进行格式检查
305
+ hints = get_type_hints(SaveData)
306
+ for key in hints.keys():
307
+ try:
308
+ if not isinstance(data[key], hints[key]):
309
+ raise ValueError(f'文件"{self.full_path}"的格式错误:"{key}" 的类型应为 "{hints[key]}", 却找到了 "{type(data[key])}"')
310
+ except KeyError:
311
+ raise ValueError(f'文件"{self.full_path}"的格式错误:缺少键 "{key}"')
312
+ self.save_data = data
313
+ task.checkpoint(f'格式检查完成')
314
+
315
+ # 产生节点映射表
316
+ for node in self.root_node.walk():
317
+ self.uuid_to_node_table[node.uuid] = node
318
+ task.checkpoint(f'节点映射表编写完成')
319
+
320
+ self._load_floder_config()
321
+
322
+ def _load_floder_config(self):
323
+ with LoggedTask('查找并加载工作区配置', logger=self.log) as task:
324
+ config_name = self.full_path.parent / '.ilinaconfig'
325
+ if config_name.exists() and config_name.is_file():
326
+ with ConfigLoader(config_name, IlinaConfig) as config:
327
+ if config.workpath:
328
+ self.workpath = Path(config.workpath)
329
+ self.log.info(f'将工作目录更新为:{self.workpath}')
330
+ else:
331
+ self.log.info(f'工作区配置未找到')
332
+
333
+ def __enter__(self) -> "Tree":
334
+ self._load()
335
+ return self
336
+
337
+ def __exit__(
338
+ self,
339
+ exc_type: type[BaseException] | None,
340
+ exc_val: BaseException | None,
341
+ exc_tb: object | None,
342
+ ) -> None:
343
+ self.log.debug(f'预备保存的对话树:\n{repr(self.save_data["root_node"])}')
344
+ self._save()
IlinaEngine/type.py ADDED
@@ -0,0 +1,8 @@
1
+ from .call_openai import (
2
+ NodeEvent,
3
+ NodeEventTypes,
4
+ )
5
+
6
+ from ._ilina_message import IlinaMessage, IlinaToolCall
7
+
8
+ from .tree import Node
@@ -0,0 +1,189 @@
1
+ Metadata-Version: 2.4
2
+ Name: IlinaEngine
3
+ Version: 0.8.1
4
+ Summary: Ilina Engine 是一个以对话树为核心的 AI 聊天后端系统。支持树状对话、读写文件、MCP 和 Skills
5
+ Project-URL: Repository, https://github.com/Foves7017/IlinaEngine
6
+ Author: Foves7017
7
+ License: MIT License
8
+
9
+ Copyright (c) 2026 Foves7017
10
+
11
+ Permission is hereby granted, free of charge, to any person obtaining a copy
12
+ of this software and associated documentation files (the "Software"), to deal
13
+ in the Software without restriction, including without limitation the rights
14
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
15
+ copies of the Software, and to permit persons to whom the Software is
16
+ furnished to do so, subject to the following conditions:
17
+
18
+ The above copyright notice and this permission notice shall be included in all
19
+ copies or substantial portions of the Software.
20
+
21
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
22
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
24
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
26
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
27
+ SOFTWARE.
28
+ License-File: LICENSE
29
+ Requires-Python: >=3.12
30
+ Requires-Dist: fovesconfig
31
+ Requires-Dist: foveslog
32
+ Requires-Dist: mcp
33
+ Requires-Dist: openai
34
+ Description-Content-Type: text/markdown
35
+
36
+ # IlinaEngine
37
+ Ilina Engine 是一个以对话树为核心的 AI 聊天后端系统。支持树状对话、读写文件、MCP 和 Skills
38
+
39
+ ## 重要说明
40
+ 当前版本 (0.8.x) 还是一个开发中版本,许多功能也许不稳定,也有可能会增删接口,请务必注意。
41
+
42
+ 关于版本更新的详细内容请见后文。
43
+
44
+ ## 安装方法
45
+ Ilina Engine 已经发布到 PyPI,你可以使用下面的命令安装:
46
+
47
+ ```
48
+ pip install IlinaEngine
49
+ ```
50
+
51
+ 你也可以前往这个页面,看看我为 Ilina Engine 开发的 GUI,然后直接使用。
52
+ https://github.com/Foves7017/IlinaGUI
53
+
54
+ 本项目的地址:
55
+ https://github.com/Foves7017/IlinaEngine
56
+
57
+ # 配置 IlinaEngine
58
+ 首次启动后,IlinaEngine会使用默认配置创建 `./configs/ai.json`,你可以在启动前手动创建它,也可以启动后手动修改,但之后可能需要重新运行程序。
59
+ [这里](./docs/config.md)介绍了所有的配置项。
60
+
61
+ 请首先至少配置可用的主要模型和辅助模型。
62
+ # 对话引擎
63
+ 对于对话树的操作全部可以通过 Engine 模块实现,Engine 模块同时也提供了直接调用 API 生成回复的能力。
64
+ [这里](./docs/engine.md)记录了所有引擎可以调用的接口,上述的 GUI 则是使用这些接口的示例。
65
+
66
+ # 对话树(Conversation Tree)
67
+
68
+ 与传统聊天软件的线性消息列表不同,IlinaEngine 使用**树结构(Tree)**存储整个对话。
69
+
70
+ 每一条消息都是一个节点(Node),节点之间通过父子关系连接,形成一棵完整的对话树。
71
+
72
+ ## 基本结构
73
+
74
+ ```text
75
+ System
76
+ └── User:你好
77
+ └── Assistant:你好,请问有什么可以帮助你?
78
+ ├── User:介绍一下 Python
79
+ │ └── Assistant:Python 是一种...
80
+
81
+ └── User:介绍一下 C++
82
+ └── Assistant:C++ 是一种...
83
+ ```
84
+
85
+ 每个节点都可以是:
86
+
87
+ * User(用户消息)
88
+ * Assistant(模型回复)
89
+ * Tool(工具调用结果)
90
+ * System(系统提示词)
91
+ * Error(错误节点)
92
+
93
+ 树中的每一条路径都代表一次完整的对话上下文。
94
+
95
+ ---
96
+
97
+ ## 分支机制
98
+
99
+ 在任意节点上:
100
+
101
+ * 修改消息
102
+ * 重新生成回答
103
+ * 从历史节点继续对话
104
+
105
+ 都不会覆盖原有内容。
106
+
107
+ 系统会创建一个新的子节点作为新的分支。
108
+
109
+ 例如:
110
+
111
+ ```text
112
+ User:推荐一本书
113
+ └── Assistant:推荐《三体》
114
+ ├── User:为什么?
115
+ │ └── Assistant:因为...
116
+
117
+ └── User:换一本
118
+ └── Assistant:推荐《银河帝国》
119
+ ```
120
+
121
+ 原始对话会被完整保留。
122
+
123
+ 你可以随时切换到任意分支继续对话。
124
+
125
+ ---
126
+
127
+ ## 当前分支
128
+
129
+ 每个节点都记录一个指向子节点的指针(Pointer)。
130
+
131
+ Pointer 用来标记:
132
+
133
+ > “当前正在使用哪一条分支”。
134
+
135
+ 例如:
136
+
137
+ ```text
138
+ User
139
+ └── Assistant
140
+ ├── User A
141
+ │ └── Assistant A
142
+
143
+ └▶ User B
144
+ └▶ Assistant B
145
+ ```
146
+
147
+ 带有 `▶` 标记的路径就是当前活跃分支。
148
+
149
+ 当模型生成回复时,系统会沿着这条路径向下收集消息,构造发送给模型的上下文。
150
+
151
+ ---
152
+
153
+ ## 修改历史
154
+
155
+ 由于所有旧节点都会被保留,因此:
156
+
157
+ * 可以查看任意历史版本
158
+ * 可以比较不同回答
159
+ * 可以回到过去的任意时间点
160
+ * 可以从任意节点创建新的对话路线
161
+
162
+ 整个对话过程类似于 Git 的提交树:
163
+
164
+ ```text
165
+ A
166
+ └── B
167
+ ├── C
168
+ │ └── D
169
+
170
+ └── E
171
+ └── F
172
+ ```
173
+
174
+ 不同的是,这里的节点不是代码提交,而是一次次对话消息。
175
+
176
+ ---
177
+
178
+ ## 优势
179
+
180
+ 相比传统线性聊天记录,对话树能够:
181
+
182
+ * 保留所有历史版本
183
+ * 支持无限分支探索
184
+ * 支持多次重新生成而不丢失旧结果
185
+ * 支持修改任意历史消息
186
+ * 支持从任意节点继续对话
187
+ * 为 AI Agent、工具调用与长期记忆提供更灵活的上下文管理能力
188
+
189
+ 对话不再是一条线,而是一张不断生长的思维树。
@@ -0,0 +1,15 @@
1
+ IlinaEngine/__init__.py,sha256=aguBEuGH-tkbx1dWOi_tNk1DOx4p0WMyA9jo0mGYbEM,100
2
+ IlinaEngine/_config_models.py,sha256=AbTDCNt-JhGVzyQKhSGxP-okFHKXlwgPH2ZLgkOIzXY,985
3
+ IlinaEngine/_ilina_message.py,sha256=mg-S6pg6DVIpNvuCaeHKPgZPoM876Nrpuu5SvwEFMus,961
4
+ IlinaEngine/call_openai.py,sha256=J-KcQ4mLVe2iB8m4wgtdGFp7j5JN6gFXWZmEXEkHGI0,10053
5
+ IlinaEngine/engine.py,sha256=IVGcedVBZHuYSc8L6WoRlikw5fpkrDxrlBc6hd__KEI,5603
6
+ IlinaEngine/exceptions.py,sha256=K28BRBlG-JKfanEk426Y9xZIOtKXmUZ_SWHUShy7l0U,745
7
+ IlinaEngine/sync_mcp.py,sha256=yJtvStrv46SdHqkRQMPuZVVd0SmX4kycHgbFiNJBNqg,7533
8
+ IlinaEngine/sysprompt.py,sha256=FXoe8XmOJFzGKz4fwWHpQ7c8CusOh7aMNMQA2tEHAHA,814
9
+ IlinaEngine/tools.py,sha256=kxp4j8f2h_73RGS2CXe4ZfRAoe0PImaDT624AzVVUTo,2387
10
+ IlinaEngine/tree.py,sha256=iaYtWFN0I8wM0HZVjLucE57-rn8FQFmHliZ1YGxT92U,13469
11
+ IlinaEngine/type.py,sha256=H8QeO3Rba56MCop7AAkpBTQfg8H4JOhN_P2zY_xDNro,151
12
+ ilinaengine-0.8.1.dist-info/METADATA,sha256=tQRRBs6Jr0R04BQIbGGEjJwVPCHIQ7YiM4n_6Ehm0Wk,5690
13
+ ilinaengine-0.8.1.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
14
+ ilinaengine-0.8.1.dist-info/licenses/LICENSE,sha256=lXz0pZeICx_-HWjVeatkJhfuZp5Wd4jOPYzVs57mLfM,1087
15
+ ilinaengine-0.8.1.dist-info/RECORD,,
@@ -0,0 +1,4 @@
1
+ Wheel-Version: 1.0
2
+ Generator: hatchling 1.30.1
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 Foves7017
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.