alayaflow 0.1.1__py3-none-any.whl → 0.1.3__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.1"
+__version__ = "0.1.3"
 

alayaflow/api/api_singleton.py

@@ -1,10 +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):
@@ -23,8 +24,10 @@ 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)
@@ -66,10 +69,15 @@ 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,

alayaflow/component/llm_node.py

@@ -18,17 +18,14 @@ class LLMComponent:
     def __init__(
         self,
         *,
-        # ===== 模型 & prompt =====
         model_id: str,
         system_prompt: str,
         prompt: str,
 
-        # ===== 采样参数 =====
         temperature: Optional[float] = None,
         top_p: Optional[float] = None,
         max_tokens: Optional[int] = None,
 
-        # ===== 输出控制 =====
         response_format: ResponseFormat = ResponseFormat.TEXT,
         json_schema: Optional[Dict[str, Any]] = None,
         outputs: Optional[Dict[str, str]] = None,
@@ -47,6 +44,9 @@ class LLMComponent:
         self.json_schema = json_schema
         self.outputs = outputs or {}
         self.retry_json_once = retry_json_once
+        
+        # —— 依赖注入（获取全局单例 ModelManager）——
+        self._model_manager = ModelManager()
     
     def _get_llm(self) -> Runnable:
         bind_kwargs: Dict[str, Any] = {}
@@ -58,8 +58,7 @@ class LLMComponent:
         if self.max_tokens is not None:
             bind_kwargs["max_tokens"] = self.max_tokens
 
-        model_manager = ModelManager()
-        llm = model_manager.get_model(self.model_id, runtime_config=bind_kwargs)
+        llm = self._model_manager.get_model(self.model_id, runtime_config=bind_kwargs)
 
         return llm
     

alayaflow/component/retrieve_node.py

@@ -7,5 +7,71 @@ class RetrieveComponent:
         self.client = client
     
     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 []
+        # result = self.client.vdb_query([query], limit, collection_name)
+        # return result.get('documents', [[]])[0] if result.get('documents') else []
+        docs: list[str] = []
+
+        # -------- 基础个人信息 --------
+        if "姓名" in query:
+            docs.extend([
+                "用户姓名是张三。",
+                "张三，男，1989年出生。",
+            ])
+
+        if "性别" in query:
+            docs.append("张三，男，工程师。")
+
+        if "年龄" in query:
+            docs.append("张三，1989年出生，今年34岁。")
+
+        # -------- 联系方式 --------
+        if "电话" in query:
+            docs.extend([
+                "联系电话：15643431212。",
+                "手机号为15643431212，可随时联系。",
+            ])
+
+        if "邮箱" in query or "Email" in query:
+            docs.extend([
+                "电子邮箱：273230101@qq.com。",
+                "邮箱地址为 27223221@qq.com。",
+            ])
+
+        # -------- 工作单位 / 职务 --------
+        if "工作单位" in query or "公司" in query:
+            docs.extend([
+                "现就职于 Google 中国。",
+                "工作单位为 Google。",
+            ])
+
+        if "职务" in query or "职位" in query:
+            docs.append("现任高级软件工程师。")
+
+        # -------- 教育背景 --------
+        if "学历" in query or "教育" in query:
+            docs.extend([
+                "2016年本科毕业于清华大学计算机科学与技术专业。",
+                "2020年博士毕业于北京大学。",
+            ])
+
+        # -------- 工作 / 学习经历 --------
+        if "学习工作经历" in query or "经历" in query:
+            docs.extend([
+                "2010.09-2014.07 就读于清华大学计算机科学与技术专业。",
+                "2014.07-2016.08 在百度公司担任软件开发工程师。",
+                "2016.09 至今在 Google 担任高级软件工程师。",
+            ])
+
+        # -------- 路径/父字段增强（模拟更真实检索）--------
+        if "个人信息" in query and not docs:
+            docs.append("张三，男，1989年出生，工程师，现居北京。")
+
+        if "联系方式" in query and not docs:
+            docs.append("联系方式：电话 15643431212，邮箱 272040101@qq.com。")
+
+        # -------- 兜底：模拟找不到 --------
+        if not docs:
+            return []
+
+        # limit 截断
+        return docs[:limit]

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/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/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/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.1.dist-info → alayaflow-0.1.3.dist-info}/METADATA

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

{alayaflow-0.1.1.dist-info → alayaflow-0.1.3.dist-info}/RECORD

@@ -1,15 +1,16 @@
-alayaflow/__init__.py,sha256=bySS__ThcDNVwLrPl9W0L9QXcNaJJVIJ-7XLW-kTh6I,121
+alayaflow/__init__.py,sha256=ZQGg3KVDYQcq_LK1TLcFuRtc1KFKYva_AoumPeVlOMM,121
 alayaflow/api/__init__.py,sha256=y6nWgqC3jhOffTqixKlv3OU_NEAFxbzHSvwJrE5bHNs,187
-alayaflow/api/api_singleton.py,sha256=vQA17a29e62ezdPPAuUYnRRD0rAyjVVAALxBUTgYrXI,11159
+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/llm_node.py,sha256=2QfiWew99EVQ05UpA7IkepefpOFHqb4Wxo_tYOipekc,3496
 alayaflow/component/memory.py,sha256=Xl5ABW89dswC9oMwkeS6NFAZniKFw8BuGuLAAHddRRA,1448
-alayaflow/component/retrieve_node.py,sha256=mPzAiAXMCGWFYLnmU7kHuf9ejUhzpE_QSHJwc7LOlAU,451
+alayaflow/component/retrieve_node.py,sha256=FlKyGH767ZbS4ss5SGcprp-sRsY4jA0YdlNP4OFzgQU,2842
+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
@@ -17,7 +18,7 @@ alayaflow/component/model/__init__.py,sha256=v63w90tlcAIEy6cN0wTvnvx3afqwNYSn_fW
 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=fQm8RO4HwEIfbqdwA4Qc76TgrJiBj39NCaJuy7XhaAk,16672
+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
@@ -27,14 +28,14 @@ alayaflow/execution/executors/naive_executor.py,sha256=ICXslitv-9ONvJ3kLC-3-vTTB
 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=9IqNPAtWt7DzASmxg48iTRu-STymvUBd8_7-JsR2pgk,250
+alayaflow/workflow/__init__.py,sha256=mzqmL4P7q7ixTp2b0rZE-5iEBh7vv4YaTyvqnQZKmos,267
 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/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.1.dist-info/METADATA,sha256=w0pT-euiwT0DNwZp5-Yly3hb7dGTTWBNtQ7takFhh7s,1925
-alayaflow-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-alayaflow-0.1.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-alayaflow-0.1.1.dist-info/RECORD,,
+alayaflow-0.1.3.dist-info/METADATA,sha256=aMfRyIwPukXW80Am5h8ApPNiBbL4mRZcfaOtMQflVL4,1925
+alayaflow-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+alayaflow-0.1.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+alayaflow-0.1.3.dist-info/RECORD,,