vectorvein 0.1.83__tar.gz → 0.1.84__tar.gz

{vectorvein-0.1.83 → vectorvein-0.1.84}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.83
+Version: 0.1.84
 Summary: VectorVein python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.83 → vectorvein-0.1.84}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.83"
+version = "0.1.84"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.83 → vectorvein-0.1.84}/src/vectorvein/workflow/nodes/__init__.py

@@ -47,7 +47,7 @@ from .text_processing import (
     TemplateCompose,
 )
 from .tools import CodebaseAnalysis, TextTranslation, TextSearch, ProgrammingFunction, ImageSearch, WorkflowInvoke
-from .vector_db import VectorDbAddData, VectorDbDeleteData, VectorDbSearchData
+from .vector_db import AddData, DeleteData, Search
 from .video_generation import KlingVideo, CogVideoX
 from .web_crawlers import TextCrawler, BilibiliCrawler, DouyinCrawler, YoutubeCrawler
 
@@ -125,9 +125,9 @@ __all__ = [
     "TextTranslation",
     "TextTruncation",
     "Tts",
-    "VectorDbAddData",
-    "VectorDbDeleteData",
-    "VectorDbSearchData",
+    "AddData",
+    "DeleteData",
+    "Search",
     "VideoEditing",
     "VideoScreenshot",
     "WorkflowInvoke",

{vectorvein-0.1.83 → vectorvein-0.1.84}/src/vectorvein/workflow/nodes/llms.py

@@ -65,6 +65,94 @@ class AliyunQwen(Node):
         )
 
 
+class Baichuan(Node):
+    def __init__(self, id: Optional[str] = None):
+        super().__init__(
+            node_type="Baichuan",
+            category="llms",
+            task_name="llms.baichuan",
+            node_id=id,
+            ports={
+                "prompt": InputPort(
+                    name="prompt",
+                    port_type=PortType.TEXTAREA,
+                    value="",
+                ),
+                "llm_model": InputPort(
+                    name="llm_model",
+                    port_type=PortType.SELECT,
+                    value="Baichuan3-Turbo",
+                    options=[
+                        {"value": "Baichuan4", "label": "Baichuan4"},
+                        {"value": "Baichuan3-Turbo", "label": "Baichuan3-Turbo"},
+                        {"value": "Baichuan3-Turbo-128k", "label": "Baichuan3-Turbo-128k"},
+                        {"value": "Baichuan2-Turbo", "label": "Baichuan2-Turbo"},
+                        {"value": "Baichuan2-53B", "label": "Baichuan2-53B"},
+                    ],
+                ),
+                "temperature": InputPort(
+                    name="temperature",
+                    port_type=PortType.TEMPERATURE,
+                    value=0.7,
+                ),
+                "top_p": InputPort(
+                    name="top_p",
+                    port_type=PortType.NUMBER,
+                    value=0.95,
+                ),
+                "stream": InputPort(
+                    name="stream",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                ),
+                "system_prompt": InputPort(
+                    name="system_prompt",
+                    port_type=PortType.TEXTAREA,
+                    value="",
+                ),
+                "response_format": InputPort(
+                    name="response_format",
+                    port_type=PortType.SELECT,
+                    value="text",
+                    options=[
+                        {"value": "text", "label": "Text"},
+                        {"value": "json_object", "label": "JSON"},
+                    ],
+                ),
+                "use_function_call": InputPort(
+                    name="use_function_call",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                ),
+                "functions": InputPort(
+                    name="functions",
+                    port_type=PortType.SELECT,
+                    value=[],
+                ),
+                "function_call_mode": InputPort(
+                    name="function_call_mode",
+                    port_type=PortType.SELECT,
+                    value="auto",
+                    options=[
+                        {"value": "auto", "label": "auto"},
+                        {"value": "none", "label": "none"},
+                    ],
+                ),
+                "output": OutputPort(
+                    name="output",
+                ),
+                "function_call_output": OutputPort(
+                    name="function_call_output",
+                    condition="return fieldsData.use_function_call.value",
+                ),
+                "function_call_arguments": OutputPort(
+                    name="function_call_arguments",
+                    condition="return fieldsData.use_function_call.value",
+                ),
+            },
+        )
+
+
 class BaiduWenxin(Node):
     def __init__(self, id: Optional[str] = None):
         super().__init__(

{vectorvein-0.1.83 → vectorvein-0.1.84}/src/vectorvein/workflow/nodes/vector_db.py

@@ -4,7 +4,7 @@ from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
-class VectorDbAddData(Node):
+class AddData(Node):
     def __init__(self, id: Optional[str] = None):
         super().__init__(
             node_type="AddData",
@@ -86,7 +86,7 @@ class VectorDbAddData(Node):
         )
 
 
-class VectorDbDeleteData(Node):
+class DeleteData(Node):
     def __init__(self, id: Optional[str] = None):
         super().__init__(
             node_type="DeleteData",
@@ -110,10 +110,10 @@ class VectorDbDeleteData(Node):
         )
 
 
-class VectorDbSearchData(Node):
+class Search(Node):
     def __init__(self, id: Optional[str] = None):
         super().__init__(
-            node_type="SearchData",
+            node_type="Search",
             category="vector_db",
             task_name="vector_db.search_data",
             node_id=id,

{vectorvein-0.1.83 → vectorvein-0.1.84}/src/vectorvein/workflow/utils/json_to_code.py

@@ -158,11 +158,11 @@ def generate_python_code(
                 f"port_type={to_python_str(port['field_type'])}",
                 f"value={to_python_str(port['value'])}",
             ]
-            if port["show"]:
+            if port.get("show"):
                 params.append(f"show={port['show']}")
-            if port["options"]:
+            if port.get("options"):
                 params.append(f"options={port['options']}")
-            if port["is_output"]:
+            if port.get("is_output"):
                 params.append(f"is_output={bool(port['is_output'])}")
             code.append(f"{info['var_name']}.add_port({', '.join(params)})")
 
@@ -183,9 +183,3 @@ def generate_python_code(
         code.append(f'workflow.connect({source_var}, "{source_handle}", {target_var}, "{target_handle}")')
 
     return "\n".join(code)
-
-
-if __name__ == "__main__":
-    # 使用示例
-    code = generate_python_code(json_file="workflow.json")
-    print(code)