alayaflow 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

alayaflow/__init__.py

@@ -1,5 +1,5 @@
 """AlayaFlow - A desktop platform for executing LangGraph workflows with multiple executors."""
 
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"
 

alayaflow/api/__init__.py

@@ -1,5 +1,8 @@
 from .api_singleton import APISingleton as Flow
+from alayaflow.component.model import ModelProfile, GenerationConfig
 
 __all__ = [
     "Flow",
+    "ModelProfile",
+    "GenerationConfig",
 ]

alayaflow/api/api_singleton.py

@@ -1,9 +1,11 @@
-from typing import Generator, Dict, List, Self
+from typing import Generator, Dict, List, Self, Optional
 
 from alayaflow.utils.singleton import SingletonMeta
 from alayaflow.workflow import WorkflowManager
 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
 
 
 class APISingleton(metaclass=SingletonMeta):
@@ -12,6 +14,7 @@ class APISingleton(metaclass=SingletonMeta):
         self.executor_manager = ExecutorManager(
             workflow_manager=self.workflow_manager
         )
+        self.model_manager = ModelManager()
         self._inited = False                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                    
 
     def is_inited(self) -> bool:
@@ -21,17 +24,26 @@ class APISingleton(metaclass=SingletonMeta):
         if not self._inited:
             raise ValueError("Flow APISingleton 未初始化，请先调用 init 方法")
 
-    def init(self, config: dict = {}) -> Self:
+    def init(self, config: dict = None) -> Self:
         """初始化 Flow APISingleton"""
+        if config is None:
+            config = {}
         # Overwrite all for now
         settings.alayahub_url = config.get("alayahub_url", settings.alayahub_url)
         settings.langfuse_enabled = config.get("langfuse_enabled", settings.langfuse_enabled)
         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)
+        
         self._inited = True
         return self
 
+    def register_models(self, model_profiles: List[dict | ModelProfile]) -> None:
+        self._check_init()
+        profiles = [ModelProfile(**profile) if isinstance(profile, dict) else profile for profile in model_profiles]
+        for profile in profiles:
+            self.model_manager.register_profile(profile)
+
     def list_loaded_workflow_ids(self) -> List[str]:
         """列举已加载的工作流"""
         self._check_init()
@@ -57,25 +69,31 @@ class APISingleton(metaclass=SingletonMeta):
         self._check_init()
         return self.workflow_manager.uninstall_workflow(workflow_id)
 
-    def load_workflow(self, workflow_id: str, version: str, init_args: dict = {}):
-        """加载工作流"""
+    def load_workflow(self, workflow_id: str, version: str, init_args: Optional[Dict] = None, storage_path: str = settings.workflow_storage_path) -> None:
+        """加载本地工作流"""
         self._check_init()
-        return self.workflow_manager.load_workflow(workflow_id, version, init_args)
+        return self.workflow_manager.load_workflow(workflow_id, version, init_args or {}, storage_path)
+
+    def register_workflow(self, runnable: BaseRunnableWorkflow, requirements: List[str] = None) -> None:
+        """加载Python项目中的工作流"""
+        self._check_init()
+        self.workflow_manager.register_workflow(runnable, requirements or [])
 
     def exec_workflow(
         self,
         workflow_id: str,
         version: str,
-        input_data: dict,
-        user_config: dict,
+        inputs: dict,
+        context: dict,
         executor_type: str | ExecutorType = ExecutorType.NAIVE
     ) -> Generator[dict, None, None]:
         """执行工作流"""
         self._check_init()
-        yield from self.executor_manager.exec_workflow(
+        for event in self.executor_manager.exec_workflow(
             workflow_id=workflow_id,
             version=version,
-            input_data=input_data,
-            user_config=user_config,
+            inputs=inputs,
+            context=context,
             executor_type=executor_type
-        )
+        ):
+            yield event

alayaflow/component/chat_model.py

@@ -1,13 +1,12 @@
-import requests
+import os
 
 from langchain_openai import ChatOpenAI
 
-from alayaflow.common.config import settings
 
 def mk_chat_model_deepseek():
     return ChatOpenAI(
         model="deepseek-chat",
-        api_key="sk-4fe7cd96f5e948c79168025372e2327c",
+        api_key=os.getenv("DEEPSEEK_API_KEY"),
         base_url="https://api.deepseek.com/v1",
     )
 

alayaflow/component/llm_node.py

@@ -2,13 +2,13 @@ from __future__ import annotations
 
 import json
 from enum import IntEnum
-from functools import cached_property
 from typing import Any, Dict, Optional
 
-from langchain_openai import ChatOpenAI
 from langchain_core.messages import SystemMessage, HumanMessage, AIMessage, BaseMessage
 from langchain_core.runnables import Runnable
 
+from alayaflow.component.model import ModelManager
+
 
 class ResponseFormat(IntEnum):
     TEXT = 0
@@ -19,7 +19,7 @@ class LLMComponent:
         self,
         *,
         # ===== 模型 & prompt =====
-        model_name: str,
+        model_id: str,
         system_prompt: str,
         prompt: str,
 
