IlinaEngine 0.8.1__tar.gz

ilinaengine-0.8.1/.gitignore

@@ -0,0 +1,219 @@
+.obsidian
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+# uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#   that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#   you could uncomment the following to ignore the entire vscode folder
+# .vscode/
+# Temporary file for partial code execution
+tempCodeRunnerFile.py
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml

ilinaengine-0.8.1/IlinaEngine/__init__.py

@@ -0,0 +1,2 @@
+from .engine import Engine
+from .type import NodeEvent, NodeEventTypes, IlinaMessage, IlinaToolCall

ilinaengine-0.8.1/IlinaEngine/_config_models.py

@@ -0,0 +1,32 @@
+from pydantic import BaseModel
+
+class IlinaConfig(BaseModel):
+    """ 文件夹下的 .ilinaconfig 内的数据模型 """
+    workpath: str|None = None
+    open_or_alarm: bool = False  # 如果是 True 倾向于打开文件，如果是 False 倾向于发送通知
+
+class MCPConfig(BaseModel):
+    command: str
+    args: list[str]
+
+class ModelConfig(BaseModel):
+    base_url: str
+    api_key: str
+    model_name: str
+
+class AIConfig(BaseModel):
+    """ 模型配置 """
+    main_model: ModelConfig = ModelConfig(
+        base_url='http://localhost:11434/v1/', 
+        api_key='sk-xxx', 
+        model_name='qwen3-vl:4b'
+    )
+
+    sub_model: ModelConfig = ModelConfig(
+        base_url='http://localhost:11434/v1/', 
+        api_key='sk-xxx', 
+        model_name='qwen3-vl:4b'
+    )
+
+    mcps: dict[str, MCPConfig] = {}
+    default_system_prompt_template: str = '当前工作目录为 {{workpath}}，当你编辑文件之后，你应该{{open_or_alarm}}'

ilinaengine-0.8.1/IlinaEngine/_ilina_message.py

@@ -0,0 +1,25 @@
+# Ilina Message
+from typing import Literal
+from pydantic import BaseModel
+
+class IlinaToolDefinition(BaseModel):
+    """ 工具定义 """
+    name: str  # 工具名称
+    description: str  # 工具说明
+    arguments: dict[str, object]  # 参数的 JSON Schema
+
+class IlinaToolCall(BaseModel):
+    """ 工具调用 """
+    name: str = ''
+    arguments: str = ''
+    tool_call_id: str = ''
+
+class IlinaMessage(BaseModel):
+    """ 对话消息 """
+    role: Literal['user', 'assistant', 'system', 'tool', 'error']
+    content: str = ''
+    reasoning_content: str = ''  # 仅在 assistant 中使用
+    tool_calls: list[IlinaToolCall] = []  # 仅在 assistant 中使用
+    tool_call_id: str = ''  # 仅在 tool 中使用
+    tool_name: str = ''  # 保存工具名，仅在 tool 中使用
+    # tool_readable_content: str = ''  # 返回一个人类可读的消息，如果指定了这一项，那么渲染时建议用这个替换content

ilinaengine-0.8.1/IlinaEngine/call_openai.py

