PyPI - alayaflow - Versions diffs - 0.1.3__tar.gz → 0.1.4__tar.gz - Mend

alayaflow 0.1.3tar.gz → 0.1.4tar.gz

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.

Files changed (82) hide show

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/alayamem_chat/1.0.0/metadata.json ADDED Viewed

@@ -0,0 +1,9 @@
+{
+    "id": "alayamem_chat",
+    "name": "AlayaMem Chat",
+    "description": "一个集成 AlayaMem 记忆功能的对话工作流",
+    "version": "1.0.0",
+    "tags": ["chat", "memory", "alayamem"],
+    "entry_file": "workflow.py",
+    "entry_point": "create_graph"
+}

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/alayamem_chat/1.0.0/schemas.py ADDED Viewed

@@ -0,0 +1,61 @@
+from typing import TypedDict, List, Optional
+from pydantic import BaseModel, Field
+from langchain_core.messages import BaseMessage, AIMessageChunk
+class WorkflowInitArgs(BaseModel):
+    alayamem_url: str = Field(..., description="AlayaMem URL")
+class Input(BaseModel):
+    messages: List[BaseMessage] = Field(..., description="List of input messages")
+class WorkflowContext(BaseModel):
+    user_id: str = Field(..., description="User ID")
+    session_id: str = Field(..., description="Session ID")
+    chat_model_id: str = Field(..., description="Chat Model ID")
+    system_prompt: str = Field(
+        default="你是一个有帮助的AI助手。",
+        description="系统基础提示词"
+    )
+    user_profile_prompt: str = Field(
+        default="## 用户画像\n{user_profile}",
+        description="用户画像提示词模板，使用 {user_profile} 作为占位符"
+    )
+    session_summary_prompt: str = Field(
+        default="## 会话摘要\n{session_summary}",
+        description="会话摘要提示词模板，使用 {session_summary} 作为占位符"
+    )
+    retrieved_docs_prompt: str = Field(
+        default="## 相关参考资料\n{retrieved_docs}",
+        description="检索文档提示词模板，使用 {retrieved_docs} 作为占位符"
+    )
+    history_turns_prompt: str = Field(
+        default="## 历史对话\n{history_turns}",
+        description="历史对话提示词模板，使用 {history_turns} 作为占位符"
+    )
+    context_wrapper_prompt: str = Field(
+        default="# 上下文信息\n\n{context_parts}\n\n请基于以上信息来回答用户的问题。",
+        description="上下文包装提示词模板，使用 {context_parts} 作为占位符"
+    )
+class Output(BaseModel):
+    chat_response: dict = Field(..., description="Chat response")
+class WorkflowState(TypedDict):
+    messages: List[BaseMessage]
+    memory_initialized: bool
+    retrieved_docs: Optional[List[str]]
+    stream_chunks: List[AIMessageChunk]
+    chat_response: Optional[dict]
+    context: Optional[str]
+    # 新增字段
+    history_turns: Optional[List[dict]]
+    session_summary: Optional[str]
+    user_profile: Optional[dict]

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/alayamem_chat/1.0.0/workflow.py ADDED Viewed

