osis-python 0.1.0__py3-none-any.whl

osis_python-0.1.0.dist-info/METADATA

@@ -0,0 +1,64 @@
+Metadata-Version: 2.4
+Name: osis-python
+Version: 0.1.0
+Summary: A Python client library for Osis APDL
+Author-email: "CCCC Highway Consultant Co. Ltd." <support@osisbim.com>
+License: MIT License
+        
+        Copyright (c) 2025 CCCC Highway Consultant Co. Ltd.
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+Project-URL: Homepage, https://github.com/osis-ai/pyosis
+Project-URL: Documentation, https://github.com/osis-ai/pyosis#readme
+Project-URL: Repository, https://github.com/osis-ai/pyosis.git
+Project-URL: BugTracker, https://github.com/osis-ai/pyosis/issues
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: langchain>=1.0.3
+Requires-Dist: langchain-core>=1.0.3
+Requires-Dist: langchain-openai>=1.0.2
+Requires-Dist: pydantic>=2.12.4
+Requires-Dist: pydantic_core>=2.41.5
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: black>=23.0; extra == "dev"
+Requires-Dist: mypy>=1.0; extra == "dev"
+Provides-Extra: test
+Requires-Dist: pytest>=7.0; extra == "test"
+Requires-Dist: pytest-cov>=4.0; extra == "test"
+Dynamic: license-file
+
+# pyosis
+
+中交公路规划设计院自研OSIS桥隧分析平台的Python接口
+
+## Installation
+
+```bash
+pip install osis-python
+
+```

osis_python-0.1.0.dist-info/RECORD

