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

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

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

@@ -1,4 +1,4 @@
-alayaflow/__init__.py,sha256=_EoEy39ORwhkwio9UhypCa7wCJSLmxv6reHH1Aqt5Kw,121
+alayaflow/__init__.py,sha256=ZQGg3KVDYQcq_LK1TLcFuRtc1KFKYva_AoumPeVlOMM,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
@@ -7,9 +7,9 @@ alayaflow/common/config.py,sha256=pi4zH_Pi0u6Fb8ZIs4u3qFOUOUeqxxUqqksoUp-hynM,38
 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
@@ -35,7 +35,7 @@ alayaflow/workflow/workflow_manager.py,sha256=dUFS6B5V64mdsopxrM6f3LvJn497eTBqMJ
 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.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,,