@@ -0,0 +1,246 @@
+from langgraph.graph import StateGraph, START, END
+from langgraph.runtime import Runtime
+from alayaflow.component.memory import query_file, commit_turn, turns, summary, profile
+from alayaflow.component.model import ModelManager
+from .schemas import WorkflowInitArgs, WorkflowState, WorkflowContext, Input, Output
+def mk_turns_node(alayamem_url: str):
+    def turns_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        session_id = runtime.context.session_id
+        user_id = runtime.context.user_id
+        original_result = turns(alayamem_url, session_id, user_id)
+        # 打印返回字段
+        print(f"\n[1. turns 接口返回]")
+        print(f"  返回类型: {type(original_result)}")
+        print(f"  返回值: {original_result}")
+        updated_state = state.copy()
+        # turns 接口直接返回列表
+        updated_state["history_turns"] = original_result if isinstance(original_result, list) else []
+        return updated_state
+    return turns_node
+def mk_query_file_node(alayamem_url: str):
+    def query_file_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        user_id = runtime.context.user_id
+        messages = state.get("messages", [])
+        message = messages[-1] if messages else None
+        original_result = query_file(alayamem_url, user_id, message)
+        # 打印返回字段
+        print(f"\n[2. query_file 接口返回]")
+        print(f"  返回类型: {type(original_result)}")
+        if isinstance(original_result, dict):
+            print(f"  字段列表: {list(original_result.keys())}")
+        print(f"  返回值: {original_result}")
+        updated_state = state.copy()
+        # query_file 返回 dict，results 是 dict，包含 documents 字段
+        if isinstance(original_result, dict):
+            results = original_result.get("results", {})
+            updated_state["retrieved_docs"] = results.get("documents", []) if isinstance(results, dict) else []
+        else:
+            updated_state["retrieved_docs"] = []
+        return updated_state
+    return query_file_node
+def mk_summary_node(alayamem_url: str):
+    def summary_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        session_id = runtime.context.session_id
+        user_id = runtime.context.user_id
+        original_result = summary(alayamem_url, session_id, user_id)
+        # 打印返回字段
+        print(f"\n[2. summary 接口返回]")
+        print(f"  返回类型: {type(original_result)}")
+        if isinstance(original_result, dict):
+            print(f"  字段列表: {list(original_result.keys())}")
+        print(f"  返回值: {original_result}")
+        updated_state = state.copy()
+        # summary 可能是 None，需要处理
+        if isinstance(original_result, dict):
+            summary_value = original_result.get("summary", "")
+            updated_state["session_summary"] = summary_value if summary_value is not None else ""
+        else:
+            updated_state["session_summary"] = ""
+        return updated_state
+    return summary_node
+def mk_profile_node(alayamem_url: str):
+    def profile_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        user_id = runtime.context.user_id
+        original_result = profile(alayamem_url, user_id)
+        # 打印返回字段
+        print(f"\n[3. profile 接口返回]")
+        print(f"  返回类型: {type(original_result)}")
+        if isinstance(original_result, dict):
+            print(f"  字段列表: {list(original_result.keys())}")
+        print(f"  返回值: {original_result}")
+        updated_state = state.copy()
+        # profile 可能是 None，需要处理
+        if isinstance(original_result, dict):
+            profile_value = original_result.get("profile", {})
+            updated_state["user_profile"] = profile_value if profile_value is not None else {}
+        else:
+            updated_state["user_profile"] = {}
+        return updated_state
+    return profile_node
+def mk_commit_turn_node(alayamem_url: str):
+    def commit_turn_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        session_id = runtime.context.session_id
+        user_id = runtime.context.user_id
+        messages = state.get("messages", [])
+        chat_response = state.get("chat_response")
+        # 提取用户消息和助手回复
+        user_text = ""
+        if messages:
+            last_msg = messages[-1]
+            if hasattr(last_msg, 'content'):
+                user_text = last_msg.content
+            elif isinstance(last_msg, dict):
+                user_text = last_msg.get('content', '')
+        assistant_text = ""
+        if chat_response:
+            if hasattr(chat_response, 'content'):
+                assistant_text = chat_response.content
+            elif isinstance(chat_response, dict):
+                assistant_text = chat_response.get('content', '')
+        # 调用 commit_turn
+        original_result = commit_turn(alayamem_url, session_id, user_text,
+                                     assistant_text, user_id, window_size=3)
+        # 打印返回字段
+        print(f"\n[4. commit_turn 接口返回]")
+        print(f"  返回类型: {type(original_result)}")
+        if isinstance(original_result, dict):
+            print(f"  字段列表: {list(original_result.keys())}")
+        print(f"  返回值: {original_result}")
+        return state.copy()
+    return commit_turn_node
+def mk_chat_node():
+    model_manager = ModelManager()
+    def chat_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        model_id = runtime.context.chat_model_id
+        chat_model = model_manager.get_model(model_id)
+        if not chat_model:
+            raise ValueError(f"无法找到模型ID为 '{model_id}' 的模型配置")
+        from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
+        # 获取 prompt 模板配置
+        ctx = runtime.context
+        system_prompt = ctx.system_prompt
+        user_profile_prompt = ctx.user_profile_prompt
+        session_summary_prompt = ctx.session_summary_prompt
+        retrieved_docs_prompt = ctx.retrieved_docs_prompt
+        history_turns_prompt = ctx.history_turns_prompt
+        context_wrapper_prompt = ctx.context_wrapper_prompt
+        # 构建完整的消息列表
+        full_messages = []
+        # 添加系统基础提示词
+        if system_prompt:
+            full_messages.append(SystemMessage(content=system_prompt))
+        # 构建上下文信息部分
+        context_parts = []
+        # 添加用户画像
+        # user_profile = state.get("user_profile", {})
+        # if user_profile:
+        #     profile_text = user_profile_prompt.format(user_profile=str(user_profile))
+        #     context_parts.append(profile_text)
+        # 添加会话摘要
+        session_summary = state.get("session_summary", "")
+        if session_summary:
+            summary_text = session_summary_prompt.format(session_summary=session_summary)
+            context_parts.append(summary_text)
+        # 添加检索文档
+        retrieved_docs = state.get("retrieved_docs", [])
+        if retrieved_docs:
+            docs_content = "\n\n".join([str(doc) for doc in retrieved_docs])
+            docs_text = retrieved_docs_prompt.format(retrieved_docs=docs_content)
+            context_parts.append(docs_text)
+        # 添加历史对话
+        history_turns = state.get("history_turns", [])
+        if history_turns:
+            history_content = "\n".join([
+                f"用户: {turn.get('user_text', '')}\n助手: {turn.get('assistant_text', '')}"
+                for turn in history_turns
+            ])
+            history_text = history_turns_prompt.format(history_turns=history_content)
+            context_parts.append(history_text)
+        # 如果有上下文信息，添加上下文系统消息
+        if context_parts:
+            context_content = context_wrapper_prompt.format(
+                context_parts="\n\n".join(context_parts)
+            )
+            full_messages.append(SystemMessage(content=context_content))
+        # 添加当前用户消息
+        current_messages = state["messages"]
+        full_messages.extend(current_messages)
+        # 打印最终输入到模型的消息
+        print(f"\n[5. 模型输入消息]")
+        print(f"  消息数量: {len(full_messages)}")
+        print("-" * 50)
+        full_content = ""
+        for msg in full_messages:
+            msg_type = type(msg).__name__
+            content = msg.content if hasattr(msg, 'content') else str(msg)
+            full_content += f"[{msg_type}]\n{content}\n\n"
+        print(full_content)
+        print("-" * 50)
+        # 调用模型
+        response = chat_model.invoke(full_messages)
+        updated_state = state.copy()
+        updated_state['chat_response'] = response
+        return updated_state
+    return chat_node
+def create_graph(init_args: WorkflowInitArgs | dict):
+    if isinstance(init_args, dict):
+        init_args = WorkflowInitArgs(**init_args)
+    alayamem_url = init_args.alayamem_url
+    graph = StateGraph(WorkflowState, WorkflowContext, input_type=Input, output_type=Output)
+    # 添加节点
+    graph.add_node("turns_node", mk_turns_node(alayamem_url))
+    graph.add_node("query_file_node", mk_query_file_node(alayamem_url))
+    #
+    graph.add_node("summary_node", mk_summary_node(alayamem_url))
+    graph.add_node("profile_node", mk_profile_node(alayamem_url))
+    graph.add_node("chat_node", mk_chat_node())
+    graph.add_node("commit_turn_node", mk_commit_turn_node(alayamem_url))
+    # 添加边：获取历史 -> 获取摘要 -> 获取画像 -> 聊天 -> 提交记录
+    graph.add_edge(START, "turns_node")
+    graph.add_edge("turns_node",  "summary_node")
+    graph.add_edge("summary_node", "profile_node")
+    graph.add_edge("profile_node", "chat_node")
+    graph.add_edge("chat_node", "commit_turn_node")
+    graph.add_edge("commit_turn_node", END)
+    return graph.compile()

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/simple_chat/1.0.0/metadata.py ADDED Viewed