@@ -0,0 +1,55 @@
+osis_python-0.1.0.dist-info/licenses/LICENSE,sha256=1KNXhaRdSyN3J-czGUxuAufGyzuSt8i1ucA9exR9ErQ,1108
+pyosis/__init__.py,sha256=KMNJXqbfFe-EoaW0cgDjI_Y6DFcYrc2fyuEMySVuXWY,418
+pyosis/ai/__init__.py,sha256=F8JuPPOZg622KKtVW3SXeOMu12uPAss6xpJR2o1c3v8,52
+pyosis/ai/agents/BaseAgent.py,sha256=C9WYzh6--2Gl7Rpm93kJyKtSwSzSDoCl6qGEbWt-q4w,3727
+pyosis/ai/agents/DecisionAgent.py,sha256=VHe3AQ5Tx86bYGTFcUuHFa7VWvqRhf4Kq3R6p7vNHVU,7924
+pyosis/ai/agents/MaterialAgent.py,sha256=4UTiIcv8yIEJdXYG2TzXuEUTjSub3Z_YHnxztbpR9eg,1727
+pyosis/ai/agents/ModelAgent.py,sha256=WM5W-qRABrPI4yPoFGkijdhg-IFM-h-LuNPSpiGgA5c,1680
+pyosis/ai/agents/QuickBuildingAgent.py,sha256=rpPaymWL7QeTMxPdVc9syc4gdap7OMDl0McqGNokv-U,2845
+pyosis/ai/agents/SectionAgent.py,sha256=RkPsuBQy0EqnSELhDb9oF2b7wMWhN8kTN73fHmOgwt0,1474
+pyosis/ai/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pyosis/boundary/__init__.py,sha256=1ruHAqESmajvyTWcG1sySJAv9H2k8Xl8ZZbSVpwl1cE,71
+pyosis/boundary/interface.py,sha256=0L9acFFR-Re3paLdTrNKImvM_hNb90DNccVE7RnfbP0,3031
+pyosis/control/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/control/interface.py,sha256=c1G4RrOiNGAU_sn7ok54UBeV7-GQfjhhD0Ybhnmc_Ik,3768
+pyosis/core/PyInterface.pyi,sha256=G6RGmYo2k2rCzEGinn3MQLhkZyHrNA7ycwQU9hsx6Fg,6211
+pyosis/core/__init__.py,sha256=6oA_VA9LRvj8Ok3akC_rUoOMDOLsdzoANNIB0LnPkO4,420
+pyosis/core/command.py,sha256=dvyMc0UpG7eFfgsZVlm7PM3XoLMs4BTsPBvNVyRokBY,6730
+pyosis/core/engine.py,sha256=kSg6vCkxoP_560-oEDEUnT9p3cZz6-IO0AdsJqjVhMM,522
+pyosis/element/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/element/interface.py,sha256=8opOnue2OJfTqy9l5EgU70ktyfhym19O2bJhWj7caJU,8860
+pyosis/general/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/general/interface.py,sha256=MKM3DVCOkx8t_ISJCo2qCMnoyJ-aGkzyW3q7xQTH_ok,553
+pyosis/live/__init__.py,sha256=VSn2AhvIcGbsyLIlhktUyLRJABKqTwFc6zZx5O828VA,66
+pyosis/live/analysis.py,sha256=A9ND6yTir3o7JIE3zGP1aJfJeMEDGCjYYkRuUmVB82Q,4468
+pyosis/live/grade.py,sha256=Fv5o2YooL5uO5f1YM7mO5d5kgAQ0IXaJlleNBSe9z7M,4329
+pyosis/live/lane.py,sha256=W8B-6Myyb12BSFwfzjErMqBApuHfreqjNECzXdSFLZo,1678
+pyosis/load/__init__.py,sha256=lf-dhGcE9lOFARlHK6O459YY91mFgyx2k8hQy8tkx2w,69
+pyosis/load/loadcase.py,sha256=I2ZQ54uXOoOh6DJ4oOv8JK-JwcCS76QVRw5mDr3UQMw,1585
+pyosis/load/static.py,sha256=eVD_bmOiKHsjJ2RRFYg2LbxPM6_F1Gi2eRw3rCrGKxg,11264
+pyosis/load/tendon.py,sha256=fvcoY15SL1nSycPiNuhjcunCIuWP4HVvyU_S-zAsEdE,11837
+pyosis/material/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/material/interface.py,sha256=rnh39VE7y-VnrUrndB-vgyUQE-rTnfYDSWs5LQBKB7U,8325
+pyosis/node/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/node/interface.py,sha256=BuwoH7V876gBZIoB_6miG5W7r0id0Tm2Fi4b6iIS9vA,1267
+pyosis/post/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/post/interface.py,sha256=9MbRgCs7O0qH12C3nrw1pCiHx45BP52LbGPEmCYrsJI,860
+pyosis/property/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pyosis/property/coordinate.py,sha256=ASfVBE6fIS_nEzzH3NfF_J9EcMauHcvMe2gBvyLF6Jw,69
+pyosis/property/creep_shrink.py,sha256=S-udAbxeJjNXuNtVEEMZTuuTC7eALIGodNi4gbpWD1E,1513
+pyosis/property/damping.py,sha256=KXcNfLUXrRXsFd45Ai7hTw3l4xbXFswhT5HfuiHZjUs,822
+pyosis/property/pu_curve.py,sha256=d-ETXtU4TxXWVqJmbi-a_oP1ZsqZ_L8HkCXQEBbg1eQ,1587
+pyosis/quick_building/__init__.py,sha256=dyvge-GgUpT3BeDwm_Lss1xqNlxAQdt63WJ7QcuZ_O4,24
+pyosis/quick_building/interface.py,sha256=XEXgPxgNwZzgrc-jDi2FpC5Oqe5IP-5HleqsERcEOwU,13957
+pyosis/section/__init__.py,sha256=JMkVPheKBR9CVjfPtQkEzxzL8EcNKm85PL99pcEXE1s,89
+pyosis/section/common.py,sha256=dmpWREBp1nqYT4KWCzPZrKinG68N1tTy8gpJStCG99I,13382
+pyosis/section/numeric.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pyosis/section/param.py,sha256=uSeL20vQCStokrByES99ASECfzDv17q0fCvPugCulaE,3067
+pyosis/section/steel.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pyosis/stage/__init__.py,sha256=HJMZj4xCKlI7c0rXkLn0HlVP_Mop44FHQdFyEClSrNc,47
+pyosis/stage/define.py,sha256=vQu5h7ptRVNyTQuDbRKxUCi4rCV5CtQGjTLT5PUJOdg,4078
+pyosis/stage/overall.py,sha256=dWpa8Z5rmB5WC8QmrWOdX4YZ7vxzHGHompO_IHOi3Ow,2206
+osis_python-0.1.0.dist-info/METADATA,sha256=9-V0rZU4fgQFeVLuKrvCnrTjWvMSRAb1-4vTjfMknMI,2775
+osis_python-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+osis_python-0.1.0.dist-info/top_level.txt,sha256=6VDNySkQKwipRXcN8ayKVci4vAXi4oPyWLqFhNVcIfM,7
+osis_python-0.1.0.dist-info/RECORD,,

