fix: update uapi entries to match actual GPU service API specs

scripts/update_uapi_specs.py

@@ -0,0 +1,92 @@
+"""Update uapi entries to match actual GPU service API specs"""
+import asyncio
+from sqlor.dbpools import DBPools
+from appPublic.jsonConfig import getConfig
+
+async def main():
+    c = getConfig('.')
+    DBPools(c.databases)
+    db = DBPools()
+    async with db.sqlorContext('rag') as s:
+
+        headers_tmpl = '{\n    "Content-Type": "application/json"\n}'
+
+        # VDB — fix param names
+        vdb_fixes = [
+            ("vdb-upsert", "upsert", "/v1/upsert",
+             '{"colname": "{{colname}}", "data": {{json.dumps(data)}}}'),
+            ("vdb-search", "search", "/v1/query",
+             '{"colname": "{{colname}}", "vector": {{json.dumps(vector)}}, "pagerows": {{top_k}}, "output_fields": ["*"]}'),
+            ("vdb-delete", "delete", "/v1/delete",
+             '{"colname": "{{colname}}", "pks": {{json.dumps(ids)}}}'),
+        ]
+        for aid, name, path, data in vdb_fixes:
+            await s.sqlExe(
+                "UPDATE uapi SET name=" + repr(name) + ", path=" + repr(path) +
+                ", data=" + repr(data) + " WHERE id=" + repr(aid), {})
+            print(f"VDB: {aid} → {path}")
+
+        # CLIP — fix to /api/embed with texts+images
+        await s.sqlExe(
+            "UPDATE uapi SET data=" + repr(
+                '{"texts": {{json.dumps(texts)}}, "images": {{json.dumps(images)}}}'
+            ) + " WHERE id='emb-embed'", {})
+        print("CLIP: /api/embed updated")
+
+        # Reranker — add top_k parameter
+        await s.sqlExe(
+            "UPDATE uapi SET data=" + repr(
+                '{"query": "{{query}}", "documents": {{json.dumps(documents)}}, "top_k": {{top_k}}}'
+            ) + " WHERE id='reranker-rerank'", {})
+        print("Reranker: data updated")
+
+        # NER — fix path to /api/extract + add entities/threshold params
+        await s.sqlExe(
+            "UPDATE uapi SET path=" + repr("/api/extract") +
+            ", data=" + repr(
+                '{"text": "{{text}}", "entities": {{json.dumps(entities)}}, "threshold": {{threshold}}}'
+            ) + " WHERE id='ner-entities'", {})
+        print("NER: /api/extract updated")
+
+        # Face — fix param names (images list)
+        await s.sqlExe(
+            "UPDATE uapi SET data=" + repr('{"images": {{json.dumps(images)}}}') +
+            " WHERE id='face-detect'", {})
+        await s.sqlExe(
+            "UPDATE uapi SET data=" + repr('{"images": {{json.dumps(images)}}}') +
+            " WHERE id='face-recognize'", {})
+        await s.sqlExe(
+            "UPDATE uapi SET data=" + repr('{"image1": "{{image1}}", "image2": "{{image2}}"}') +
+            " WHERE id='face-compare'", {})
+        print("Face: params fixed")
+
+        # Graph — add missing endpoints
+        new_graph = [
+            ("graph-add-node", "rag-graph", "add_node", "添加节点", "POST",
+             "/api/graph/add_node",
+             '{"graph": "{{graph}}", "node_id": "{{node_id}}", "attrs": {{json.dumps(attrs)}}}',
+             "io-graph"),
+            ("graph-add-edge", "rag-graph", "add_edge", "添加边", "POST",
+             "/api/graph/add_edge",
+             '{"graph": "{{graph}}", "source": "{{source}}", "target": "{{target}}", "attrs": {{json.dumps(attrs)}}}',
+             "io-graph"),
+            ("graph-neighbors", "rag-graph", "neighbors", "邻居查询", "GET",
+             "/api/graph/neighbors",
+             '{"graph": "{{graph}}", "node_id": "{{node_id}}"}',
+             "io-graph"),
+        ]
+        for aid, upappid, name, title, method, path, data, ioid in new_graph:
+            await s.sqlExe(
+                "INSERT IGNORE INTO uapi (id, upappid, name, title, httpmethod, path, data, headers, ioid) "
+                "VALUES (" + repr(aid) + "," + repr(upappid) + "," + repr(name) + "," +
+                repr(title) + "," + repr(method) + "," + repr(path) + "," +
+                repr(data) + "," + repr(headers_tmpl) + "," + repr(ioid) + ")", {})
+            print(f"Graph: added {aid}")
+
+        # Verify all
+        print("\n=== All uapi entries ===")
+        for r in await s.sqlExe(
+            "SELECT id, upappid, name, httpmethod, path FROM uapi ORDER BY upappid, name", {}):
+            print(f"  {r.id:25s} {r.upappid:15s} {r.name:12s} {r.httpmethod:4s} {r.path}")
+
+asyncio.run(main())