@@ -0,0 +1,16 @@
+from pathlib import Path
+from alayaflow.workflow import WorkflowInfo
+def get_metadata():
+    meta = {
+        "id": "simple_chat",
+        "name": "Simple Chatbot",
+        "description": "一个简单的 LLM 对话工作流示例",
+        "version": "1.0.0",
+        "tags": ["chat", "basic"],
+        "entry_file": "workflow.py",
+        "entry_point": "create_graph",
+        "wf_dir": Path(__file__).parent
+    }
+    return WorkflowInfo(**meta)

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/simple_chat/1.0.0/requirements.txt ADDED Viewed

@@ -0,0 +1,11 @@
+# LangGraph 核心依赖
+langgraph>=0.2.0
+# LangChain Community (用于 ChatOpenAI)
+langchain-community>=0.3.0
+# OpenAI SDK (DeepSeek API 兼容 OpenAI 格式)
+openai>=1.0.0
+# Langfuse
+langfuse>=3.0.0,<4.0.0

alayaflow-0.1.4/.alaya.ai/alayaflow/workflows/simple_chat/1.0.0/workflow.py ADDED Viewed

@@ -0,0 +1,49 @@
+from langgraph.graph import StateGraph, START, END
+from langgraph.runtime import Runtime
+from alayaflow.component.model import ModelManager
+from .schemas import WorkflowInitArgs, WorkflowState, WorkflowContext, Input, Output
+def mk_chat_node():
+    model_manager = ModelManager()
+    def chat_node(state: WorkflowState, runtime: Runtime[WorkflowContext]):
+        model_id = runtime.context.chat_model_id
+        chat_model = model_manager.get_model(model_id)
+        if not chat_model:
+            raise ValueError(f"无法找到模型ID为 '{model_id}' 的模型配置")
+        messages = state["messages"].copy()
+        updated_state = state.copy()
+        retrieved_docs = state.get("retrieved_docs", [])
+        if retrieved_docs:
+            context_text = "\n\n".join([str(doc) for doc in retrieved_docs])
+            from langchain_core.messages import SystemMessage
+            context_message = SystemMessage(
+                content=f"以下是相关的参考资料，请基于这些资料回答用户的问题：\n\n{context_text}"
+            )
+            messages.insert(0, context_message)
+        response = chat_model.invoke(messages)
+        updated_state['chat_response'] = response
+        return updated_state
+    return chat_node
+def create_graph(init_args: WorkflowInitArgs | dict):
+    if isinstance(init_args, dict):
+        init_args = WorkflowInitArgs(**init_args)
+    alayamem_url = init_args.alayamem_url
+    graph = StateGraph(WorkflowState, WorkflowContext, input_type=Input, output_type=Output)
+    graph.add_node("chat_node", mk_chat_node())
+    graph.add_edge(START, "chat_node")
+    graph.add_edge("chat_node",  END)
+    return graph.compile()