osis_python-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

osis_python-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 CCCC Highway Consultant Co. Ltd.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

osis_python-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+pyosis

pyosis/__init__.py

@@ -0,0 +1,15 @@
+"""
+pyosis
+
+========
+A library for A library for calling OSIS functions.
+A library for extending OSIS functionality.
+A library that allows users to use CAE with great flexibility.
+"""
+
+__version__ = "0.1.0"
+__author__ = "Li Zihao" # 中交公路规划设计院 - 工程技术研究中心 - 生产数智化研究室
+
+# from .core import engine
+# from .ai.agents import BaseAgent
+from .core import osis_run

pyosis/ai/__init__.py

@@ -0,0 +1,6 @@
+"""
+pyosis.ai
+
+========
+OSIS AI相关功能
+"""

pyosis/ai/agents/BaseAgent.py

@@ -0,0 +1,79 @@
+from langchain_openai import ChatOpenAI
+from langchain.agents import create_agent
+from langgraph.checkpoint.memory import InMemorySaver
+import pydantic
+from typing import Literal
+
+class BaseAgent:
+    """Agent base class"""
+    
+    def __init__(self, model="qwen-flash", api_key="", base_url=""):
+        """Initialize the agent"""
+        if model == "" or api_key == "" or base_url == "":
+            raise ValueError("API key and Base URL must be provided for Agent.")
+        self.llm = ChatOpenAI(
+            model=model,
+            api_key=pydantic.SecretStr(api_key),
+            base_url=base_url)
+
+    def create_agent(self, tools, system_prompt):
+        self.system_prompt = system_prompt
+        self.checkpointer = InMemorySaver()          # 记忆管理
+        self.agent = create_agent(model=self.llm, 
+                            tools=tools, 
+                            system_prompt=self.system_prompt, 
+                            checkpointer=self.checkpointer)
+
+    def invoke(self, user_input, thread_id="1"):
+        """向智能体提问"""
+        result = self.agent.invoke(
+            {"messages": [{"role": "user", "content": user_input}]}, 
+            {"configurable": {"thread_id": thread_id}}
+        )
+        return result
+    def stream(self, user_input, thread_id="1", stream_mode: Literal["values", "updates", "checkpoints", "tasks"] = "updates"):
+        """流式输出智能体回复"""
+        return self.agent.stream(
+            {"messages": [{"role": "user", "content": user_input}]}, {"configurable": {"thread_id": thread_id}},
+             stream_mode=stream_mode)
+    
+    def ask_agent(self, user_input, thread_id="1"):
+        """向智能体提问，回答仅文本"""
+        result = self.agent.invoke(
+            {"messages": [{"role": "user", "content": user_input}]}, 
+            {"configurable": {"thread_id": thread_id}}
+        )
+        return result['messages'][-1].content
+    
+    def ask_agent_stream(self, user_input, thread_id="1", stream_mode: Literal["values", "updates", "checkpoints", "tasks"] = "updates"):
+        """流式输出智能体回复，回答仅文本"""
+        for chunk in self.agent.stream(
+            {"messages": [{"role": "user", "content": user_input}]}, {"configurable": {"thread_id": thread_id}}, 
+            stream_mode=stream_mode):
+            # 提取消息内容
+            for step, data in chunk.items():
+                if step == "tools":     # 工具调用结果不需要显示
+                    continue
+                ai_response = data['messages'][-1].content
+                yield ai_response
+    
+    def run_example(self, stream=True):
+        """运行智能体"""
+        while True:
+            user_input = input("User: ")
+            if user_input == "exit" or user_input == "quit":
+                break
+            if stream:
+                print("Agent: ", end="")
+                for chunk in self.stream(user_input):
+                    # 提取消息内容
+                    for step, data in chunk.items():
+                        if 'messages' in data and len(data['messages']) > 0:
+                            ai_response = f"\nstep: {step}\ncontent: {data['messages'][-1].content_blocks}"     # 调试信息
+                            print(ai_response)
+                            if step == "tools":     # 工具调用结果不需要显示
+                                continue
+                            ai_response = data['messages'][-1].content                                          # 一般回复
+                
+            else:
+                print("Agent: ", self.ask_agent(user_input))

