基本搭建完成多智能体框架

_backend/tools.py

@@ -1,7 +1,7 @@
 import requests
 
 
-def retrieval_from_knowledge_base(
+def vector_retrieval_from_knowledge_base(
         query: str,
         topk: int = 3
         ) -> str:
@@ -16,10 +16,77 @@ def retrieval_from_knowledge_base(
     str: The result of the knowledge retrieval in JSON format.
     """
     url = 'http://100.85.52.31:7080/v1/chat-messages'
+    # retrieval_from_knowledge_base
     headers = {
         'Authorization': f'Bearer app-uJgo3TQKcS1O9PMCDHko71Fp',
         'Content-Type': 'application/json'
     }
+
+    data = {
+        "inputs": {"topK": topk},
+        "query": query,
+        "response_mode": "blocking",
+        "user": "tangger",
+        "files": []
+    }
+
+    response = requests.post(url, headers=headers, json=data)
+
+    if response.status_code == 524:
+        print("Server is not responding. Please try again later. Maybe GPU was down in the container.")
+        return None
+
+    try:
+        result = response.json()
+    except ValueError:
+        return [{"error": "Response is not in JSON format"}]
+
+    useful_results = []
+    try:
+        answer = eval(result.get("answer", "[]"))
+        for item in answer:
+            metadata = item.get("metadata", {})
+            useful_info = {
+                "id": metadata.get("document_id"),
+                "title": item.get("title"),
+                "content": item.get("content"),
+                "metadata": None,
+                "embedding": None,
+                "score": metadata.get("score")
+            }
+            useful_results.append(useful_info)
+    except Exception as e:
+        return [{"error": f"Error processing result: {e}", "status": "TERMINATE"}]
+    if useful_results == []:
+        useful_results = "NULL"
+    return str(useful_results)
+
+
+def hybird_retrieval_from_knowledge_base(
+        query: str,
+        topk: int = 3
+        ) -> str:
+    """
+    Retrieval for knowledge from the knowledge base based on the specified query and returns the topk results.
+    
+    Parameters:
+    query (str): The query for knowledge retrieval.
+    topk (int): The number of top results to return, default is 3.
+    
+    Returns:
+    str: The result of the knowledge retrieval in JSON format.
+    """
+    url = 'http://100.85.52.31:7080/v1/chat-messages'
+    # retrieval_from_knowledge_base
+    # headers = {
+    #     'Authorization': f'Bearer app-uJgo3TQKcS1O9PMCDHko71Fp',
+    #     'Content-Type': 'application/json'
+    # }
+    # retrieval_from_knowledge_base_new
+    headers = {
+        'Authorization': f'Bearer app-PP7g4pIQORsXwha4SBvSEqBs',
+        'Content-Type': 'application/json'
+    }
     data = {
         "inputs": {"topK": topk},
         "query": query,
@@ -91,47 +158,47 @@ def scheme_convert_to_json():
         转换合成方案时，必须严格遵守如下预定义的JSON格式，每个JSON结构的字段必须填充完整，即使有些字段留空：
         ```json
         {
-            "task_id": "",
-            "experiment_name": "",
-            "materials": [
-                { "material_id": "", "name": "", "amount": "", "unit": "", "purity": "", "state": ""}
+            "TaskId": "",
+            "ExperimentName": "",
+            "Materials": [
+                { "MaterialId": "", "Name": "", "Formula": "", "Amount": "", "Unit": "", "Purity": "", "State": ""}
                 // ...可在此处添加更多material对象
             ],
 
-            "containers": [{
-                "container_id": "", "name": "", "capacity": "", "unit": "", "material_of_construction": "", "shape": "", "heat_resistant": "", "pressure_rating": "",
+            "Containers": [{
+                "ContainerId": "", "Name": "", "Capacity": "", "Unit": "", "MaterialOfConstruction": "", "Shape": "", "HeatResistant": "", "PressureRating": "",
                 // ...可在此处添加更多container对象
             ],
 
-            "equipments": [{
-                "equipment_id": "", "name": "", 
-                "parameters": {
+            "Equipments": [{
+                "EquipmentId": "", "Name": "", 
+                "Parameters": {
                     // 具体设备参数（例如 rpm 范围, 温度范围等）
                 },
                 }
                 // ...可在此处添加更多equipment对象
             ],
 
-            "robot_workflow": [{ 
-                "step_id": "", "description": "",
-                "actions": [{
-                    "action_type": "",  // limited robot action: "pick_container""place_container""pick_container_with_material""place_container_into_equipment""remove_container_from_equipment"
-                    "container_id": "",
-                    "material_id": "",
-                    "equipment_id": "",
+            "RobotWorkflow": [{ 
+                "Stepid": "", "Description": "",
+                "Actions": [{
+                    "ActionType": "",  // limited robot action: "pick_container""place_container""pick_container_with_material""place_container_into_equipment""remove_container_from_equipment"
+                    "ContainerId": "",
+                    "MaterialId": "",
+                    "EquipmentId": "",
                     }
                     // ...可在此处添加更多子动作
                 ],
-                "dependencies": [
+                "Dependencies": [
                     // 若需要依赖之前的若干 step_id，可列在这里，如 ["1", "2"]
                 ],
-                "step_output": {
-                    "container_id": "",
-                    "contents": [
+                "StepOutput": {
+                    "ContainerId": "",
+                    "Contents": [
                     {
-                        "material_id": "",
-                        "amount": "",
-                        "unit": ""
+                        "MaterialId": "",
+                        "Amount": "",
+                        "Unit": ""
                     }
                     // ...可在此处列出执行完本步骤后容器中的产物或状态
                     ]
@@ -143,16 +210,16 @@ def scheme_convert_to_json():
         ```
 
         ### JSON结构主要字段说明
-        1. task_id 类型: 字符串 说明: 任务的唯一标识符，用于区分不同的任务 限制: 必须唯一，不能重复
-        2. materials 类型: 数组 说明: 使用的材料列表，每个材料包含ID、名称、数量和单位
-        3. containers 类型: 数组 说明: 使用的容器列表，每个容器包含ID、类型、容量、单位和可选的附加参数
-            限制: a.数组中的每个对象必须包含以下字段：a.1 "type": 容器类型（如烧杯、锥形瓶、离心管等） a.2 "capacity": 容器的容量 a.3 "unit": 容量的单位 a.4 "additional_parameters" 为可选字段，可包含材质、耐温范围等信息
-        4. equipments 类型: 数组 说明: 使用的设备列表，每个设备包含ID、名称和可选的参数 限制: 数组中的每个对象必须包含 "name" 字段 "parameters" 为可选，可根据设备实际需求进行配置（如搅拌速度、超声功率、温度范围等）
-        5. workflow 类型: 数组 说明: 包含所有步骤的列表 限制:每个步骤都是一个对象 顺序重要（一般按步骤顺序依次执行）
-        6. step_id 类型: 整数 说明: 步骤的唯一标识符，用于区分不同的步骤 限制: 必须唯一，不能重复
-        7. actions 类型: 数组 说明: 包含该步骤中所有动作的列表 限制: a.每个动作都是一个对象 b.动作在数组中的顺序通常会影响执行顺序
-        8. action_id 类型: 字符串 说明: 动作的唯一标识符，用于区分同一步骤内的不同动作 限制: 在同一步骤内必须唯一
-        9. action_type 类型: 字符串 说明: 动作的类型，但此处特别强调仅限于机械臂可执行的动作 限制: 
+        1. TaskId 类型: 字符串 说明: 任务的唯一标识符，用于区分不同的任务 限制: 必须唯一，不能重复
+        2. Materials 类型: 数组 说明: 使用的材料列表，每个材料包含ID、名称、数量和单位
+        3. Containers 类型: 数组 说明: 使用的容器列表，每个容器包含ID、类型、容量、单位和可选的附加参数
+            限制: a.数组中的每个对象必须包含以下字段：a.1 "Type": 容器类型（如烧杯、锥形瓶、离心管等） a.2 "Capacity": 容器的容量 a.3 "Unit": 容量的单位 
+        4. Equipments 类型: 数组 说明: 使用的设备列表，每个设备包含ID、名称和可选的参数 限制: 数组中的每个对象必须包含 "Name" 字段 "Parameters" 为可选，可根据设备实际需求进行配置（如搅拌速度、超声功率、温度范围等）
+        5. Workflow 类型: 数组 说明: 包含所有步骤的列表 限制:每个步骤都是一个对象 顺序重要（一般按步骤顺序依次执行）
+        6. StepId 类型: 整数 说明: 步骤的唯一标识符，用于区分不同的步骤 限制: 必须唯一，不能重复
+        7. Actions 类型: 数组 说明: 包含该步骤中所有动作的列表 限制: a.每个动作都是一个对象 b.动作在数组中的顺序通常会影响执行顺序
+        8. ActionId 类型: 字符串 说明: 动作的唯一标识符，用于区分同一步骤内的不同动作 限制: 在同一步骤内必须唯一
+        9. ActionType 类型: 字符串 说明: 动作的类型，但此处特别强调仅限于机械臂可执行的动作 限制: 
             a.必须是以下预定义类型之一（对应机械臂操作）：
                 a.1 "pick_container" （拿容器）
                 a.2 "place_container" （放容器）
@@ -160,8 +227,8 @@ def scheme_convert_to_json():
                 a.4 "place_container_into_equipment" （将容器放进某设备）
                 a.5 "remove_container_from_equipment" （从设备中取出容器）
             b. 诸如“搅拌”、“超声”、“离心”等动作不在此列，它们属于设备自身的潜在动作，不在机械臂的动作范围内
-        10. dependencies 类型: 数组 说明: 依赖的前一步骤的 step_id 列表 限制: 每个依赖项必须是有效的 step_id 当本步骤需要等待前面若干步骤完成后再执行时，可通过此字段进行控制
-        11. step_output 类型: 字符串 说明: 步骤的输出标识符，用于后续步骤的输入或引用 限制: 标识符应唯一且有意义 可用来表示该步骤总体产出或容器中的新溶液名等
+        10. Dependencies 类型: 数组 说明: 依赖的前一步骤的 step_id 列表 限制: 每个依赖项必须是有效的 step_id 当本步骤需要等待前面若干步骤完成后再执行时，可通过此字段进行控制
+        11. StepOutput 类型: 字符串 说明: 步骤的输出标识符，用于后续步骤的输入或引用 限制: 标识符应唯一且有意义 可用来表示该步骤总体产出或容器中的新溶液名等
 """
 
 # def send_instruction_to_robot_platform():
@@ -288,7 +355,7 @@ def upload_to_s3(json_data: str):
         # 尝试解析清理后的JSON数据
         data = json.loads(json_data_cleaned)
         # 取得task id
-        task_id = data['task_id']
+        task_id = data['TaskId']
         # print("解析后的JSON数据:", data)
         with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
             try:
@@ -345,7 +412,7 @@ def get_latest_exp_log():
         # print(res)
         return res
 
-    pl_latest = get_pl_latest_file() 
+    pl_latest = get_pl_latest_file()
     uv_latest = get_uv_latest_file()
 
     return pl_latest + "\n" + uv_latest
@@ -364,23 +431,7 @@ def generate_task_id():
     return task_id
 
 
-def sendScheme2RobotPlatform(task_id: str, scheme_url: str):
-    # 首先检查task_id是否和scheme_url是否匹配
-    if task_id not in scheme_url:
-        return {"status": "error", "message": "task_id and scheme_url do not match, Request human/user intervention."}
-    
-    # 读取scheme_url的内容
-    import requests
-    try:
-        response = requests.get(scheme_url)
-        response.raise_for_status()
-        scheme_content = response.text
-        # 读取scheme_content的内容为JSON
-        import json
-        scheme_data = json.loads(scheme_content)
-        # print(scheme_data)
-    except requests.exceptions.RequestException as e:
-        return {"status": "error", "message": f"Error reading scheme_url: {e}"}
+def sendScheme2RobotWorkstation(task_id: str, scheme_url: str):
     
     def mol2mg(formula: str, source_unit: str, target_unit: str, value: float):
         """
@@ -418,19 +469,126 @@ def sendScheme2RobotPlatform(task_id: str, scheme_url: str):
         
         except ValueError as e:
             return {"status": "error", "message": f"Invalid chemical formula: {formula}. Error: {str(e)}"}
-        
-
 
+    # 首先检查task_id是否和scheme_url是否匹配
+    if task_id not in scheme_url:
+        return {"status": "error", "message": "task_id and scheme_url do not match, Request human/user intervention."}
+    
+    # 读取scheme_url的内容
     import requests
-    url = "http://100.122.132.69:50000/sendScheme2RobotPlatform"
+    try:
+        response = requests.get(scheme_url)
+        response.raise_for_status()
+        scheme_content = response.text
+        # 读取scheme_content的内容为JSON
+        import json
+        scheme_data = json.loads(scheme_content)
+        robot_scheme = {}
+        robot_scheme['TaskId'] = scheme_data['TaskId']
+        robot_scheme['ExperimentName'] = scheme_data['ExperimentName']
+        materials = []
+        for mat in scheme_data['Materials']:
+            materials.append({
+                "MaterialId": mat['MaterialId'],
+                "Name": mat['Name'],
+                "Amount": mat['Amount'] if "mol" not in mat['Unit'] else mol2mg(mat['Formula'], mat['Unit'], 'mg', float(mat['Amount']))["value"],
+                "Unit": mat['Unit'] if "mol" not in mat['Unit'] else "mg",
+                "Purity": mat['Purity'],
+                "State": mat['State']
+            })
+        robot_scheme['Materials'] = materials
+        robot_scheme['Containers'] = scheme_data['Containers']
+        robot_scheme['Equipments'] = scheme_data['Equipments']
+        robot_scheme['RobotWorkflow'] = scheme_data['RobotWorkflow']
+        # print(scheme_data)
+    except requests.exceptions.RequestException as e:
+        return {"status": "error", "message": f"Error reading scheme_url: {e}"}
+    
+    import requests
+    # url = "http://100.122.132.69:50000/sendScheme2RobotPlatform"
+    url = "http://localhost:50000/sendScheme2RobotWorkstation"
     data = {"status": "ok"}
     try:
-        response = requests.post(url, json=data)
+        response = requests.post(url, json=robot_scheme)
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
-        print(f"Error sending scheme to robot platform: {e}")
+        print(f"Error sending scheme to robot workstation: {e}")
+        return None
+
+def sendScheme2MobileRobot(task_id: str, scheme_url: str):
+    
+    def mol2mg(formula: str, source_unit: str, target_unit: str, value: float):
+        """
+        将mol转换为mg
+        Args:
+            formula: 化学式，如CsPb
+            source_unit: 源单位 (mol或mmol)
+            target_unit: 目标单位 (mg)
+            value: 要转换的值
+        """
+        import requests
+        from periodictable import formula as chem_formula
+        
+        # 检查单位是否有效
+        if source_unit.lower() not in ['mol', 'mmol'] or target_unit.lower() != 'mg':
+            return {"status": "error", "message": "Invalid units. Only mol/mmol to mg conversion supported"}
+            
+        try:
+            # 解析化学式并计算摩尔质量
+            compound = chem_formula(formula)
+            molar_mass = compound.mass  # 获取化合物摩尔质量 (g/mol)
+            
+            # 转换计算
+            if source_unit.lower() == 'mol':
+                mg_value = value * molar_mass * 1000  # mol -> g -> mg
+            elif source_unit.lower() == 'mmol':
+                mg_value = value * molar_mass  # mmol -> g -> mg
+                
+            return {
+                "status": "success",
+                "formula": formula,
+                "value": round(mg_value, 4),  # 保留4位小数
+                "unit": "mg"
+            }
+        
+        except ValueError as e:
+            return {"status": "error", "message": f"Invalid chemical formula: {formula}. Error: {str(e)}"}
+
+    # 首先检查task_id是否和scheme_url是否匹配
+    if task_id not in scheme_url:
+        return {"status": "error", "message": "task_id and scheme_url do not match, Request human/user intervention."}
+    
+    # 读取scheme_url的内容
+    import requests
+    try:
+        response = requests.get(scheme_url)
+        response.raise_for_status()
+        scheme_content = response.text
+        # 读取scheme_content的内容为JSON
+        import json
+        scheme_data = json.loads(scheme_content)
+        robot_scheme = {}
+        robot_scheme['TaskId'] = scheme_data['TaskId']
+        robot_scheme['ExperimentName'] = scheme_data['ExperimentName']
+        robot_scheme['Containers'] = scheme_data['Containers']
+        robot_scheme['Equipments'] = scheme_data['Equipments']
+
+        # print(scheme_data)
+    except requests.exceptions.RequestException as e:
+        return {"status": "error", "message": f"Error reading scheme_url: {e}"}
+    
+    import requests
+    # url = "http://100.122.132.69:50000/sendScheme2RobotPlatform"
+    url = "http://localhost:50000/sendScheme2MobileRobot"
+    data = {"status": "ok"}
+    try:
+        response = requests.post(url, json=robot_scheme)
+        response.raise_for_status()
+        return str(response.json()) + "\n" + "task_id: " + task_id + "\n" + "scheme_url: " + scheme_url + "\n"
+    except requests.exceptions.RequestException as e:
+        print(f"Error sending scheme to robot workstation: {e}")
         return None
 
 if __name__ == "__main__":
-    print(sendScheme2RobotPlatform())
+    print(sendScheme2RobotWorkstation(task_id="task_20250122133404", scheme_url="https://s3-api.siat-mic.com/temp/robotExprimentScheme_task_20250122133404.json?AWSAccessKeyId=9bUtQL1Gpo9JB6o3pSGr&Signature=o9SGRdIN6h6%2BL8V7BCYALTc5s8k%3D&Expires=1737527678"))