matrix-for-agents 0.1.2__py3-none-any.whl

agentmatrix/agents/post_office.py

@@ -0,0 +1,212 @@
+import asyncio
+from typing import Dict, Optional
+from ..db.database import AgentMailDB
+import os
+import json
+import textwrap
+from pathlib import Path
+from ..core.message import Email
+from ..core.log_util import AutoLoggerMixin
+class PostOffice(AutoLoggerMixin):
+    def __init__(self, matrix_path):
+        self.directory = {}
+        self.queue = asyncio.Queue()
+        email_db_path = os.path.join(matrix_path,".matrix" , "matrix_mails.db")
+        self.email_db = AgentMailDB(email_db_path) # 初始化数据库连接
+        self._paused = False
+        self.vector_db = None
+
+        # 初始化 user_sessions 管理
+        self.user_sessions = {}
+        self.user_sessions_file = os.path.join(matrix_path, ".matrix", "user_sessions.json")
+        self._load_user_sessions()
+
+    def _load_user_sessions(self):
+        """从文件加载 user_sessions 数据"""
+        try:
+            file_path = Path(self.user_sessions_file)
+            if file_path.exists():
+                with open(file_path, 'r', encoding='utf-8') as f:
+                    self.user_sessions = json.load(f)
+                self.logger.info(f"Loaded {len(self.user_sessions)} user sessions from {self.user_sessions_file}")
+            else:
+                # 文件不存在，确保目录存在
+                file_path.parent.mkdir(parents=True, exist_ok=True)
+                self.user_sessions = {}
+                self.logger.info(f"User sessions file not found. Starting with empty sessions.")
+        except json.JSONDecodeError as e:
+            self.logger.error(f"Failed to parse user sessions file: {e}. Starting with empty sessions.")
+            self.user_sessions = {}
+        except Exception as e:
+            self.logger.exception(f"Error loading user sessions: {e}. Starting with empty sessions.")
+            self.user_sessions = {}
+
+    def _save_user_sessions(self):
+        """保存 user_sessions 数据到文件（使用原子写入）"""
+        try:
+            file_path = Path(self.user_sessions_file)
+            # 确保目录存在
+            file_path.parent.mkdir(parents=True, exist_ok=True)
+
+            # 使用原子写入：先写临时文件，再重命名
+            temp_file = file_path.with_suffix('.tmp')
+            with open(temp_file, 'w', encoding='utf-8') as f:
+                json.dump(self.user_sessions, f, ensure_ascii=False, indent=2)
+
+            # 原子重命名
+            temp_file.replace(file_path)
+
+            self.logger.debug(f"Saved {len(self.user_sessions)} user sessions to {self.user_sessions_file}")
+        except Exception as e:
+            self.logger.exception(f"Failed to save user sessions: {e}")
+
+
+
+
+    def register(self, agent):
+        self.directory[agent.name] = agent
+        agent.post_office = self
+
+    def unregister(self, agent):
+        del self.directory[agent.name]
+
+    def yellow_page(self):
+        yellow_page = ""
+        for name, agent in self.directory.items():
+            # 给 description 添加 2 个空格的缩进
+            
+            description = textwrap.indent(agent.description, "  ")
+            
+            # 给 instruction 添加 4 个空格的缩进
+            instruction = textwrap.indent(agent.instruction_to_caller, "    ")
+            
+            yellow_page += f"- {name}: \n"
+            yellow_page += f"{description} \n"
+            
+            yellow_page += f"  [How to talk to {agent.name}]\n"
+            yellow_page += f"{instruction}\n\n"
+        return yellow_page
+
+    def yellow_page_exclude_me(self, myname):
+        yellow_page = ""
+        for name, agent in self.directory.items():
+            # 给 description 添加 2 个空格的缩进
+            if name == myname:
+                continue
+            description = textwrap.indent(agent.description, "  ")
+            
+            # 给 instruction 添加 4 个空格的缩进
+            instruction = textwrap.indent(agent.instruction_to_caller, "    ")
+            
+            yellow_page += f"- {name}: \n"
+            yellow_page += f"{description} \n"
+            
+            yellow_page += f"  [How to talk to {agent.name}]\n"
+            yellow_page += f"{instruction}\n\n"
+        return yellow_page
+
+    def get_contact_list(self, exclude =None):
+        contact_list = []
+        for name, agent in self.directory.items():
+            if name == exclude:
+                continue
+            contact_list.append(name)
+        return contact_list
+
+
+
+
+
+    def pause(self):
+        self._paused = True
+
+    def resume(self):
+        self._paused = False
+
+    async def dispatch(self, email):
+        self.email_db.log_email(email)
+        self.vector_db.add_documents("email", [str(email)],
+            metadatas={"created_at": email.timestamp,
+                       "sender": email.sender,
+                       "recipient": email.recipient,
+                       "user_session_id": email.user_session_id
+                },
+            ids=[email.id]
+        )
+
+        # 维护 user_sessions
+        if email.user_session_id:
+            if email.user_session_id not in self.user_sessions:
+                # 新的 session，添加记录
+                self.user_sessions[email.user_session_id] = {
+                    "name": email.subject,
+                    "last_email_time": str(email.timestamp)
+                }
+                self.logger.info(f"New user session created: {email.user_session_id} - {email.subject}")
+            else:
+                # 已存在的 session，更新时间戳
+                self.user_sessions[email.user_session_id]["last_email_time"] = str(email.timestamp)
+                self.logger.debug(f"User session updated: {email.user_session_id}")
+
+            # 同步到磁盘
+            self._save_user_sessions()
+
+        self.logger.debug(f"Sending email from {email.sender} to {email.recipient} ")
+        await self.queue.put(email)
+        self.logger.debug("Mail delivered")
+
+    async def run(self):
+        self.logger.info("[PostOffice] Service Started")
+        while True:
+            if not self._paused:
+                email = await self.queue.get()
+                if email.recipient in self.directory:
+                    target = self.directory[email.recipient]
+                    await target.inbox.put(email)
+                else:
+                    self.logger.warning(f"Dropped mail to {email.recipient}")
+                self.queue.task_done()
+            else:
+                await asyncio.sleep(0.1)
+
+    def get_mails_by_range(self, user_session_id, agent_name, start=0, end=1):
+        """查询某个Agent的指定Range的邮件
+        Args:
+            agent_name: Agent名称
+            start: 起始索引（0表示最新邮件）
+            end: 结束索引
+        Returns:
+            指定范围内的邮件列表
+        """
+        email_records = self.email_db.get_mails_by_range(user_session_id, agent_name, start, end)
+        emails = []
+        for record in email_records:
+            email = Email(
+                id=record['id'],
+                timestamp=record['timestamp'],
+                sender=record['sender'],
+                recipient=record['recipient'],
+                subject=record['subject'],
+                body=record['body'],
+                in_reply_to=record['in_reply_to'],
+                user_session_id=record.get('user_session_id', None)
+            )
+            emails.append(email)
+        return emails
+
+    def get_user_sessions(self, user_session_id: Optional[str] = None) -> Dict:
+        """
+        获取 user_sessions 数据
+
+        Args:
+            user_session_id: 可选，如果提供则返回指定 session 的数据，否则返回所有 sessions
+
+        Returns:
+            如果提供了 user_session_id：返回该 session 的信息字典，不存在则返回 None
+            如果未提供：返回所有 sessions 的副本（避免外部修改）
+        """
+        if user_session_id:
+            return self.user_sessions.get(user_session_id)
+        else:
+            # 返回深拷贝，避免外部修改影响内部数据
+            return json.loads(json.dumps(self.user_sessions))