@@ -35,7 +35,7 @@ class LLMComponent:
         retry_json_once: bool = True,
     ):
         # —— 配置即成员变量（= Spec）——
-        self.model_name = model_name
+        self.model_id = model_id
         self.system_prompt = system_prompt
         self.prompt = prompt
 
@@ -47,14 +47,8 @@ class LLMComponent:
         self.json_schema = json_schema
         self.outputs = outputs or {}
         self.retry_json_once = retry_json_once
-
-
-    @cached_property
-    def llm(self) -> Runnable:
-        return ChatOpenAI(model=self.model_name, api_key="sk-4fe7cd96f5e948c79168025372e2327c", base_url="https://api.deepseek.com/v1")
     
     def _get_llm(self) -> Runnable:
-        llm = self.llm
         bind_kwargs: Dict[str, Any] = {}
 
         if self.temperature is not None:
@@ -64,11 +58,8 @@ class LLMComponent:
         if self.max_tokens is not None:
             bind_kwargs["max_tokens"] = self.max_tokens
 
-        if bind_kwargs:
-            try:
-                llm = llm.bind(**bind_kwargs)
-            except Exception:
-                pass
+        model_manager = ModelManager()
+        llm = model_manager.get_model(self.model_id, runtime_config=bind_kwargs)
 
         return llm
     

alayaflow/component/model/__init__.py

@@ -0,0 +1,8 @@
+from .schemas import ModelProfile, GenerationConfig
+from .model_manager import ModelManager
+
+__all__ = [
+    "ModelProfile",
+    "GenerationConfig",
+    "ModelManager",
+]

alayaflow/component/model/model_manager.py

@@ -0,0 +1,60 @@
+from typing import Dict, Optional, Any, Union
+from langchain_core.language_models import BaseChatModel
+from langchain_openai import ChatOpenAI
+
+from alayaflow.utils.singleton import SingletonMeta
+from alayaflow.component.model.schemas import ModelProfile, GenerationConfig
+
+
+class ModelManager(metaclass=SingletonMeta):
+    def __init__(self):
+        self._profiles: Dict[str, ModelProfile] = {}
+
+    def register_profile(self, profile: ModelProfile, override: bool = False) -> None:
+        if profile.model_id in self._profiles and not override:
+            raise ValueError(f"Profile with model ID {profile.model_id} already exists.")
+        self._profiles[profile.model_id] = profile
+
+    def get_model(
+        self, 
+        model_id: str, 
+        runtime_config: Optional[Union[GenerationConfig, Dict[str, Any]]] = None
+    ) -> BaseChatModel:
+        if model_id not in self._profiles:
+            raise ValueError(f"Model ID '{model_id}' not found. Please register it first.")
+
+        profile = self._profiles[model_id]
+        
+        # Merge config
+
+        final_params = profile.default_config.model_dump(exclude={"extra_kwargs"})
+        extra_kwargs = profile.default_config.extra_kwargs.copy()
+
+        if runtime_config:
+            if isinstance(runtime_config, GenerationConfig):
+                override_dict = runtime_config.model_dump(exclude_unset=True, exclude={"extra_kwargs"})
+                final_params.update(override_dict)
+                extra_kwargs.update(runtime_config.extra_kwargs)
+            elif isinstance(runtime_config, dict):
+                final_params.update(runtime_config)
+
+        # Instantiate model
+
+        if profile.model_type == "OpenAI":
+            return ChatOpenAI(
+                model=profile.model_name,
+                openai_api_key=profile.api_key.get_secret_value(),
+                openai_api_base=profile.base_url,
+                # Generation parameters
+                temperature=final_params.get("temperature"),
+                max_tokens=final_params.get("max_tokens"),
+                top_p=final_params.get("top_p"),
+                frequency_penalty=final_params.get("frequency_penalty"),
+                presence_penalty=final_params.get("presence_penalty"),
+                model_kwargs=extra_kwargs
+            )
+        else:
+            raise ValueError(f"Unsupported model type: {profile.model_type}")
+
+    def get_profile(self, model_id: str) -> Optional[ModelProfile]:
+        return self._profiles.get(model_id)

alayaflow/component/model/schemas.py