@@ -0,0 +1,222 @@
+# 调用 API
+
+from enum import Enum
+from typing import Generator
+from logging import getLogger
+from dataclasses import dataclass
+from FovesConfig import ConfigLoader
+
+
+# from tools import Toolset
+from .tree import Node
+from .sync_mcp import MCPLoader
+from ._config_models import AIConfig
+from ._ilina_message import IlinaMessage, IlinaToolCall
+from openai import OpenAI
+from openai.types.chat import (
+    ChatCompletionAssistantMessageParam,
+    ChatCompletionMessageFunctionToolCallParam,
+    ChatCompletionUserMessageParam,
+    ChatCompletionSystemMessageParam,
+    ChatCompletionToolMessageParam,
+
+    ChatCompletionMessageParam,
+    ChatCompletionChunk,
+    ChatCompletion,
+)
+
+class NodeEventTypes(str, Enum):
+    CREATED = 'CREATED'
+    UPDATED = 'UPDATED'
+    FINISNED = 'FINISHED'
+    ERROR = 'ERROR'
+
+@dataclass
+class NodeEvent:
+    """ 节点发生变化的事件 """
+    node: Node
+    _type: NodeEventTypes
+
+class OpenAIClient:
+    def __init__(self, is_main_model: bool, mcp_loader: MCPLoader):
+        with ConfigLoader('./configs/ai.json', AIConfig) as config:
+            if is_main_model:
+                modelcfg = config.main_model
+            else:
+                modelcfg = config.sub_model
+        self.client = OpenAI(base_url=modelcfg.base_url, api_key=modelcfg.api_key)
+        self.model = modelcfg.model_name
+        # self.toolset = Toolset()
+        self.mcp_loader = mcp_loader
+        self.log = getLogger(f"Model_{self.model}")
+    
+    def ilina_to_openai(self, ilina: IlinaMessage) -> ChatCompletionMessageParam|None:
+        """ 将 IlinaMessage 转化为  ChatCompletionMessageParam, 会过滤掉 error 类型"""
+        if ilina.role == 'assistant':
+            tool_calls = []
+            for tool_call in ilina.tool_calls:
+                tool_calls.append(ChatCompletionMessageFunctionToolCallParam(
+                    type='function',
+                    id=tool_call.tool_call_id,
+                    function={
+                        'name': tool_call.name,
+                        'arguments': tool_call.arguments
+                    }
+                ))
+
+            if len(tool_calls) > 0:
+                return ChatCompletionAssistantMessageParam(
+                    role='assistant',
+                    content=ilina.content,
+                    tool_calls=tool_calls
+                )
+            else:
+                return ChatCompletionAssistantMessageParam(
+                    role='assistant',
+                    content=ilina.content,
+                )
+        
+        elif ilina.role == 'user':
+            return ChatCompletionUserMessageParam(
+                role='user',
+                content=ilina.content
+            )
+        
+        elif ilina.role == 'system':
+            return ChatCompletionSystemMessageParam(
+                role='system',
+                content=ilina.content
+            )
+        
+        elif ilina.role == 'tool':
+            return ChatCompletionToolMessageParam(
+                role='tool',
+                content=ilina.content,
+                tool_call_id=ilina.tool_call_id
+            )
+        
+        elif ilina.role == 'error':
+            return None
+
+    def chat(self, messages: list[IlinaMessage]) -> Generator[NodeEvent, None, None]:
+        """ 调用模型，会首先用生成器返回流失输出结果，最后return合并的Node """
+        new_messages: list[IlinaMessage] = []  # 存储本轮调用生成的消息
+        stop_reason: str = ''  # 存储停止原因
+        self.log.info('开始调用模型')
+        while stop_reason != 'stop':
+            # 发起请求
+            self.log.info('发起 API 请求')
+            try:
+                # 首先将 messages 转换成 openai 格式
+                openai_messages = []
+                for item in map(self.ilina_to_openai, messages + new_messages):
+                    if item is not None:
+                        openai_messages.append(item)
+
+                # 发起调用
+                res = self.client.chat.completions.create(
+                    messages=openai_messages,
+                    model=self.model,
+                    tools=self.mcp_loader.get_list_openai(),
+                    stream=True
+                ) 
+            except Exception as e:
+                yield NodeEvent(Node(IlinaMessage(role='error', content=f'{e}')), NodeEventTypes.ERROR)
+                return
+            
+            assistant_node = Node(IlinaMessage(role='assistant'))
+            # 传递节点开始事件
+            yield NodeEvent(assistant_node, NodeEventTypes.CREATED)
+
+            # 逐帧解析
+            tool_call_nodes: dict[str, Node] = {}  # 保存工具 ID 对应的节点
+            for chunk in res:
+                chunk: ChatCompletionChunk
+                delta = chunk.choices[0].delta
+
+                # 保存流式信息
+                if delta.content:
+                    assistant_node.message.content += delta.content
+                    yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
+
+                # 保存思考信息                
+                reasoning_content_delta = ''
+                try:
+                    self.log.debug(delta)
+                    if hasattr(delta, 'reasoning_content'):  # Deepseek：用 reasoning_content 输出思考流
+                        reasoning_content_delta = delta.reasoning_content or '' # pyright: ignore[reportAttributeAccessIssue]
+                    elif hasattr(delta, 'reasoning'):  # Ollama: 用 reasoning 输出思考流
+                        reasoning_content_delta = delta.reasoning or '' # pyright: ignore[reportAttributeAccessIssue]
+                except AttributeError:  # 如果获取不到，也设置成空
+                    reasoning_content_delta = ''
+                    
+                assistant_node.message.reasoning_content += reasoning_content_delta
+                yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
+                
+                # 保存工具调用
+                if delta.tool_calls:  # 如果工具调用不为 None
+                    for tool_call in delta.tool_calls:  # 对于工具调用列表中的每个工具
+                        try:  # 尝试直接追加到现有索引处
+                            if tool_call.id:  # 覆写 ID
+                                assistant_node.message.tool_calls[tool_call.index].tool_call_id = tool_call.id
+
+                            if tool_call.function:  # 更新 function
+                                if tool_call.function.name:  # 覆写 name
+                                    assistant_node.message.tool_calls[tool_call.index].name = tool_call.function.name
+                                    if tool_call.id:
+                                        tool_call_nodes[tool_call.id].message.tool_name = tool_call.function.name
+                                if tool_call.function.arguments:  # 追加 arguments
+                                    assistant_node.message.tool_calls[tool_call.index].arguments += tool_call.function.arguments
+                            
+                            yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
+
+                        except IndexError:  # 不存在则新增
+                            assert tool_call.id is not None
+                            assert tool_call.function is not None
+                            assert tool_call.function.name is not None
+                            assert tool_call.function.arguments is not None
+                            assistant_node.message.tool_calls.append(IlinaToolCall(
+                                tool_call_id=tool_call.id,
+                                name=tool_call.function.name,
+                                arguments=tool_call.function.arguments
+                            ))
+                            tool_call_node = Node(IlinaMessage(role='tool', tool_call_id=tool_call.id, tool_name=tool_call.function.name))
+                            tool_call_nodes[tool_call.id] = tool_call_node
+                            yield NodeEvent(tool_call_node, NodeEventTypes.CREATED)
+                            yield NodeEvent(assistant_node, NodeEventTypes.UPDATED)
+
+            # 更新停止原因
+            if chunk.choices[0].finish_reason:
+                stop_reason = chunk.choices[0].finish_reason
+            self.log.info(f'停止原因：{stop_reason}')
+
+            # 添加助手信息
+            new_messages.append(assistant_node.message)
+            yield NodeEvent(assistant_node, NodeEventTypes.FINISNED)
+
+            # 调用并添加工具信息
+            # 返回流式传输的工具块
+            for call in assistant_node.message.tool_calls:
+                self.log.info(f'调用工具 {call.name}')
+                result = self.mcp_loader.call(call)
+                self.log.info(f'工具返回 {result}')
+                # 根据 ID 获取对应的节点并进行修改和发送
+                tool_call_node = tool_call_nodes[call.tool_call_id]
+                tool_call_node.message.content = result
+                # 向新消息列表中添加工具消息
+                new_messages.append(tool_call_node.message)
+                yield NodeEvent(tool_call_node, NodeEventTypes.UPDATED)
+                yield NodeEvent(tool_call_node, NodeEventTypes.FINISNED)
+
+            self.log.debug(f'当前的new_messages:\n{'\n'.join([str(m) for m in new_messages])}')
+
+    def once(self, sysprompt: str, user_input: str) -> str|None:
+        """ 进行一次调用 """ 
+        res: ChatCompletion = self.client.chat.completions.create(
+            messages=[
+                {'role': 'system', 'content': sysprompt},
+                {'role': 'user', 'content': user_input}
+            ],
+            model=self.model,
+        )
+        return res.choices[0].message.content