agentmatrix/agents/report_writer.py

@@ -0,0 +1,14 @@
+from ..agents.base import BaseAgent
+from ..skills.report_writer import ReportWriterSkillMixin
+import asyncio
+import logging
+
+class ReportWriter(BaseAgent, ReportWriterSkillMixin):
+    _custom_log_level = logging.DEBUG
+    def __init__(self, profile ):
+        super().__init__(profile)
+        self.research_state = None
+        #self.sem = asyncio.Semaphore(5)
+        
+
+

agentmatrix/agents/secretary.py

@@ -0,0 +1,10 @@
+import json
+from ..agents.base import BaseAgent
+
+
+class SecretaryAgent(BaseAgent):
+    def __init__(self, profile ):
+        super().__init__(profile)
+        
+
+

agentmatrix/agents/stateful.py

@@ -0,0 +1,10 @@
+from ..agents.base import BaseAgent
+from ..skills.project_management import ProjectManagementMixin
+from ..skills.notebook import NotebookMixin
+import json
+class StatefulAgent(BaseAgent, ProjectManagementMixin, NotebookMixin):
+    def __init__(self, profile):
+        super().__init__(profile)
+        # 这就是那个“动态文档”，可以是 JSON，也可以是 Markdown
+        self.project_board = None
+        self.vector_db = None

agentmatrix/agents/user_proxy.py