@@ -0,0 +1,33 @@
+from typing import Optional
+from pydantic import BaseModel, Field, ConfigDict, SecretStr
+
+
+class GenerationConfig(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    temperature: float = Field(default=0.7, ge=0.0, le=2.0, description="采样温度")
+    max_tokens: Optional[int] = Field(default=None, description="最大生成 Token 数")
+    top_p: float = Field(default=1.0, description="核采样阈值")
+    frequency_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    presence_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    
+    extra_kwargs: dict = Field(default_factory=dict, description="其他参数")
+
+
+class ModelProfile(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    # Local used fields
+    name: str = Field(..., description="模型名称")
+    model_id: str = Field(..., description="系统内部使用的唯一 ID")
+    model_type: str = Field(default="OpenAI", description="模型类型工厂标识")
+    provider_name: str = Field(default="Unknown", description="提供商名称")
+
+    # Connection Credentials
+    model_name: str = Field(..., description="厂商的模型名称")
+    base_url: str = Field(..., description="API Base URL")
+    api_key: SecretStr = Field(default=SecretStr(""), description="API Key")
+
+    # Default Generation Config
+    default_config: GenerationConfig = Field(default_factory=GenerationConfig)
+

alayaflow/component/retrieve_node.py

@@ -8,10 +8,4 @@ class RetrieveComponent:
     
     def __call__(self, query: str, collection_name: str, limit: int = 3) -> list[str]:
         result = self.client.vdb_query([query], limit, collection_name)
-        return result.get('documents', [[]])[0] if result.get('documents') else []
-
-if __name__ == "__main__":
-    client = HttpAlayaMemClient("http://10.16.70.46:5555")
-    res = client.vdb_query(messages="姓名", limit=5, collection_name="file_watcher_collection")
-
-    print(res)
+        return result.get('documents', [[]])[0] if result.get('documents') else []

alayaflow/component/search_node.py

@@ -0,0 +1,147 @@
+from __future__ import annotations
+
+import requests
+import urllib.parse
+from typing import List, Dict, Any, Optional, Literal
+
+class WebSearchJinaComponent:
+    def __init__(
+        self,
+        *,
+        api_key: str,
+        
+        query: str,
+        
+        top_k: int = 5,
+        site_limit: Optional[str] = None,  
+        timeout: int = 60,
+        
+        no_cache: bool = False,
+        
+        response_format: Literal['markdown', 'html', 'text'] = 'markdown',
+        
+        retain_images: Literal['none', 'all', 'occluded'] = 'none',
+        
+        with_generated_alt: bool = True,
+        
+        with_links_summary: bool = False,
+    ):
+        self.api_key = api_key
+        self.query = query
+        self.top_k = top_k
+        self.site_limit = site_limit
+        self.timeout = timeout
+        
+        self.no_cache = no_cache
+        self.response_format = response_format
+        self.retain_images = retain_images
+        self.with_generated_alt = with_generated_alt
+        self.with_links_summary = with_links_summary
+        
+        self.base_url = "https://s.jina.ai/"
+
+    def _construct_query(self) -> str:
+        final_query = self.query
+        if self.site_limit:
+            final_query = f"{final_query} site:{self.site_limit}"
+        return final_query
+
+    def _get_headers(self) -> Dict[str, str]:
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Accept": "application/json",
+            
+            "X-Retain-Images": self.retain_images,
+            "X-Return-Format": self.response_format,
+            "X-With-Generated-Alt": "true" if self.with_generated_alt else "false",
+            "X-With-Links-Summary": "true" if self.with_links_summary else "false",
+        }
+        
+        if self.no_cache:
+            headers["X-No-Cache"] = "true"
+            
+        return headers
+
+    def __call__(self) -> List[Dict[str, Any]]:
+        query_str = self._construct_query()
+        encoded_query = urllib.parse.quote(query_str)
+        url = f"{self.base_url}{encoded_query}"
+        
+        headers = self._get_headers()
+
+        try:
+            response = requests.get(url, headers=headers, timeout=self.timeout)
+            
+            if response.status_code != 200:
+                print(f"Error: API returned status {response.status_code}")
+                print(f"Message: {response.text}")
+                return []
+            json_resp = response.json()
+            
+            items = []
+            if isinstance(json_resp, dict):
+                if "data" in json_resp and isinstance(json_resp["data"], list):
+                    items = json_resp["data"]
+                elif "results" in json_resp:
+                    items = json_resp["results"]
+            elif isinstance(json_resp, list):
+                items = json_resp
+            if not items:
+                return []
+            results = items[:self.top_k]
+            clean_results = []
+            for item in results:
+                entry = {
+                    "title": item.get("title", "No Title"),
+                    "url": item.get("url", ""),
+                    "description": item.get("description", ""),
+                    "content": item.get("content", ""), 
+                }
+                if self.with_links_summary and "links" in item:
+                    entry["links"] = item["links"]
+                    
+                clean_results.append(entry)
+                
+            return clean_results
+
+        except requests.exceptions.Timeout:
+            print("Error: Search request timed out.")
+            return []
+        except requests.exceptions.RequestException as e:
+            print(f"Error calling Jina API: {e}")
+            return []
+        except Exception as e:
+            print(f"Unexpected error: {e}")
+            return []
+
+    def to_string_context(self) -> str:
+        results = self.__call__()
+        if not results:
+            return "No search results found."
+
+        parts = []
+        for i, item in enumerate(results, 1):
+            # limit content length to avoid overly long context
+            content_preview = item['content'][:2000] if item['content'] else item['description']
+            
+            parts.append(
+                f"### Result {i}\n"
+                f"**Title**: {item['title']}\n"
+                f"**Source**: {item['url']}\n"
+                f"**Content**:\n{content_preview}\n"
+            )
+        return "\n\n".join(parts)
+    
+if __name__ == "__main__":
+    import os
+    JINA_API_KEY = os.getenv("JINA_API_KEY")
+    search_query = "DeepSeek-V3 的技术架构特点"
+    search_node = WebSearchJinaComponent(
+        api_key=JINA_API_KEY,
+        query=search_query,
+        top_k=3
+    )
+    print(f"正在搜索: {search_query} ...")
+    search_context = search_node.to_string_context()
+    print("搜索完成，部分上下文预览：")
+    print(search_context + "...\n")

alayaflow/component/web_search.py

@@ -111,16 +111,16 @@ def get_config_schema() -> Type[TypedDict]:
     return WFConfig
 
 
-graph = create_graph(search_api_key="xxx")  # 2
-result = graph.invoke({
-    "query": "唐博",     # 3
-}, config={
-    "configurable": WFConfig(
-        search_api_key="abc",
-        search_url="https://xxxx",
-    )
-})
-
-
-print(f"result: {result}")
+if __name__ == "__main__":
+    # Example usage
+    graph = create_graph(search_api_key="your-api-key-here")
+    result = graph.invoke({
+        "query": "search query",
+    }, config={
+        "configurable": WFConfig(
+            search_api_key="your-api-key-here",
+            search_url="https://your-search-api-url",
+        )
+    })
+    print(f"result: {result}")
 

alayaflow/execution/env_manager.py

@@ -50,6 +50,7 @@ class EnvManager:
 
     def _save_metadata(self):
         """保存环境元数据"""
+        os.makedirs(os.path.dirname(self.metadata_file), exist_ok=True)
         with open(self.metadata_file, 'w') as f:
             json.dump(self.env_metadata, f, indent=2)
 

alayaflow/execution/executor_manager.py

@@ -1,4 +1,4 @@
-from typing import Generator, Dict, Optional
+from typing import Dict, Generator, Any
 from enum import Enum
 
 from alayaflow.execution.executors.base_executor import BaseExecutor
@@ -43,10 +43,10 @@ class ExecutorManager:
             self, 
             workflow_id: str,
             version: str, 
-            input_data: dict,
-            user_config: dict, 
+            inputs: dict,
+            context: dict, 
             executor_type: ExecutorType | str = ExecutorType.NAIVE
-    ) -> Generator[Dict, None, None]:
+    ) -> Generator[Dict[str, Any], None, None]:
         if isinstance(executor_type, str):
             executor_type = ExecutorType(executor_type)
         if executor_type not in self._executor_map:
@@ -55,5 +55,5 @@ class ExecutorManager:
                 f"Supported kinds: {list(self._executor_map.keys())}"
             )
         executor = self._executor_map[executor_type]
-        yield from executor.execute_stream(workflow_id, version, input_data, user_config)
+        yield from executor.execute_stream(workflow_id, version, inputs, context)
 

alayaflow/execution/executors/base_executor.py

@@ -4,6 +4,6 @@ from typing import Generator, Dict
 
 class BaseExecutor(ABC):
     @abstractmethod
-    def execute_stream(self, workflow_id: str, version: str, input_data: dict, config: dict) -> Generator[Dict, None, None]:
+    def execute_stream(self, workflow_id: str, version: str, inputs: dict, context: dict) -> Generator[Dict, None, None]:
         pass
 

alayaflow/execution/executors/naive_executor.py

@@ -36,8 +36,8 @@ class NaiveExecutor(BaseExecutor):
             self, 
             workflow_id: str, 
             version: str, 
-            input_data: dict, 
-            user_config: dict
+            inputs: dict, 
+            context: dict
     ) -> Generator[Dict, None, None]:
 
         # 1) resolve workflow
@@ -47,7 +47,7 @@ class NaiveExecutor(BaseExecutor):
             yield {"error": str(e), "workflow_id": workflow_id, "version": version}
             return
 
-        print(f"NaiveExecutor execute_stream: {workflow_id} {version} {input_data} {user_config}")
+        print(f"NaiveExecutor execute_stream: {workflow_id} {version} {inputs} {context}")
 
         # TODO: Support langflow workflow
         # Only support StateGraphRunnableWorkflow now. 
@@ -60,7 +60,7 @@ class NaiveExecutor(BaseExecutor):
 
         def run_async_producer():
             try:
-                asyncio.run(self._produce_events_to_queue(runnable, input_data, user_config, event_queue))
+                asyncio.run(self._produce_events_to_queue(runnable, inputs, context, event_queue))
             except Exception as e:
                 event_queue.put({"error": str(e), "traceback": traceback.format_exc(), "workflow_id": workflow_id, "version": version})
             finally:
@@ -95,20 +95,18 @@ class NaiveExecutor(BaseExecutor):
             yield self._serialize_event(item)
 
     
-    async def _produce_events_to_queue(self, runnable: BaseRunnableWorkflow, input_data: dict, user_config: dict, event_queue: queue.Queue):
+    async def _produce_events_to_queue(self, runnable: BaseRunnableWorkflow, inputs: dict, context: dict, event_queue: queue.Queue):
         try:
             # Setup tracing
             tracing = get_tracing(settings)
             langfuse_cb = tracing.build_callback()
-            
-            # Merge user_config and tracing config
-            merged_config = {
-                "configurable": user_config
-            }
+        
             if langfuse_cb:
-                merged_config.update(tracing.build_config(input_data, runnable.info, langfuse_cb))
+                config = tracing.build_config(inputs, runnable.info, langfuse_cb)
+            else:
+                config = {}
             