ilinaengine-0.8.1/IlinaEngine/engine.py

@@ -0,0 +1,147 @@
+# 引擎层
+import logging
+from uuid import UUID
+from typing import Generator
+
+from FovesLog import LoggedTask
+from .tree import Tree, Node
+from .call_openai import *
+from ._ilina_message import IlinaMessage
+from .exceptions import *
+
+class Engine:
+    def __init__(self, filename: str) -> None:
+        self.log = logging.getLogger('对话引擎')
+        self.log.setLevel(logging.INFO)
+
+        with LoggedTask('初始化', logger=self.log) as task:
+            self.tree = Tree(filename)
+            task.checkpoint(f'建立文件树')
+            self.mcp_loader = MCPLoader()
+            task.checkpoint(f'建立MCP工具')
+            self.main_model = OpenAIClient(True, self.mcp_loader)
+    
+    @property
+    def workpath(self) -> str:
+        """ 获取工作目录 """
+        return str(self.tree.workpath)
+
+    @property
+    def name(self) -> str:
+        """ 获取树的名字 """
+        return self.tree.name
+    
+    @name.setter
+    def name(self, value: str):
+        self.tree.name = value
+    
+    def set_name(self, new_name: str):
+        """ 设置名字的函数版本，适用于某些无法使用属性赋值的情况
+        
+        说的就是你, lambda 语句
+        """
+        self.name = new_name
+    
+    @property
+    def message_list(self) -> tuple[list[UUID], list[IlinaMessage]]:
+        """ 获取当前的消息链 """
+        return self.tree.root_node._to_message_list()
+
+    @property
+    def readonly_root_node(self) -> Node:
+        """ 获取根节点。注意：只应该用来获取树结构 """
+        return self.tree.root_node
+
+    @property
+    def readonly_leaves(self) -> list[Node]:
+        """ 获取所有叶子节点。注意：只应该用来获取树结构 """
+        return self.tree.root_node._get_leaves()
+
+    @property
+    def readonly_now_node(self) -> UUID:
+        """ 获取当前最新节点的 UUID """
+        return self.tree.root_node._get_now().uuid
+
+    def get_message_by_uuid(self, uuid: UUID) -> IlinaMessage:
+        """ 通过 UUID 获取消息内容 """
+        return self.tree.uuid_to_node_table[uuid].message
+
+    def get_parent(self, uuid: UUID) -> UUID:
+        """ 获取某个节点的父节点。对于根节点和其他找不到父节点的节点会报错 """
+        parent = self.tree.root_node._get_parent(uuid)
+        if parent:
+            return parent
+        else:
+            raise ParentNotFoundError(uuid)
+
+    def edit_node(self, target: UUID, new_message: IlinaMessage) -> UUID:
+        """ 修改节点内容，不会实际修改，而是作为父节点的新子节点插入，会返回新节点的 UUID """
+        new_node = Node(new_message)
+        with self.tree as tree:
+            tree.insert(new_node, self.get_parent(target))
+        return new_node.uuid
+
+    def invoke(self, start_from: UUID|None=None) -> Generator[NodeEvent, bool, None]:
+        """ 将当前的对话发送给主模型，并获取其回复，默认从最新分支开始
+            并会自动将 NodeFinished 事件的节点添加到节点树中
+            如果指定了 start_from，会从那里截断并插入节点。
+            如果 start_from 不在当前分支，会首先切换过去。
+        """
+        self.log.info(f'调用 AI')
+
+        # 如果未指定 start_from，就设置到当前的末尾
+        if start_from is None:
+            start_from = self.readonly_now_node
+
+        # 如果不在当前分支，就切换过去
+        uuids, messages = self.message_list
+        if start_from not in uuids:
+            self.move_to_node(start_from)
+            uuids, messages = self.message_list
+
+        # 如果节点类型是 assistant，就获从其父节点出截断，否则从节点处截断
+        if self.get_message_by_uuid(start_from).role == 'assistant':
+            messages = messages[:uuids.index(start_from)]
+            append_point_uuid = self.get_parent(start_from)
+        else:
+            messages = messages[:uuids.index(start_from) + 1]
+            append_point_uuid = start_from
+
+        # 调用并进行处理
+        for event in self.main_model.chat(messages):
+            stop: bool|None = yield event
+
+            if event._type == NodeEventTypes.ERROR:
+                stop = True
+            elif event._type == NodeEventTypes.FINISNED:
+                with self.tree as tree:
+                    tree.insert(event.node, append_point_uuid)
+                    append_point_uuid = event.node.uuid
+            
+            if stop:
+                return
+    
+    def send(self, message: IlinaMessage) -> UUID:
+        """ 将消息插入到当前节点之后，会返回新节点的 UUID """
+        self.log.info(f'插入用户消息...')
+        new_node = Node(message)
+        with self.tree as tree:
+            if not tree.insert(new_node, tree.now_node.uuid):
+                raise NodeNotFoundError(tree.now_node.uuid)
+        return new_node.uuid
+
+    
+    def move_to_node(self, uuid: UUID):
+        """ 将指针设置到指定 UUID 所在分支的末尾 """
+        self.log.info(f'移动指针到 {uuid}')
+        with self.tree as tree:
+            if not tree.set_pointer(uuid):
+                raise NodeNotFoundError(uuid)
+
+    
+    def delete_node(self, uuid: UUID):
+        """ 删除指定 UUID 的节点 """
+        self.log.info(f'删除节点： {uuid}')
+        with self.tree as root_node:
+            if not root_node.delete(uuid):
+                raise NodeNotFoundError(uuid)

ilinaengine-0.8.1/IlinaEngine/exceptions.py

@@ -0,0 +1,23 @@
+# 自定义异常
+
+from uuid import UUID
+
+class NodeNotFoundError(Exception):
+    """ 未找到指定 UUID 所对应的节点 """
+    def __init__(self, uuid: UUID, *args: object) -> None:
+        message = f'未找到 UUID 为 {uuid} 的节点'
+        super().__init__(message, *args)
+        self.message = message
+    
+    def __str__(self):
+        return self.message
+
+class ParentNotFoundError(Exception):
+    """ 未找到指定 UUID 所对应的父节点 """
+    def __init__(self, uuid: UUID, *args: object) -> None:
+        message = f'未找到 UUID 为 {uuid} 的节点的父节点'
+        super().__init__(message, *args)
+        self.message = message
+    
+    def __str__(self):
+        return self.message