{alayaflow-0.1.3 → alayaflow-0.1.4}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alayaflow
-Version: 0.1.3
+Version: 0.1.4
 Summary: A desktop platform for executing LangGraph workflows with uv-managed sandboxes.
 Author-email: alayaflow group <dev@example.com>
 License: MIT

alayaflow-0.1.4/examples/alayamem_demo.py ADDED Viewed

@@ -0,0 +1,142 @@
+import os
+import sys
+from dotenv import load_dotenv
+from alayaflow.api import Flow
+# Load environment variables from .env file
+load_dotenv()
+def init_flow():
+    """Initialize flow api singleton"""
+    # Flow() is a singleton instance
+    # So it can be "constructed" in multiple places
+    # But there is only one instance
+    flow = Flow()
+    flow.init({
+        "alayahub_url": "http://your-alayahub-url",
+    })
+    flow.register_models([
+        {
+            # Local used fields
+            "name": "DeepSeek Chat",
+            "model_id": "deepseek-chat",
+            "provider_name": "DeepSeek",
+            # Connection credentials
+            "model_name": "deepseek-chat",
+            "base_url": "https://api.deepseek.com/v1",
+            "api_key": os.getenv("DEEPSEEK_API_KEY"),
+        }
+    ])
+def init_workflows():
+    flow = Flow()
+    workflow_id = 'alayamem_chat'
+    workflow_version = '1.0.0'
+    init_args = {
+        "alayamem_url": "http://localhost:5555",
+    }
+    print(f"\n正在加载工作流: {workflow_id} v{workflow_version}")
+    try:
+        result = flow.load_workflow(workflow_id, workflow_version, init_args)
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+def run_single_turn(flow, workflow_id, workflow_version, question, context):
+    """执行单轮对话"""
+    inputs = {
+        "messages": [{"role": "user", "content": question}]
+    }
+    def handle_chat_model_stream(event):
+        content = event["data"]["chunk"]['content']
+        if content:
+            print(content, end="", flush=True)
+    def handle_error(event):
+        sys.stderr.write(f"\n[Error Event Detected]\n")
+        sys.stderr.write("--- Error message ---\n")
+        sys.stderr.write(f"{event.get('error', 'Unknown error')}\n")
+        if 'traceback' in event:
+            sys.stderr.write("\n--- Stack Trace ---\n")
+            sys.stderr.write(f"{event['traceback']}\n")
+        # 打印完整的 event 信息以便调试
+        sys.stderr.write("\n--- Complete Event ---\n")
+        sys.stderr.write(f"{event}\n")
+    for event in flow.exec_workflow(workflow_id, workflow_version, inputs, context):
+        if "error" in event:
+            handle_error(event)
+            return False
+        kind = event["event"]
+        if kind == "on_chat_model_stream":
+            handle_chat_model_stream(event)
+    print()
+    return True
+def run_workflow():
+    """运行多轮对话测试"""
+    # 使用已初始化的 Flow 单例
+    flow = Flow()
+    workflow_id = 'alayamem_chat'
+    workflow_version = '1.0.0'
+    context = {
+        "user_id": "kekeke",
+        "session_id": "test_session_multi_turn",
+        "chat_model_id": "deepseek-chat",
+    }
+    # 定义多轮对话问题
+    questions = [
+        "你是谁",
+        "介绍一下transformers",
+        "我问了你几个问题",
+        "你觉得我是做什么工作的"
+    ]
+    print("\n" + "="*80)
+    print("开始多轮对话测试 - 测试 AlayaMem 记忆功能")
+    print("="*80)
+    for i, question in enumerate(questions, 1):
+        print(f"\n{'='*80}")
+        print(f"第 {i} 轮对话")
+        print(f"{'='*80}")
+        print(f"\n用户: {question}")
+        print(f"助手: ", end="")
+        success = run_single_turn(flow, workflow_id, workflow_version, question, context)
+        if not success:
+            print(f"\n第 {i} 轮对话失败，停止测试")
+            break
+        print(f"\n{'-'*80}")
+    print(f"\n{'='*80}")
+    print("多轮对话测试完成")
+    print(f"{'='*80}\n")
+def main():
+    init_flow()
+    init_workflows()
+    # 添加调试信息：验证工作流是否已加载
+    flow = Flow()
+    print(f"已加载的工作流: {flow._workflow_manager._workflows.keys() if hasattr(flow, '_workflow_manager') else 'Unknown'}")
+    run_workflow()
+if __name__ == "__main__":
+    main()