-            async for chunk in runnable.astream_events(input_data, merged_config):
+            async for chunk in runnable.stream_events_async(inputs, context, config):
                 event_queue.put(chunk)  # Put each event immediately (real-time)
         except Exception as e:
             # If execution fails, put error event in queue

alayaflow/execution/executors/uv_executor.py

@@ -28,8 +28,8 @@ class UvExecutor(BaseExecutor):
         self,
         workflow_id: str,
         version: str,
-        input_data: dict,
-        user_config: dict,
+        inputs: dict,
+        context: dict,
     ) -> Generator[Dict, None, None]:
         raise NotImplementedError("uv executor not supported yet")
 

alayaflow/execution/executors/worker_executor.py

@@ -7,6 +7,6 @@ class WorkerExecutor(BaseExecutor):
     def __init__(self, workflow_manager: WorkflowManager):
         self.workflow_manager = workflow_manager
     
-    def execute_stream(self, workflow_id: str, version: str, input_data: dict, user_config: dict) -> Generator[Dict, None, None]:
+    def execute_stream(self, workflow_id: str, version: str, inputs: dict, context: dict) -> Generator[Dict, None, None]:
         raise NotImplementedError("worker executor not supported yet")
 

alayaflow/workflow/__init__.py

@@ -2,5 +2,9 @@ from alayaflow.workflow.workflow_info import WorkflowInfo
 from alayaflow.workflow.workflow_loader import WorkflowLoader
 from alayaflow.workflow.workflow_manager import WorkflowManager
 
-__all__ = ["WorkflowInfo", "WorkflowLoader", "WorkflowManager"]
+__all__ = [
+    "WorkflowInfo", 
+    "WorkflowLoader", 
+    "WorkflowManager",
+]
 

alayaflow/workflow/runnable/base_runnable_workflow.py

@@ -1,5 +1,7 @@
 from abc import ABC
-from typing import Dict, Generator
+from typing import Dict, Any, AsyncGenerator
+
+from langgraph.graph.state import RunnableConfig
 
 from alayaflow.workflow.workflow_info import WorkflowInfo
 
@@ -11,9 +13,9 @@ class BaseRunnableWorkflow(ABC):
     def info(self) -> WorkflowInfo:
         return self._info
 
-    def invoke(self, input_data: dict, user_config: dict) -> dict:
-        raise NotImplementedError("invoke method must be implemented in derived classes")
-    
-    def astream_events(self, input_data: dict, user_config: dict) -> Generator[Dict, None, None]:
+    def invoke(self, inputs: dict, context: dict, config: RunnableConfig) -> dict:
         raise NotImplementedError("invoke method must be implemented in derived classes")
 
+    async def stream_events_async(self, inputs: dict, context: dict, config: RunnableConfig) -> AsyncGenerator[Dict[str, Any], None]:
+        raise NotImplementedError("stream_events_async method must be implemented in derived classes")
+

alayaflow/workflow/runnable/state_graph_runnable_workflow.py

@@ -1,7 +1,6 @@
-from functools import cached_property
-from typing import Dict, Generator
+from typing import Dict, Any, AsyncGenerator
 
-from langgraph.graph.state import CompiledStateGraph
+from langgraph.graph.state import CompiledStateGraph, RunnableConfig
 
 from alayaflow.workflow.runnable.base_runnable_workflow import BaseRunnableWorkflow
 from alayaflow.workflow.workflow_info import WorkflowInfo
@@ -11,13 +10,20 @@ class StateGraphRunnableWorkflow(BaseRunnableWorkflow):
         super().__init__(info)
         self._graph = graph
 
-    def invoke(self, input_data: dict, user_config: dict) -> dict:
-        return self._graph.invoke(input_data, {
-            "configurable": user_config
-        })
+    def invoke(self, inputs: dict, context: dict, config: RunnableConfig) -> dict:
+        return self._graph.invoke(inputs, config, context=context)
     
-    def astream_events(self, input_data: dict, user_config: dict) -> Generator[Dict, None, None]:
-        return self._graph.astream_events(input_data, {
-            "configurable": user_config
-        })
+    async def stream_events_async(
+        self, 
+        inputs: dict, 
+        context: dict, 
+        config: RunnableConfig
+    ) -> AsyncGenerator[Dict[str, Any], None]:
+        async for event in self._graph.astream_events(
+            inputs, 
+            config, 
+            version="v2", 
+            context=context
+        ):
+            yield event
 

alayaflow/workflow/workflow_loader.py

@@ -105,8 +105,11 @@ class WorkflowLoader:
 
         return get_input_schema_func, create_graph_func
 
-    def load_workflow(self, init_args: dict = {}, force_reload: bool = False) -> BaseRunnableWorkflow:
+    def load_workflow(self, init_args: dict = None, force_reload: bool = False) -> BaseRunnableWorkflow:
         """加载工作流，返回包含 info 和 creator 的加载结果"""
+        if init_args is None:
+            init_args = {}
+
         info = self.load_info()
 
         # now we only support StateGraph
@@ -116,12 +119,14 @@ class WorkflowLoader:
 
         return runnable
     
