migrate 2to4

_backend/tools.py

@@ -0,0 +1,86 @@
+import requests
+
+
+def retrieval_from_knowledge_base(
+        query: str,
+        topk: int
+        ) -> 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'
+    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 search_from_oqmd_by_composition(composition: str) -> str:
+        """
+        Search materials from OQMD database by chemical composition.
+
+        Args:
+            composition (str): Chemical composition string. Example: "CsPbBr3"
+
+        Returns:
+            str: JSON string containing material data or error message
+        """
+        # 构建请求参数
+        param = {
+            'composition': composition
+        }
+        tool_endpoint = "http://100.84.94.73:8020"
+        try:
+            # 发送请求到/oqmd/search路由
+            response = requests.get(
+                tool_endpoint + "/oqmd/search",
+                params=param
+            )
+            response.raise_for_status()
+            return str(response.json()['data'])
+        except requests.exceptions.RequestException as e:
+            return f"Error: {str(e)}"