pyosis/ai/agents/DecisionAgent.py

@@ -0,0 +1,189 @@
+from langchain_core.messages import AIMessage, HumanMessage
+from .BaseAgent import BaseAgent
+from langchain.tools import tool
+from .MaterialAgent import MaterialAgent
+from .SectionAgent import SectionAgent
+from .ModelAgent import ModelAgent
+from .QuickBuildingAgent import QuickBuildingAgent
+
+
+material_agent = None
+section_agent = None
+model_agent = None
+quick_building_agent = None
+
+def log_to_file(msg,fn = "D:/log.txt"):
+    fp = open("D:/log.txt", "a", encoding="utf-8")
+    fp.write(msg)
+    print(msg)
+    fp.write('\r\n')
+    fp.close()
+
+def call_agent(agent, request: str):
+    ai_response = agent.invoke(request, "4")
+    messages = ai_response['messages']
+    last_human_index = 0
+    for i in range(len(messages)-1, -1, -1):
+        if isinstance(messages[i], HumanMessage):
+            last_human_index = i
+            break
+
+    print("v" * 30)
+    for message in messages[last_human_index + 1:]:
+        if isinstance(message, AIMessage):
+            # print(f"content: {message.content}")
+            # print(f"tool_calls: {message.tool_calls}\n")
+            log_to_file(f"content: {message.content}")
+            log_to_file(f"tool_calls: {message.tool_calls}\n")
+    print("^" * 30)
+
+    return ai_response['messages'][-1].content
+
+@tool
+def call_material_agent(request: str):
+    """
+    调用材料智能体
+
+    Args:
+        request (str): 对材料智能体的请求
+    Returns:
+        str: 材料智能体的回答
+    """
+    global material_agent
+    return call_agent(material_agent, request)
+    # for chunk in material_agent.ask_agent_stream(request, "1"):
+    #     # 提取消息内容
+    #     for step, data in chunk.items():
+    #         ai_response = f"\nstep: {step}\ncontent: {data['messages'][-1].content_blocks}"     # 调试信息
+    #         print(ai_response)
+    #         yield data['messages'][-1].content                                          # 一般回复
+    
+@tool
+def call_section_agent(request: str):
+    '''
+    调用截面智能体
+
+    Args:
+        request (str): 对截面智能体的请求
+    Returns:
+        str: 截面智能体的回答
+    '''
+    global section_agent
+    return call_agent(section_agent, request)
+    # for chunk in section_agent.ask_agent_stream(request, "2"):
+    #     # 提取消息内容
+    #     for step, data in chunk.items():
+    #         ai_response = f"\nstep: {step}\ncontent: {data['messages'][-1].content_blocks}"     # 调试信息
+    #         print(ai_response)
+    #         yield data['messages'][-1].content                                          # 一般回复
+
+@tool
+def call_model_agent(request: str):
+    '''
+    调用模型智能体
+        
+    Args:
+        request (str): 对模型智能体的请求
+    Returns:
+        str: 模型智能体的回答
+    '''
+    global model_agent
+    return call_agent(model_agent, request)
+
+@tool
+def call_quick_building_agent(request: str):
+    '''
+    调用快速建模智能体
+        
+    Args:
+        request (str): 对快速建模智能体的请求
+    Returns:
+        str: 快速建模智能体的回答
+    '''
+    global quick_building_agent
+    return call_agent(quick_building_agent, request)
+
+class DecisionAgent(BaseAgent):
+    """决策智能体"""
+    def __init__(self, model="qwen-max", api_key="", base_url=""):
+        super().__init__(model, api_key, base_url)
+        global material_agent
+        global section_agent
+        global model_agent
+        global quick_building_agent
+        material_agent = MaterialAgent('qwen-flash', api_key, base_url)
+        material_agent.create_agent()
+        section_agent = SectionAgent('qwen-flash', api_key, base_url)
+        section_agent.create_agent()
+        model_agent = ModelAgent('qwen-flash', api_key, base_url)
+        model_agent.create_agent()
+        quick_building_agent = QuickBuildingAgent('qwen-plus', api_key, base_url)
+        quick_building_agent.create_agent()
+
+    def create_agent(self):
+        tools = [
+            call_material_agent, # 调用材料智能体
+            call_section_agent,  # 调用截面智能体
+            call_model_agent,    # 调用模型智能体
+            call_quick_building_agent # 调用快速建模智能体
+        ]
+    
+#         system_prompt = """\
+# 你是一个桥梁设计总监，负责协调材料、截面、模型、显示等专业智能体的工作。
+        
+# 你的职责：
+# 1. 分析用户需求，确定设计参数
+# 2. 协调各个智能体的工作顺序
+# 3. 检查设计进度，确保所有组件都正确创建
+# 4. 处理设计过程中的冲突和错误
+# 5. 调用操作必须一步步来，等上一步完成了才能调用下一个函数。
+
+# 你的手下掌握着以下几个智能体：
+# 1. 材料智能体：用于生成满足用户需求的材料。
+# 2. 截面智能体：用于生成满足用户需求的截面。
+# 3. 模型智能体：用于生成满足用户需求的模型，具体包含创建节点与创建单元。创建时需要提供材料与截面编号。
+# 4. 快速建模智能体：直接创建 小箱梁 T梁 连续小箱梁 连续T梁 空心板 等模型
+
+# 工作流程
+# - 调用工具前告诉用户你的想法。
+# - 你最重要的工作是确定当前用户的需求需要调用哪个智能体，你可能并不知道用户所描述的参数具体含义是什么，你只需要将需求与必要信息整理并传递给具体功能智能体即可。
+# - 若需要创建完整悬浇梁，流程为创建材料-创建截面-创建节点-创建单元-刷新界面。
+
+# 工作原则：
+# 1. **严格顺序执行**：每次只调用一个工具，等待该工具执行完成并返回结果后，再根据结果决定下一步
+# 2. **确认执行结果**：每个工具调用后，必须检查返回的结果状态，确认成功后再继续
+# 3. **分步规划**：即使你知道完整流程，也不能一次性生成所有工具调用，必须一步步来，也一定不要做用户要求之外的事情，比如只要求创建材料，你就不要继续剩下的流程
+
+# 不要做超出用户要求的事。所有流程结束后，请告知用户可以继续下一步。若功能智能体出现问题，请告知失败原因。
+# """
+        system_prompt = \
+"""
+你是一个桥梁设计总监，负责协调材料、截面、模型、显示等专业智能体的工作。
+
+你的手下掌握着以下几个智能体：
+1. 材料智能体：用于生成满足用户需求的材料。
+2. 截面智能体：用于生成满足用户需求的截面。
+3. 模型智能体：用于创建节点与创建单元。创建一定要提供材料与截面编号，如果没有材料或者截面，新建一个。
+4. 快速建模智能体：这是一个独立的模块，直接创建 空心板 小箱梁 T梁 连续小箱梁 连续T梁 等模型。当用户想要直接创建桥梁时，**确保将用户的所有需求（包括桥梁类型、尺寸、荷载、钢束、施工阶段等）一次性全部转交给快速建模智能体，不能分布执行**
+
+你的职责：
+1. 分析用户需求，确定设计参数
+2. 协调各个智能体的工作顺序
+3. 检查设计进度，确保所有组件都正确创建
+4. 处理设计过程中的冲突和错误
+
+工作流程：
+- 先告诉用户你的思考和计划
+- 调用需要的智能体
+- 等待智能体执行完成并返回结果
+- 检查结果，确认是否成功，并详细告知你调用的智能体做了什么
+- 如果成功，继续下一步；如果失败，处理错误
+
+特殊情况处理：
+- 如果工具返回错误，分析错误原因并告诉用户
+- 如果用户需求不完整，先询问缺失信息
+- 所有流程结束后，告知用户可以继续下一步
+
+"""
+        super().create_agent(tools, system_prompt)
+