-    def _sanitize_module_name(self, wf_id: str, version: str) -> str:
+    @classmethod
+    def _sanitize_module_name(cls, wf_id: str, version: str) -> str:
         safe_version = version.replace(".", "_")
         safe_wf_id = re.sub(r'[^0-9a-zA-Z_]', '_', wf_id)
         return f"workflow.{safe_wf_id}.{safe_version}"
-    
-    def _load_module(self, module_name: str, filepath: Path) -> Any:
+        
+    @classmethod
+    def _load_module(cls, module_name: str, filepath: Path) -> Any:
         """加载 Python 模块，支持相对导入"""
         try:
             # 工作流目录（包含 workflow.py 的目录）
@@ -151,7 +156,8 @@ class WorkflowLoader:
                 del sys.modules[module_name]
             raise ImportError(f"加载工作流模块失败: {e}") from e
     
-    def _resolve_entry_point(self, module: Any, entry_point: str, workflow_id: str) -> Any:
+    @classmethod
+    def _resolve_entry_point(cls, module: Any, entry_point: str, workflow_id: str) -> Any:
         parts = entry_point.split('.')
         obj = module
         
@@ -164,5 +170,3 @@ class WorkflowLoader:
             obj = getattr(obj, part)
         
         return obj
-
-    

alayaflow/workflow/workflow_manager.py

@@ -4,7 +4,7 @@ import shutil
 import tempfile
 
 from pathlib import Path
-from typing import Optional, Dict, List, Tuple, Callable, Any
+from typing import Dict, List, Optional
 import requests
 
 from alayaflow.workflow.workflow_info import WorkflowKey, WorkflowInfo
@@ -24,11 +24,23 @@ class WorkflowManager:
         self._workflow_cache.clear()
         self._requirements_cache.clear()
     
-    def load_workflow(self, workflow_id: str, version: str, init_args: dict = {}) -> None:
-        loader = WorkflowLoader(settings.workflow_storage_path, workflow_id, version)
-        self._info_cache[WorkflowKey(workflow_id, version)] = loader.load_info()
-        self._workflow_cache[WorkflowKey(workflow_id, version)] = loader.load_workflow(init_args)
-        self._requirements_cache[WorkflowKey(workflow_id, version)] = loader.load_requirements()
+    def load_workflow(self, workflow_id: str, version: str, init_args: Optional[Dict] = None, storage_path: str = settings.workflow_storage_path) -> None:
+        if init_args is None:
+            init_args = {}
+        loader = WorkflowLoader(storage_path, workflow_id, version)
+        runnable = loader.load_workflow(init_args)
+        requirements = loader.load_requirements()
+        self._register_workflow(runnable, requirements)
+
+    def register_workflow(self, runnable: BaseRunnableWorkflow, requirements: List[str] = None) -> None:
+        self._register_workflow( runnable, requirements or [])
+
+    def _register_workflow(self, runnable: BaseRunnableWorkflow, requirements: List[str]) -> None:
+        info = runnable.info
+        key = WorkflowKey(info.id, info.version)
+        self._info_cache[key] = info
+        self._workflow_cache[key] = runnable
+        self._requirements_cache[key] = requirements
 
     # Keep for future reference
     # def load_workflows(self) -> None:

{alayaflow-0.1.0.dist-info → alayaflow-0.1.2.dist-info}/METADATA

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

alayaflow-0.1.2.dist-info/RECORD

