test

app.py

@@ -40,7 +40,7 @@ raw_jsons = json.loads(decrypted_content)
 docs = []
 metas = []
 
-for vhjx_item in raw_jsons[:1]:
+for vhjx_index, vhjx_item in enumerate(raw_jsons):
     chapter = vhjx_item[0]
     for jvvi_item in vhjx_item[1:]:
         content = jvvi_item["原文"]
@@ -52,28 +52,29 @@ for vhjx_item in raw_jsons[:1]:
             "critique": jvvi_item.get("批判", ""),
             "chapter": chapter
         })
+    
+    # 一个章节一次
+    # 批量生成 embeddings（每个为 list[float]）
+    emb_result = llm.create_embedding(docs)
+    embeddings = [item["embedding"] for item in emb_result["data"]]  # List[List[float]]
 
-# 4. 批量生成 embeddings（每个为 list[float]）
-emb_result = llm.create_embedding(docs)
-embeddings = [item["embedding"] for item in emb_result["data"]]  # List[List[float]]
-
-# 准备数据
-milvus_data = []
-for i, emb in enumerate(embeddings):
-    item = metas[i]
-    milvus_data.append({
-        "id" : i,
-        "index": item["index"],
-        "vector": emb,
-        "text": item["text"],
-        "annotation": item["annotation"],
-        "critique": item["critique"],
-        "chapter": item["chapter"]
-    })
+    # 准备数据
+    milvus_data = []
+    for i, emb in enumerate(embeddings):
+        item = metas[i]
+        milvus_data.append({
+            "id" : vhjx_index * 100 + i,
+            "index": item["index"],
+            "vector": emb,
+            "text": item["text"],
+            "annotation": item["annotation"],
+            "critique": item["critique"],
+            "chapter": item["chapter"]
+        })
 
-# 插入数据
-client.insert(collection_name="collection_1", data=milvus_data)
-print(f"✅ 插入完成：共 {len(milvus_data)} 条数据")
+    # 插入数据
+    client.insert(collection_name="collection_1", data=milvus_data)
+    print(f"✅ 插入完成：共 {len(milvus_data)} 条数据")
 
 def greet(name):
     embeddings = llm.create_embedding(name)
@@ -83,8 +84,7 @@ def greet(name):
         limit=2,
         output_fields=["text", "id"],
     )
-    print(res)
-    return "Hello " + name + "!!"
+    return res
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+demo = gr.Interface(fn=greet, inputs="text", outputs=gr.JSON(label="查询结果"))
 demo.launch(mcp_server=True)