pyosis/ai/agents/MaterialAgent.py

@@ -0,0 +1,44 @@
+from langchain.tools import tool
+from .BaseAgent import BaseAgent
+from ...property.creep_shrink import osis_creep_shrink
+from ...material import *
+
+class MaterialAgent(BaseAgent):
+    """材料设计智能体"""
+    def __init__(self, model="qwen-flash", api_key="", base_url=""):
+        super().__init__(model, api_key, base_url)
+
+    def create_agent(self):
+        tools = [
+            tool(osis_creep_shrink), 
+            tool(osis_material_conc),
+            tool(osis_material_steel),
+            tool(osis_material_reber),
+            tool(osis_material_prestressed),
+            tool(osis_material_custom),
+            tool(osis_material_del),
+            tool(osis_material_mod)
+        ]
+        system_prompt = \
+"""
+你是材料设计专家，负责桥梁材料的创建和管理。你需要配合决策智能体完成桥梁材料的创建与修改工作。
+
+重要说明：
+- 如果是混凝土材料，一定要先创建收缩徐变特性，再创建材料。其他材料直接创建材料即可
+
+工作流程：
+- 创建任何对象时编号从1递增
+- 操作前告诉用户你的想法，操作完成后先确认执行结果，如果失败结束当前调用链并告知用户
+- 若用户没规定，参数全部使用默认值
+- 若用户要修改某个材料，直接重新调用一次创建函数，参数中指定要修改的材料编号即可
+- 默认创建C50混凝土材料
+
+创建成功后，请告知决策智能体。创建失败后，请告知失败原因。
+"""
+        super().create_agent(tools, system_prompt)
+
+if __name__ == "__main__":
+    material_agent = MaterialAgent()
+    material_agent.create_agent()
+    material_agent.run_example(True)
+    