{alayaflow-0.1.3 → alayaflow-0.1.4}/pyproject.toml RENAMED Viewed

@@ -1,6 +1,6 @@
 [project]
 name = "alayaflow"
-version = "0.1.3"
+version = "0.1.4"
 description = "A desktop platform for executing LangGraph workflows with uv-managed sandboxes."
 readme = "README.md"
 requires-python = ">=3.10"

alayaflow-0.1.4/src/alayaflow/__init__.py ADDED Viewed

@@ -0,0 +1,7 @@
+"""AlayaFlow - A workflow engen for management and execution."""
+from importlib.metadata import version
+__version__ = version("alayaflow")

{alayaflow-0.1.3 → alayaflow-0.1.4}/src/alayaflow/api/api_singleton.py RENAMED Viewed

@@ -1,4 +1,4 @@
-from typing import Generator, Dict, List, Self, Optional
+from typing import Generator, Dict, List, Self, Optional, AsyncGenerator
 from alayaflow.utils.singleton import SingletonMeta
 from alayaflow.workflow import WorkflowManager
@@ -6,6 +6,10 @@ from alayaflow.execution import ExecutorManager, ExecutorType
 from alayaflow.common.config import settings
 from alayaflow.component.model import ModelManager, ModelProfile
 from alayaflow.workflow.runnable import BaseRunnableWorkflow
+from alayaflow.utils.logger import AlayaFlowLogger
+logger = AlayaFlowLogger()
 class APISingleton(metaclass=SingletonMeta):
@@ -15,7 +19,7 @@ class APISingleton(metaclass=SingletonMeta):
             workflow_manager=self.workflow_manager
         )
         self.model_manager = ModelManager()
-        self._inited = False
+        self._inited = False
     def is_inited(self) -> bool:
         return self._inited
@@ -34,6 +38,8 @@ class APISingleton(metaclass=SingletonMeta):
         settings.langfuse_public_key = config.get("langfuse_public_key", settings.langfuse_public_key)
         settings.langfuse_secret_key = config.get("langfuse_secret_key", settings.langfuse_secret_key)
         settings.langfuse_url = config.get("langfuse_url", settings.langfuse_url)
+        logger.info(f"AlayaFlow is initialized with config: {settings.model_dump()}")
         self._inited = True
         return self
@@ -97,3 +103,22 @@ class APISingleton(metaclass=SingletonMeta):
             executor_type=executor_type
         ):
             yield event
+    async def exec_workflow_async(
+        self,
+        workflow_id: str,
+        version: str,
+        inputs: dict,
+        context: dict,
+        executor_type: str | ExecutorType = ExecutorType.NAIVE
+    ) -> AsyncGenerator[dict, None]:
+        """异步执行工作流"""
+        self._check_init()
+        async for event in self.executor_manager.exec_workflow_async(
+            workflow_id=workflow_id,
+            version=version,
+            inputs=inputs,
+            context=context,
+            executor_type=executor_type
+        ):
+            yield event

alayaflow 0.1.3__tar.gz → 0.1.4__tar.gz

alayaflow 0.1.3tar.gz → 0.1.4tar.gz