@@ -0,0 +1,41 @@
+alayaflow/__init__.py,sha256=_EoEy39ORwhkwio9UhypCa7wCJSLmxv6reHH1Aqt5Kw,121
+alayaflow/api/__init__.py,sha256=y6nWgqC3jhOffTqixKlv3OU_NEAFxbzHSvwJrE5bHNs,187
+alayaflow/api/api_singleton.py,sha256=SvWeLHPYHT1ABACwmoNj2dWrFjpqLCpSMz1-KW3hsZQ,11640
+alayaflow/clients/alayamem/base_client.py,sha256=pyU2WF2jqNEgBEe8JOZSg13gHQ2pJcBgJ_6YP-5mWkw,540
+alayaflow/clients/alayamem/http_client.py,sha256=n0hAh_ddzEwFfNMsdw5s2dqvuMsyZNGOT-RcHsIXuEw,2171
+alayaflow/common/config.py,sha256=pi4zH_Pi0u6Fb8ZIs4u3qFOUOUeqxxUqqksoUp-hynM,3806
+alayaflow/component/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alayaflow/component/chat_model.py,sha256=GH2xcOhtvh3YqN2LMITmu1j-0_-t9d4hiMuRFctF2og,402
+alayaflow/component/intent_classifier.py,sha256=5KH52LIqIDpw2hlX4gi3Ff7SFVhenCFdFV-fXag0sDM,3765
+alayaflow/component/llm_node.py,sha256=rvj1f8gqKmdGZkM8Mo8z1PDafdpxhn52nT8L19ZxmaI,3510
+alayaflow/component/memory.py,sha256=Xl5ABW89dswC9oMwkeS6NFAZniKFw8BuGuLAAHddRRA,1448
+alayaflow/component/retrieve_node.py,sha256=mPzAiAXMCGWFYLnmU7kHuf9ejUhzpE_QSHJwc7LOlAU,451
+alayaflow/component/search_node.py,sha256=JNFD6qXDd2_NPpXQf7z8xklJFZs7UrFhWet43nrs25Y,4950
+alayaflow/component/web_search.py,sha256=HZp9j0X0YMBC_mhGqzi0g0pbvmlWiLiNdRxzbEFzl6s,3403
+alayaflow/component/langflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alayaflow/component/langflow/intent_classifier.py,sha256=0xqq2wpVVnEErgjOenKYHMsXMkCrHOlCpEZRJqw3PoM,2822
+alayaflow/component/model/__init__.py,sha256=v63w90tlcAIEy6cN0wTvnvx3afqwNYSn_fWI2U5fSwY,171
+alayaflow/component/model/model_manager.py,sha256=VTqkCLXxhGqSXi7GcBxaRHw3wTTfEk1rP31UMYvPW7Y,2503
+alayaflow/component/model/schemas.py,sha256=YADx6OGltGMzpUSNEYcUbGOrrJvtV6dFdAub3FfAw9w,1404
+alayaflow/execution/__init__.py,sha256=9gj_xgIJxlq3EuwsDMb5X05yP70HoD0_T25Khd495Xc,117
+alayaflow/execution/env_manager.py,sha256=-7npm1f-FNhHyOt8NZGbGA3i7JMHqQXoWV1uAe1bpyE,16744
+alayaflow/execution/executor_manager.py,sha256=_t_tr58yUsMbIW_xfevUXv8ba-7lGbBJFhjPsQ-ricM,2147
+alayaflow/execution/langfuse_tracing.py,sha256=BuRMHDH7Gub7CMkJM5ECLzs4vjy3VqAgzh2INE9zbOI,3882
+alayaflow/execution/workflow_runner.py,sha256=XEX4Em0Hv1sI8Im0lREjXq3fN1jYVwFnMMW3pphIAZk,3243
+alayaflow/execution/executors/__init__.py,sha256=RYwYg880smrZ8EX5iwVsJe0Rtgo8-tF82pY5jA3926g,412
+alayaflow/execution/executors/base_executor.py,sha256=mtBJM9bo_VLWAG9nnuq9xCjlD83bsvk1NPY5-aeD8TQ,254
+alayaflow/execution/executors/naive_executor.py,sha256=ICXslitv-9ONvJ3kLC-3-vTTBg1dWQlp0evCtxMO_MI,4749
+alayaflow/execution/executors/uv_executor.py,sha256=IIwP4j-BuDmfNoizuLsHcW0hRmzsLArWRVtVToX3_dM,4622
+alayaflow/execution/executors/worker_executor.py,sha256=niyTqsxB1iHvkuYb3xd35UwnsQllKul-Z6ikenJZ9Hk,513
+alayaflow/utils/singleton.py,sha256=5crFVfOkr9pU_j83ywqAMaL07BvVN5Ke_VGjT9qyUN0,432
+alayaflow/workflow/__init__.py,sha256=mzqmL4P7q7ixTp2b0rZE-5iEBh7vv4YaTyvqnQZKmos,267
+alayaflow/workflow/workflow_info.py,sha256=rnpAwYE4trhiv7o8LPmQpyQ3CDFfNN2yk1CLKRnWz0w,1259
+alayaflow/workflow/workflow_loader.py,sha256=0EP_SEgcGyvSKgWF3DDL4CEmy5NcxRyCpFV5yXzFviM,6729
+alayaflow/workflow/workflow_manager.py,sha256=dUFS6B5V64mdsopxrM6f3LvJn497eTBqMJaXokypBkI,11980
+alayaflow/workflow/runnable/__init__.py,sha256=sNybFeRxLwbDLHiZxlVFXsn3w2n1Jn0Mtun2W6fvjFU,257
+alayaflow/workflow/runnable/base_runnable_workflow.py,sha256=ap53fOeC5iUh2zm45LpEDjLJ4uqfO2C6FCN6WGm13kw,776
+alayaflow/workflow/runnable/state_graph_runnable_workflow.py,sha256=PMSHks46kmNM2uDVmf5TNcLW7AR6dgfJFohxs8Dcfm4,972
+alayaflow-0.1.2.dist-info/METADATA,sha256=CMWzmY6f6eWQNd6l8vC-PXnfIsSpoyDaNP18vlEGZZo,1925
+alayaflow-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+alayaflow-0.1.2.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+alayaflow-0.1.2.dist-info/RECORD,,

alayaflow-0.1.0.dist-info/RECORD

