vectorvein 0.1.82__py3-none-any.whl → 0.1.84__py3-none-any.whl

vectorvein/chat_clients/openai_compatible_client.py

@@ -258,7 +258,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
                     else:
                         message = chunk.choices[0].delta.model_dump()
                         delta_content = message.get("content", "")
-                        buffer += delta_content
+                        buffer += delta_content or ""
 
                         while True:
                             if not in_reasoning:
@@ -584,7 +584,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                     else:
                         message = chunk.choices[0].delta.model_dump()
                         delta_content = message.get("content", "")
-                        buffer += delta_content
+                        buffer += delta_content or ""
 
                         while True:
                             if not in_reasoning:

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/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/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/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)

{vectorvein-0.1.82.dist-info → vectorvein-0.1.84.dist-info}/METADATA

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

{vectorvein-0.1.82.dist-info → vectorvein-0.1.84.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.1.82.dist-info/METADATA,sha256=wQvnVyArw5rABZLRm90QH33KMTZKtampPX7uffyBh3E,641
-vectorvein-0.1.82.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.1.82.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.1.84.dist-info/METADATA,sha256=VypGorXFaEc9vawL2I6d4AfNe0Q7GbNRgMI76-nSs-A,641
+vectorvein-0.1.84.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.1.84.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/__init__.py,sha256=omQuG4PRRPNflSAgtdU--rwsWG6vMpwMEyIGZyFVHVQ,18596
 vectorvein/chat_clients/anthropic_client.py,sha256=PGIKldH4FnGrqozoY_FZ6LqhDHC-jY7NF5J1F1zT2Ok,38257
@@ -14,7 +14,7 @@ vectorvein/chat_clients/minimax_client.py,sha256=ooJU92UCACC4TVWKJ-uo8vqQ8qF3K14
 vectorvein/chat_clients/mistral_client.py,sha256=1aKSylzBDaLYcFnaBIL4-sXSzWmXfBeON9Q0rq-ziWw,534
 vectorvein/chat_clients/moonshot_client.py,sha256=gbu-6nGxx8uM_U2WlI4Wus881rFRotzHtMSoYOcruGU,526
 vectorvein/chat_clients/openai_client.py,sha256=Nz6tV45pWcsOupxjnsRsGTicbQNJWIZyxuJoJ5DGMpg,527
-vectorvein/chat_clients/openai_compatible_client.py,sha256=VYT2qNanBaVOZVUIjBpXBUohk5x5QOFNScLwEI3J-GE,28882
+vectorvein/chat_clients/openai_compatible_client.py,sha256=HqdECEcm0JzENjmqclwTXFlnoiguYLfAZ7Z8tqYIsNE,28894
 vectorvein/chat_clients/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
 vectorvein/chat_clients/stepfun_client.py,sha256=zsD2W5ahmR4DD9cqQTXmJr3txrGuvxbRWhFlRdwNijI,519
@@ -37,12 +37,12 @@ vectorvein/workflow/graph/edge.py,sha256=xLZEJmBjAfVB53cd7CuRcKhgE6QqXv9nz32wJn8
 vectorvein/workflow/graph/node.py,sha256=nxDq4pTFcsm2x9MmNmZxghhc-HduSLRwH3Ct2i8bZ3o,2871
 vectorvein/workflow/graph/port.py,sha256=kv0YbAPqe9Hh9QPSoiB2PUTTWQT1xIdNeVD50OkWTzA,5684
 vectorvein/workflow/graph/workflow.py,sha256=BmS351-WSeHQ02Kh0jOQpxJbNgZ-AHwW0mSQLXcKceU,2805
-vectorvein/workflow/nodes/__init__.py,sha256=AB8YCjkK3xdpk1_jfzsBkDtsSp48RcS_N4631MyEJXU,3159
+vectorvein/workflow/nodes/__init__.py,sha256=jd4O27kIJdOtkij1FYZ6aJnJy2OQa7xtL1r-Yv8ylO0,3103
 vectorvein/workflow/nodes/audio_generation.py,sha256=ht2S0vnd0mIAt6FBaSWlADGbb7f_1DAySYrgYnvZT1Q,5726
 vectorvein/workflow/nodes/control_flows.py,sha256=Zc_uWuroYznLrU-BZCncyzvejC-zFl6EuN_VP8oq5mY,6573
 vectorvein/workflow/nodes/file_processing.py,sha256=Rsjc8al0z-2KuweO0nIybWvceqxbqOPQyTs0-pgy5m4,3980
 vectorvein/workflow/nodes/image_generation.py,sha256=fXOhLGodJ3OdKBPXO5a3rq4wN2GMJ0jwqKO_gJFdocU,32852
-vectorvein/workflow/nodes/llms.py,sha256=MU0tmCOcPUcN4uVI9t5RlQeCuKgiXzVNxpdtgMzLV2U,31077
+vectorvein/workflow/nodes/llms.py,sha256=lCq5MVv6k4sEbRfnImQVM1Zzzsyuud-1XuapezIjfQc,34557
 vectorvein/workflow/nodes/media_editing.py,sha256=Od0X0SdcyRhcJckWpDM4WvgWEKxaIsgMXpMifN8Sc5M,29405
 vectorvein/workflow/nodes/media_processing.py,sha256=t-azYDphXmLRdOyHDfXFTS1tsEOyKqKskDyD0y232j8,19043
 vectorvein/workflow/nodes/output.py,sha256=_UQxiddHtGv2rkjhUFE-KDgrjnh0AGJQJyq9-4Aji5A,12567
@@ -50,8 +50,8 @@ vectorvein/workflow/nodes/relational_db.py,sha256=zfzUhV25TpZGhkIzO18PmAT5xhcsJC
 vectorvein/workflow/nodes/text_processing.py,sha256=cfepHoy8JaFDqWLUzl6lmlDlKaH0vk1gWQSm7T0wmn8,7503
 vectorvein/workflow/nodes/tools.py,sha256=Vw7yPdPqENqQ-MwbPcz5LQ-5036UGApCPX6ZspPOw40,12739
 vectorvein/workflow/nodes/triggers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/workflow/nodes/vector_db.py,sha256=FG0fDc99wYLrxqchcbnTKyyFPrVjw710pld--F_j9Oo,5744
+vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs076z-7dlQQ,5712
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
-vectorvein/workflow/utils/json_to_code.py,sha256=8e08bThiXTGXXI0GuBde6lMSp4-CHt7B5S_6rQ3hkg0,6020
-vectorvein-0.1.82.dist-info/RECORD,,
+vectorvein/workflow/utils/json_to_code.py,sha256=TmeibZZQaKBNr0dcWXs8WS6c5AsgRseuMGJ9bOw5fyI,5903
+vectorvein-0.1.84.dist-info/RECORD,,