@@ -0,0 +1,82 @@
+from ..agents.base import BaseAgent
+from ..core.message import Email
+from typing import Callable, Awaitable, Optional
+import datetime
+import uuid
+
+# 定义一个回调函数的类型：接收一封邮件，返回 None (异步)
+OnMailReceived = Callable[[Email], Awaitable[None]]
+
+class UserProxyAgent(BaseAgent):
+    def __init__(self, profile) :
+        super().__init__(profile)
+        self.on_mail_received = None
+        
+    
+    def set_mail_handler(self, on_mail_received: OnMailReceived):
+        self.on_mail_received = on_mail_received
+
+
+    async def process_email(self, email: Email):
+        """
+        当 Matrix 里的 Agent (如 Planner) 给 User 发信时触发
+        """
+        print(f"[{self.name}] 收到邮件: {email.subject}")
+        # 1. 记录日志 (BaseAgent 能力)
+        await self.emit("USER_INTERACTION", f"收到来自 {email.sender} 的消息", payload={
+            "type": "MAIL_RECEIVED",
+            "mail_id": email.id,
+            "sender": email.sender,
+            "subject": email.subject,
+            "body": email.body,
+            "in_reply_to": email.in_reply_to
+        })
+
+        # 2. 触发外部钩子 (如果定义了的话)
+        if self.on_mail_received:
+            # 把这封信扔给外部程序，至于外部程序是打印还是弹窗，我不关心
+            await self.on_mail_received(email)
+        else:
+            # 默认行为：简单的打印到控制台，防止没有任何反应
+            print(f"\n[UserProxy 收到邮件] From: {email.sender}\nSubject: {email.subject}\nBody: {email.body}\n")
+
+    async def speak(self, user_session_id, to: str, subject, content: str =None, reply_to_id: str = None):
+        """
+        [Public API]
+        这是给宿主程序调用的方法。
+        相当于用户开口说话。
+        """
+        if content is None:
+            content = subject
+            subject = None
+
+        if not subject:
+            resp = await self.cerebellum.think(f"""
+            请你根据以下邮件内容生成一个高度扼要的邮件主题。明了的说明事情本质就行，不需要具体细节内容，例如“一个问题”，“下载任务” 等等
+
+            [邮件内容]
+            {content}
+
+            [输出要求]
+            你可以尽量简短的说明思路，但输出的最后一行必须是、也只能是邮件主题本身。例如如果邮件主题是ABC，那么最后一行就是ABC。不多不少，不要增加任何标点符号
+            """)
+            subject = resp['reply'] or ""
+            #取subject 最后一行
+            subject = subject.split("\n")[-1]
+        #给subject 后面加上日期yyyy-mm-dd 
+        
+        subject = subject + " " + datetime.datetime.now().strftime("%Y-%m-%d")
+        subject = subject.strip()
+
+
+        email = Email(
+            sender=self.name,
+            recipient=to,
+            subject=subject,
+            body=content,
+            in_reply_to=reply_to_id,
+            user_session_id=user_session_id
+        )
+        await self.post_office.dispatch(email)
+
+

agentmatrix/agents/worker.py

@@ -0,0 +1,30 @@
+from ..agents.base import BaseAgent
+
+class WorkerAgent(BaseAgent):
+    async def step(self, session):
+        await self.emit("THINKING", "正在思考...")
+        
+        # 调用 Mock LLM
+        response = await self.brain.think(self.name, session.history)
+        
+        # 简单的协议解析 (Action Protocol)
+        # 格式: [Action: EMAIL] To: X | Subject: Y | Body: Z
+        # 格式: [Action: REPLY] Body
+        
+        if "[Action: EMAIL]" in response:
+            # 解析 (简化版字符串操作)
+            content = response.split("]", 1)[1].strip()
+            parts = {p.split(":")[0].strip(): p.split(":")[1].strip() for p in content.split("|")}
+            
+            to = parts.get("To")
+            subj = parts.get("Subject")
+            body = parts.get("Body")
+            
+            session.history.append({"role": "assistant", "content": response})
+            # 这里认为是中间步骤，expect_reply = True
+            await self.send_email(to, subj, body, session, expect_reply=True)
+            
+        elif "[Action: REPLY]" in response:
+            body = response.split("REPLY]", 1)[1].strip()
+            # 最终回复
+            await self.send_email(session.original_sender, "Re: Task", body, session, expect_reply=False)

agentmatrix/backends/__init__.py

@@ -0,0 +1 @@
+"""Backends module for Agent-Matrix framework."""