完成api最后调试

nodes/chunk_and_embedding.py

@@ -135,5 +135,4 @@ def main(scrape_json: list, DASHSCOPE_API_KEY: str) -> dict:
     '''
     return {
         "results": result_list,
-        "len_chunks": len(chunks)
     }

nodes/iteration_output.py

@@ -1,16 +1,25 @@
-
-def main(iteration_output: dict, url: str, len_chunks: float, body: str, status_code: float):
+def check_status(status_code: float, body: str):
+    '''
+    检查状态码和约定的返回值
+    '''
+    if status_code != 200:
+        raise Exception(f"注册任务失败，状态码：{status_code}")
+    if "code" not in body or body["code"] != 1:
+        raise Exception(f"注册任务失败，返回值：{body}")
+def main(iteration_output: list[dict], status_code: float, body: str):
     '''
     迭代器运行结果记录
     iteration_output: 迭代器运行结果记录
-    url: 待处理的URL
-    len_chunks: 分块后的文本块数量
-    body: 从URL获取的响应体
     status_code: 从URL获取的响应状态码
+    body: 从URL获取的响应体
     '''
-    if iteration_output is None:
-        iteration_output = {}
+    try:
+        check_status(status_code, body)
+        data = body["data"]
+        iteration_output.append(data)
+    except Exception as e:
+        raise e
     
     return {
-        "result": arg1 + arg2,
+        "iteration_output": iteration_output,
     }

nodes/parse_pending_urls.py

@@ -17,5 +17,5 @@ def main(status_code: float, body: str):
 
     return {
         "urls": urls,
-        "iteration_output": {} # 迭代器运行结果记录，用于传入给迭代器进行记录运行情况 
+        "iteration_output": [] # 迭代器运行结果记录，用于传入给迭代器进行记录运行情况 
     }