pyosis/ai/agents/ModelAgent.py

@@ -0,0 +1,42 @@
+from langchain.tools import tool
+from .BaseAgent import BaseAgent
+from ...node import *
+from ...element import *
+
+class ModelAgent(BaseAgent):
+    """模型设计智能体"""
+    def __init__(self, model="qwen-flash", api_key="", base_url=""):
+        super().__init__(model, api_key, base_url)
+
+    def create_agent(self):
+        tools = [
+            tool(osis_node),
+            tool(osis_node_del),
+            tool(osis_node_mod),
+
+            tool(osis_element_beam3d),
+            tool(osis_element_cable),
+            tool(osis_element_truss),
+            tool(osis_element_spring),
+            tool(osis_element_shell),
+            tool(osis_element_del),
+            tool(osis_element_mod)
+        ]
+        system_prompt = \
+"""
+你是桥梁模型设计专家，负责创建桥梁的节点和单元，组成完整结构模型。
+        
+重要依赖关系：
+- 创建模型前必须确保材料和截面已经创建
+- 需要使用已创建的材料编号和截面编号，如果没有提供给你，终止调用并告诉决策智能体
+- 创建单元前一定要确保使用的节点已经创建
+        
+工作流程：
+- 创建任何对象时编号从1递增
+- 用户未规定的参数使用默认参数
+- 若用户要修改对象，直接重新调用一次创建函数，参数中指定要修改的对象编号即可
+- 若用户未说明，假设用户希望创建一个跨度为10米，分为10个单元的简单悬浇梁结构，则你需要创建11个节点和10个单元。
+
+创建成功后，请告知决策智能体。创建失败后，请告知失败原因。
+"""
+        super().create_agent(tools, system_prompt)

pyosis/ai/agents/QuickBuildingAgent.py