@@ -1,37 +0,0 @@
-alayaflow/__init__.py,sha256=Twp-G7fdsTkTspyDl9VKx9yieHHF9b8BalhEtnqNKWs,121
-alayaflow/api/__init__.py,sha256=Y33eVzsA9v20VLpRK-v3DZOF-DsjfExag6C1taK2WRc,74
-alayaflow/api/api_singleton.py,sha256=5ffWXL4umEOD3VNBOL0X2c38-PjOEEeQkK5yWbTI1BU,10721
-alayaflow/clients/alayamem/base_client.py,sha256=pyU2WF2jqNEgBEe8JOZSg13gHQ2pJcBgJ_6YP-5mWkw,540
-alayaflow/clients/alayamem/http_client.py,sha256=n0hAh_ddzEwFfNMsdw5s2dqvuMsyZNGOT-RcHsIXuEw,2171
-alayaflow/common/config.py,sha256=pi4zH_Pi0u6Fb8ZIs4u3qFOUOUeqxxUqqksoUp-hynM,3806
-alayaflow/component/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alayaflow/component/chat_model.py,sha256=aXBRDdtp0HOHlonwfXTzOIcjepjd5zBh8znlJpK9oVU,461
-alayaflow/component/intent_classifier.py,sha256=5KH52LIqIDpw2hlX4gi3Ff7SFVhenCFdFV-fXag0sDM,3765
-alayaflow/component/llm_node.py,sha256=URRqSj-snRjbPk-7S7uEd8DJk3L2srYe8Qslu8T6u28,3773
-alayaflow/component/memory.py,sha256=Xl5ABW89dswC9oMwkeS6NFAZniKFw8BuGuLAAHddRRA,1448
-alayaflow/component/retrieve_node.py,sha256=vJqK-5KC-w9ZE_cxA1hHDTWuA-vbPrvxJkionKfniYE,652
-alayaflow/component/web_search.py,sha256=PoUZV_H4vTDSwTqQarbmukQzpyUcBTatJxLZso-0Kok,3282
-alayaflow/component/langflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alayaflow/component/langflow/intent_classifier.py,sha256=0xqq2wpVVnEErgjOenKYHMsXMkCrHOlCpEZRJqw3PoM,2822
-alayaflow/execution/__init__.py,sha256=9gj_xgIJxlq3EuwsDMb5X05yP70HoD0_T25Khd495Xc,117
-alayaflow/execution/env_manager.py,sha256=fQm8RO4HwEIfbqdwA4Qc76TgrJiBj39NCaJuy7XhaAk,16672
-alayaflow/execution/executor_manager.py,sha256=EwSlU6WnsRkhO0z2cxESwtsQnaIOjfmDazVyUob6DVA,2158
-alayaflow/execution/langfuse_tracing.py,sha256=BuRMHDH7Gub7CMkJM5ECLzs4vjy3VqAgzh2INE9zbOI,3882
-alayaflow/execution/workflow_runner.py,sha256=XEX4Em0Hv1sI8Im0lREjXq3fN1jYVwFnMMW3pphIAZk,3243
-alayaflow/execution/executors/__init__.py,sha256=RYwYg880smrZ8EX5iwVsJe0Rtgo8-tF82pY5jA3926g,412
-alayaflow/execution/executors/base_executor.py,sha256=yMP2Fw2uf6TCOj9axQtFApIZCSw94QOXuxvEpzy9UW0,257
-alayaflow/execution/executors/naive_executor.py,sha256=lLqjsHZBbXGUO2HMvr_0hYD9tgUKMWep7v0QBQUF8Lk,4892
-alayaflow/execution/executors/uv_executor.py,sha256=XCqECDdieBlZ36CcMSGVyml11Oi9p4HnJLhyapBDlfQ,4630
-alayaflow/execution/executors/worker_executor.py,sha256=o_O8RjguTifGye4avuBkhKGZcrB_xAAbVuMNve8isfY,521
-alayaflow/utils/singleton.py,sha256=5crFVfOkr9pU_j83ywqAMaL07BvVN5Ke_VGjT9qyUN0,432
-alayaflow/workflow/__init__.py,sha256=9IqNPAtWt7DzASmxg48iTRu-STymvUBd8_7-JsR2pgk,250
-alayaflow/workflow/workflow_info.py,sha256=rnpAwYE4trhiv7o8LPmQpyQ3CDFfNN2yk1CLKRnWz0w,1259
-alayaflow/workflow/workflow_loader.py,sha256=fJi7i714JRY5bESahLxcTNei_f-YEL5fnZtgHG7ChG4,6623
-alayaflow/workflow/workflow_manager.py,sha256=bfPGP1UWom4B2ZfuWyyKI0tfFxyn4j2_wBMECc8-Fu8,11536
-alayaflow/workflow/runnable/__init__.py,sha256=sNybFeRxLwbDLHiZxlVFXsn3w2n1Jn0Mtun2W6fvjFU,257
-alayaflow/workflow/runnable/base_runnable_workflow.py,sha256=gN8d2pUijugu1JZr3RrHS95ziu8Of401IQQtmTM6_lc,655
-alayaflow/workflow/runnable/state_graph_runnable_workflow.py,sha256=K2ahaGN9ubr9G2wDeBTQR_0sYo03SWhQbGVUjIb7w0U,843
-alayaflow-0.1.0.dist-info/METADATA,sha256=ribiSb6KxPEN_sDGa9Yd4v8w21jFzKegHvkD0448zsw,1925
-alayaflow-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-alayaflow-0.1.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-alayaflow-0.1.0.dist-info/RECORD,,