alayaflow 0.1.1__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.1"
+__version__ = "0.1.2"
 

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/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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alayaflow
-Version: 0.1.1
+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.1.dist-info → alayaflow-0.1.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
-alayaflow/__init__.py,sha256=bySS__ThcDNVwLrPl9W0L9QXcNaJJVIJ-7XLW-kTh6I,121
+alayaflow/__init__.py,sha256=_EoEy39ORwhkwio9UhypCa7wCJSLmxv6reHH1Aqt5Kw,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
@@ -10,6 +10,7 @@ alayaflow/component/intent_classifier.py,sha256=5KH52LIqIDpw2hlX4gi3Ff7SFVhenCFd
 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
@@ -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.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,,