@@ -0,0 +1,74 @@
+from langchain.tools import tool
+from .BaseAgent import BaseAgent
+from ...quick_building import *
+
+
+class QuickBuildingAgent(BaseAgent):
+    """模型设计智能体"""
+    def __init__(self, model="qwen-flash", api_key="", base_url=""):
+        super().__init__(model, api_key, base_url)
+
+    def create_agent(self):
+        tools = [
+            tool(osis_set_qb_bridge_type),
+            tool(osis_set_qb_overall),
+            tool(osis_set_qb_portrait),
+            tool(osis_set_qb_load),
+            tool(osis_set_qb_tendon),
+            tool(osis_set_qb_stage),
+            tool(osis_create_qb_bridge)
+        ]
+        system_prompt = \
+"""
+你是快速建模助手，负责配合用户调用几个快速建模函数，可调用以下功能模块：
+
+__桥梁类型设置__ `osis_set_qb_bridge_type(eBridgeType)`
+
+   - 支持类型：HOLLOWSLAB(空心板), SMALLBOXBEAM(小箱梁), TBEAM(T梁), CONTINUOUSSMALLBOXBEAM(连续小箱梁), CONTINUOUSTBEAM(连续T梁)
+
+__总体参数配置__ `osis_set_qb_overall(eBridgeType, 跨径列表, 是否弹性连接, 支座刚度参数...)`
+
+   - 简支桥：单跨参数列表
+   - 连续桥：多跨参数列表
+   - 弹性连接需设置支座刚度
+
+__纵向参数设置__ `osis_set_qb_portrait(eBridgeType, 单元尺寸范围, 结构尺寸参数...)`
+
+   - 控制单元划分和结构几何参数
+
+__荷载配置__ `osis_set_qb_load(eBridgeType, 荷载参数...)`
+
+   - 通过零/非零值控制荷载类型启用
+   - 温度效应需同时设置升温/降温值
+
+__钢束设置__ `osis_set_qb_tendon(eBridgeType, 钢束参数列表)`
+
+   - 每个钢束包含名称、属性、几何参数、应力等详细信息
+
+__施工阶段设置__ `osis_set_qb_stage(eBridgeType, 阶段参数列表)`
+
+   - 定义施工顺序、持续时间和荷载状态
+
+__快速建模函数__ `osis_create_qb_bridge` - 按照设置好的参数创建标准桥型
+
+操作流程：
+
+- 首先通过 osis_set_qb_bridge_type 设定桥梁类型
+
+- 如果用户有特别要求，比如修改某个参数，则只需要调用该参数的对应功能模块来修改，除了要修改的参数外，其他参数不用修改
+
+- 如果用户没有特别要求，直接调用 osis_create_qb_bridge 创建即可，其他函数不用调用，皆被默认设置好了
+
+- 用户想在原桥梁修改某些参数时，只需要再次调用该参数的对应功能模块来修改即可
+
+- 用户想创建新的桥梁时，重新调用 osis_set_qb_bridge_type 来设定桥梁类型，会自动清空旧桥梁数据
+
+
+错误处理：
+
+- 检查函数返回的(成功标志, 错误信息)元组
+- 处理OSISEngine引擎层错误
+- 创建成功后，请告知用户。创建失败后，请告知失败原因。
+
+"""
+        super().create_agent(tools, system_prompt)

pyosis/ai/agents/SectionAgent.py

@@ -0,0 +1,40 @@
+from langchain.tools import tool
+from .BaseAgent import BaseAgent
+from ...section import *
+
+
+class SectionAgent(BaseAgent):
+    """截面设计智能体"""
+    def __init__(self, model="qwen-turbo", api_key="", base_url=""):
+        super().__init__(model, api_key, base_url)
+
+    def create_agent(self):
+        tools = [
+            tool(osis_section_circle),
+            tool(osis_section_Ishape),
+            tool(osis_section_Lshape),
+            tool(osis_section_Tshape),
+            tool(osis_section_smallbox),
+            tool(osis_section_hollowslab),
+            tool(osis_section_custom),
+            
+            tool(osis_section_offset),
+            tool(osis_section_mesh),
+            tool(osis_section_del),
+            tool(osis_section_mod)
+        ]
+        system_prompt = \
+"""
+你是截面设计专家，负责桥梁截面的创建和管理。你需要配合决策智能体完成桥梁截面的创建与修改工作。
+
+注意事项：
+- 创建任何对象时编号从1递增
+- 创建截面时，若用户没有明确要求，必须同时设置截面偏移和网格划分参数
+- 若用户有修改需求，不需要重新调用所有函数，只重新调用参数改变的函数即可
+- 若用户没规定，参数全部使用默认值
+- 默认创建矩形截面
+
+创建成功后，请告知决策智能体。创建失败后，请告知失败原因。
+"""
+        super().create_agent(tools, system_prompt)
+

pyosis/boundary/__init__.py

@@ -0,0 +1,7 @@
+'''
+pyosis.boundary 的 Docstring
+
+'''
+
+
+from .interface import *